OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
| 5 #include "base/synchronization/lock.h" |
| 6 |
5 #include <stdlib.h> | 7 #include <stdlib.h> |
6 | 8 |
7 #include "base/synchronization/lock.h" | 9 #include "base/compiler_specific.h" |
8 #include "base/threading/platform_thread.h" | 10 #include "base/threading/platform_thread.h" |
9 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
10 | 12 |
11 namespace base { | 13 namespace base { |
12 | 14 |
13 // Basic test to make sure that Acquire()/Release()/Try() don't crash ---------- | 15 // Basic test to make sure that Acquire()/Release()/Try() don't crash ---------- |
14 | 16 |
15 class BasicLockTestThread : public PlatformThread::Delegate { | 17 class BasicLockTestThread : public PlatformThread::Delegate { |
16 public: | 18 public: |
17 BasicLockTestThread(Lock* lock) : lock_(lock), acquired_(0) {} | 19 BasicLockTestThread(Lock* lock) : lock_(lock), acquired_(0) {} |
18 | 20 |
19 virtual void ThreadMain() { | 21 virtual void ThreadMain() OVERRIDE { |
20 for (int i = 0; i < 10; i++) { | 22 for (int i = 0; i < 10; i++) { |
21 lock_->Acquire(); | 23 lock_->Acquire(); |
22 acquired_++; | 24 acquired_++; |
23 lock_->Release(); | 25 lock_->Release(); |
24 } | 26 } |
25 for (int i = 0; i < 10; i++) { | 27 for (int i = 0; i < 10; i++) { |
26 lock_->Acquire(); | 28 lock_->Acquire(); |
27 acquired_++; | 29 acquired_++; |
28 PlatformThread::Sleep(TimeDelta::FromMilliseconds(rand() % 20)); | 30 PlatformThread::Sleep(TimeDelta::FromMilliseconds(rand() % 20)); |
29 lock_->Release(); | 31 lock_->Release(); |
(...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
84 EXPECT_GE(acquired, 20); | 86 EXPECT_GE(acquired, 20); |
85 EXPECT_GE(thread.acquired(), 20); | 87 EXPECT_GE(thread.acquired(), 20); |
86 } | 88 } |
87 | 89 |
88 // Test that Try() works as expected ------------------------------------------- | 90 // Test that Try() works as expected ------------------------------------------- |
89 | 91 |
90 class TryLockTestThread : public PlatformThread::Delegate { | 92 class TryLockTestThread : public PlatformThread::Delegate { |
91 public: | 93 public: |
92 TryLockTestThread(Lock* lock) : lock_(lock), got_lock_(false) {} | 94 TryLockTestThread(Lock* lock) : lock_(lock), got_lock_(false) {} |
93 | 95 |
94 virtual void ThreadMain() { | 96 virtual void ThreadMain() OVERRIDE { |
95 got_lock_ = lock_->Try(); | 97 got_lock_ = lock_->Try(); |
96 if (got_lock_) | 98 if (got_lock_) |
97 lock_->Release(); | 99 lock_->Release(); |
98 } | 100 } |
99 | 101 |
100 bool got_lock() const { return got_lock_; } | 102 bool got_lock() const { return got_lock_; } |
101 | 103 |
102 private: | 104 private: |
103 Lock* lock_; | 105 Lock* lock_; |
104 bool got_lock_; | 106 bool got_lock_; |
(...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
153 static void DoStuff(Lock* lock, int* value) { | 155 static void DoStuff(Lock* lock, int* value) { |
154 for (int i = 0; i < 40; i++) { | 156 for (int i = 0; i < 40; i++) { |
155 lock->Acquire(); | 157 lock->Acquire(); |
156 int v = *value; | 158 int v = *value; |
157 PlatformThread::Sleep(TimeDelta::FromMilliseconds(rand() % 10)); | 159 PlatformThread::Sleep(TimeDelta::FromMilliseconds(rand() % 10)); |
158 *value = v + 1; | 160 *value = v + 1; |
159 lock->Release(); | 161 lock->Release(); |
160 } | 162 } |
161 } | 163 } |
162 | 164 |
163 virtual void ThreadMain() { | 165 virtual void ThreadMain() OVERRIDE { |
164 DoStuff(lock_, value_); | 166 DoStuff(lock_, value_); |
165 } | 167 } |
166 | 168 |
167 private: | 169 private: |
168 Lock* lock_; | 170 Lock* lock_; |
169 int* value_; | 171 int* value_; |
170 | 172 |
171 DISALLOW_COPY_AND_ASSIGN(MutexLockTestThread); | 173 DISALLOW_COPY_AND_ASSIGN(MutexLockTestThread); |
172 }; | 174 }; |
173 | 175 |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
205 MutexLockTestThread::DoStuff(&lock, &value); | 207 MutexLockTestThread::DoStuff(&lock, &value); |
206 | 208 |
207 PlatformThread::Join(handle1); | 209 PlatformThread::Join(handle1); |
208 PlatformThread::Join(handle2); | 210 PlatformThread::Join(handle2); |
209 PlatformThread::Join(handle3); | 211 PlatformThread::Join(handle3); |
210 | 212 |
211 EXPECT_EQ(4 * 40, value); | 213 EXPECT_EQ(4 * 40, value); |
212 } | 214 } |
213 | 215 |
214 } // namespace base | 216 } // namespace base |
OLD | NEW |