OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/timer/timer.h" | 5 #include "base/timer/timer.h" |
6 | 6 |
7 #include <stddef.h> | 7 #include <stddef.h> |
8 | 8 |
9 #include <memory> | 9 #include <memory> |
10 | 10 |
11 #include "base/macros.h" | 11 #include "base/macros.h" |
12 #include "base/message_loop/message_loop.h" | 12 #include "base/message_loop/message_loop.h" |
| 13 #include "base/run_loop.h" |
13 #include "base/test/test_simple_task_runner.h" | 14 #include "base/test/test_simple_task_runner.h" |
14 #include "base/threading/thread_task_runner_handle.h" | 15 #include "base/threading/thread_task_runner_handle.h" |
15 #include "build/build_config.h" | 16 #include "build/build_config.h" |
16 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
17 | 18 |
18 using base::TimeDelta; | 19 using base::TimeDelta; |
19 using base::SingleThreadTaskRunner; | 20 using base::SingleThreadTaskRunner; |
20 | 21 |
21 namespace { | 22 namespace { |
22 | 23 |
(...skipping 86 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
109 base::RepeatingTimer timer_; | 110 base::RepeatingTimer timer_; |
110 }; | 111 }; |
111 | 112 |
112 void RunTest_OneShotTimer(base::MessageLoop::Type message_loop_type) { | 113 void RunTest_OneShotTimer(base::MessageLoop::Type message_loop_type) { |
113 base::MessageLoop loop(message_loop_type); | 114 base::MessageLoop loop(message_loop_type); |
114 | 115 |
115 bool did_run = false; | 116 bool did_run = false; |
116 OneShotTimerTester f(&did_run); | 117 OneShotTimerTester f(&did_run); |
117 f.Start(); | 118 f.Start(); |
118 | 119 |
119 base::MessageLoop::current()->Run(); | 120 base::RunLoop().Run(); |
120 | 121 |
121 EXPECT_TRUE(did_run); | 122 EXPECT_TRUE(did_run); |
122 } | 123 } |
123 | 124 |
124 void RunTest_OneShotTimer_Cancel(base::MessageLoop::Type message_loop_type) { | 125 void RunTest_OneShotTimer_Cancel(base::MessageLoop::Type message_loop_type) { |
125 base::MessageLoop loop(message_loop_type); | 126 base::MessageLoop loop(message_loop_type); |
126 | 127 |
127 bool did_run_a = false; | 128 bool did_run_a = false; |
128 OneShotTimerTester* a = new OneShotTimerTester(&did_run_a); | 129 OneShotTimerTester* a = new OneShotTimerTester(&did_run_a); |
129 | 130 |
130 // This should run before the timer expires. | 131 // This should run before the timer expires. |
131 base::ThreadTaskRunnerHandle::Get()->DeleteSoon(FROM_HERE, a); | 132 base::ThreadTaskRunnerHandle::Get()->DeleteSoon(FROM_HERE, a); |
132 | 133 |
133 // Now start the timer. | 134 // Now start the timer. |
134 a->Start(); | 135 a->Start(); |
135 | 136 |
136 bool did_run_b = false; | 137 bool did_run_b = false; |
137 OneShotTimerTester b(&did_run_b); | 138 OneShotTimerTester b(&did_run_b); |
138 b.Start(); | 139 b.Start(); |
139 | 140 |
140 base::MessageLoop::current()->Run(); | 141 base::RunLoop().Run(); |
141 | 142 |
142 EXPECT_FALSE(did_run_a); | 143 EXPECT_FALSE(did_run_a); |
143 EXPECT_TRUE(did_run_b); | 144 EXPECT_TRUE(did_run_b); |
144 } | 145 } |
145 | 146 |
146 void RunTest_OneShotSelfDeletingTimer( | 147 void RunTest_OneShotSelfDeletingTimer( |
147 base::MessageLoop::Type message_loop_type) { | 148 base::MessageLoop::Type message_loop_type) { |
148 base::MessageLoop loop(message_loop_type); | 149 base::MessageLoop loop(message_loop_type); |
149 | 150 |
150 bool did_run = false; | 151 bool did_run = false; |
151 OneShotSelfDeletingTimerTester f(&did_run); | 152 OneShotSelfDeletingTimerTester f(&did_run); |
152 f.Start(); | 153 f.Start(); |
153 | 154 |
154 base::MessageLoop::current()->Run(); | 155 base::RunLoop().Run(); |
155 | 156 |
156 EXPECT_TRUE(did_run); | 157 EXPECT_TRUE(did_run); |
157 } | 158 } |
158 | 159 |
159 void RunTest_RepeatingTimer(base::MessageLoop::Type message_loop_type, | 160 void RunTest_RepeatingTimer(base::MessageLoop::Type message_loop_type, |
160 const TimeDelta& delay) { | 161 const TimeDelta& delay) { |
161 base::MessageLoop loop(message_loop_type); | 162 base::MessageLoop loop(message_loop_type); |
162 | 163 |
163 bool did_run = false; | 164 bool did_run = false; |
164 RepeatingTimerTester f(&did_run, delay); | 165 RepeatingTimerTester f(&did_run, delay); |
165 f.Start(); | 166 f.Start(); |
166 | 167 |
167 base::MessageLoop::current()->Run(); | 168 base::RunLoop().Run(); |
168 | 169 |
169 EXPECT_TRUE(did_run); | 170 EXPECT_TRUE(did_run); |
170 } | 171 } |
171 | 172 |
172 void RunTest_RepeatingTimer_Cancel(base::MessageLoop::Type message_loop_type, | 173 void RunTest_RepeatingTimer_Cancel(base::MessageLoop::Type message_loop_type, |
173 const TimeDelta& delay) { | 174 const TimeDelta& delay) { |
174 base::MessageLoop loop(message_loop_type); | 175 base::MessageLoop loop(message_loop_type); |
175 | 176 |
176 bool did_run_a = false; | 177 bool did_run_a = false; |
177 RepeatingTimerTester* a = new RepeatingTimerTester(&did_run_a, delay); | 178 RepeatingTimerTester* a = new RepeatingTimerTester(&did_run_a, delay); |
178 | 179 |
179 // This should run before the timer expires. | 180 // This should run before the timer expires. |
180 base::ThreadTaskRunnerHandle::Get()->DeleteSoon(FROM_HERE, a); | 181 base::ThreadTaskRunnerHandle::Get()->DeleteSoon(FROM_HERE, a); |
181 | 182 |
182 // Now start the timer. | 183 // Now start the timer. |
183 a->Start(); | 184 a->Start(); |
184 | 185 |
185 bool did_run_b = false; | 186 bool did_run_b = false; |
186 RepeatingTimerTester b(&did_run_b, delay); | 187 RepeatingTimerTester b(&did_run_b, delay); |
187 b.Start(); | 188 b.Start(); |
188 | 189 |
189 base::MessageLoop::current()->Run(); | 190 base::RunLoop().Run(); |
190 | 191 |
191 EXPECT_FALSE(did_run_a); | 192 EXPECT_FALSE(did_run_a); |
192 EXPECT_TRUE(did_run_b); | 193 EXPECT_TRUE(did_run_b); |
193 } | 194 } |
194 | 195 |
195 class DelayTimerTarget { | 196 class DelayTimerTarget { |
196 public: | 197 public: |
197 bool signaled() const { return signaled_; } | 198 bool signaled() const { return signaled_; } |
198 | 199 |
199 void Signal() { | 200 void Signal() { |
200 ASSERT_FALSE(signaled_); | 201 ASSERT_FALSE(signaled_); |
201 signaled_ = true; | 202 signaled_ = true; |
202 } | 203 } |
203 | 204 |
204 private: | 205 private: |
205 bool signaled_ = false; | 206 bool signaled_ = false; |
206 }; | 207 }; |
207 | 208 |
208 void RunTest_DelayTimer_NoCall(base::MessageLoop::Type message_loop_type) { | 209 void RunTest_DelayTimer_NoCall(base::MessageLoop::Type message_loop_type) { |
209 base::MessageLoop loop(message_loop_type); | 210 base::MessageLoop loop(message_loop_type); |
210 | 211 |
211 // If Delay is never called, the timer shouldn't go off. | 212 // If Delay is never called, the timer shouldn't go off. |
212 DelayTimerTarget target; | 213 DelayTimerTarget target; |
213 base::DelayTimer timer(FROM_HERE, TimeDelta::FromMilliseconds(1), &target, | 214 base::DelayTimer timer(FROM_HERE, TimeDelta::FromMilliseconds(1), &target, |
214 &DelayTimerTarget::Signal); | 215 &DelayTimerTarget::Signal); |
215 | 216 |
216 bool did_run = false; | 217 bool did_run = false; |
217 OneShotTimerTester tester(&did_run); | 218 OneShotTimerTester tester(&did_run); |
218 tester.Start(); | 219 tester.Start(); |
219 base::MessageLoop::current()->Run(); | 220 base::RunLoop().Run(); |
220 | 221 |
221 ASSERT_FALSE(target.signaled()); | 222 ASSERT_FALSE(target.signaled()); |
222 } | 223 } |
223 | 224 |
224 void RunTest_DelayTimer_OneCall(base::MessageLoop::Type message_loop_type) { | 225 void RunTest_DelayTimer_OneCall(base::MessageLoop::Type message_loop_type) { |
225 base::MessageLoop loop(message_loop_type); | 226 base::MessageLoop loop(message_loop_type); |
226 | 227 |
227 DelayTimerTarget target; | 228 DelayTimerTarget target; |
228 base::DelayTimer timer(FROM_HERE, TimeDelta::FromMilliseconds(1), &target, | 229 base::DelayTimer timer(FROM_HERE, TimeDelta::FromMilliseconds(1), &target, |
229 &DelayTimerTarget::Signal); | 230 &DelayTimerTarget::Signal); |
230 timer.Reset(); | 231 timer.Reset(); |
231 | 232 |
232 bool did_run = false; | 233 bool did_run = false; |
233 OneShotTimerTester tester(&did_run, 100 /* milliseconds */); | 234 OneShotTimerTester tester(&did_run, 100 /* milliseconds */); |
234 tester.Start(); | 235 tester.Start(); |
235 base::MessageLoop::current()->Run(); | 236 base::RunLoop().Run(); |
236 | 237 |
237 ASSERT_TRUE(target.signaled()); | 238 ASSERT_TRUE(target.signaled()); |
238 } | 239 } |
239 | 240 |
240 struct ResetHelper { | 241 struct ResetHelper { |
241 ResetHelper(base::DelayTimer* timer, DelayTimerTarget* target) | 242 ResetHelper(base::DelayTimer* timer, DelayTimerTarget* target) |
242 : timer_(timer), target_(target) {} | 243 : timer_(timer), target_(target) {} |
243 | 244 |
244 void Reset() { | 245 void Reset() { |
245 ASSERT_FALSE(target_->signaled()); | 246 ASSERT_FALSE(target_->signaled()); |
(...skipping 18 matching lines...) Expand all Loading... |
264 | 265 |
265 base::OneShotTimer timers[20]; | 266 base::OneShotTimer timers[20]; |
266 for (size_t i = 0; i < arraysize(timers); ++i) { | 267 for (size_t i = 0; i < arraysize(timers); ++i) { |
267 timers[i].Start(FROM_HERE, TimeDelta::FromMilliseconds(i * 10), | 268 timers[i].Start(FROM_HERE, TimeDelta::FromMilliseconds(i * 10), |
268 &reset_helper, &ResetHelper::Reset); | 269 &reset_helper, &ResetHelper::Reset); |
269 } | 270 } |
270 | 271 |
271 bool did_run = false; | 272 bool did_run = false; |
272 OneShotTimerTester tester(&did_run, 300); | 273 OneShotTimerTester tester(&did_run, 300); |
273 tester.Start(); | 274 tester.Start(); |
274 base::MessageLoop::current()->Run(); | 275 base::RunLoop().Run(); |
275 | 276 |
276 ASSERT_TRUE(target.signaled()); | 277 ASSERT_TRUE(target.signaled()); |
277 } | 278 } |
278 | 279 |
279 class DelayTimerFatalTarget { | 280 class DelayTimerFatalTarget { |
280 public: | 281 public: |
281 void Signal() { | 282 void Signal() { |
282 ASSERT_TRUE(false); | 283 ASSERT_TRUE(false); |
283 } | 284 } |
284 }; | 285 }; |
(...skipping 222 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
507 TEST(TimerTest, ContinuationStopStart) { | 508 TEST(TimerTest, ContinuationStopStart) { |
508 { | 509 { |
509 ClearAllCallbackHappened(); | 510 ClearAllCallbackHappened(); |
510 base::MessageLoop loop; | 511 base::MessageLoop loop; |
511 base::Timer timer(false, false); | 512 base::Timer timer(false, false); |
512 timer.Start(FROM_HERE, TimeDelta::FromMilliseconds(10), | 513 timer.Start(FROM_HERE, TimeDelta::FromMilliseconds(10), |
513 base::Bind(&SetCallbackHappened1)); | 514 base::Bind(&SetCallbackHappened1)); |
514 timer.Stop(); | 515 timer.Stop(); |
515 timer.Start(FROM_HERE, TimeDelta::FromMilliseconds(40), | 516 timer.Start(FROM_HERE, TimeDelta::FromMilliseconds(40), |
516 base::Bind(&SetCallbackHappened2)); | 517 base::Bind(&SetCallbackHappened2)); |
517 base::MessageLoop::current()->Run(); | 518 base::RunLoop().Run(); |
518 EXPECT_FALSE(g_callback_happened1); | 519 EXPECT_FALSE(g_callback_happened1); |
519 EXPECT_TRUE(g_callback_happened2); | 520 EXPECT_TRUE(g_callback_happened2); |
520 } | 521 } |
521 } | 522 } |
522 | 523 |
523 TEST(TimerTest, ContinuationReset) { | 524 TEST(TimerTest, ContinuationReset) { |
524 { | 525 { |
525 ClearAllCallbackHappened(); | 526 ClearAllCallbackHappened(); |
526 base::MessageLoop loop; | 527 base::MessageLoop loop; |
527 base::Timer timer(false, false); | 528 base::Timer timer(false, false); |
528 timer.Start(FROM_HERE, TimeDelta::FromMilliseconds(10), | 529 timer.Start(FROM_HERE, TimeDelta::FromMilliseconds(10), |
529 base::Bind(&SetCallbackHappened1)); | 530 base::Bind(&SetCallbackHappened1)); |
530 timer.Reset(); | 531 timer.Reset(); |
531 // Since Reset happened before task ran, the user_task must not be cleared: | 532 // Since Reset happened before task ran, the user_task must not be cleared: |
532 ASSERT_FALSE(timer.user_task().is_null()); | 533 ASSERT_FALSE(timer.user_task().is_null()); |
533 base::MessageLoop::current()->Run(); | 534 base::RunLoop().Run(); |
534 EXPECT_TRUE(g_callback_happened1); | 535 EXPECT_TRUE(g_callback_happened1); |
535 } | 536 } |
536 } | 537 } |
537 | 538 |
538 } // namespace | 539 } // namespace |
OLD | NEW |