| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include <deque> | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/bind_helpers.h" | |
| 9 #include "base/test/test_pending_task.h" | |
| 10 #include "base/test/test_simple_task_runner.h" | |
| 11 #include "cc/base/delayed_unique_notifier.h" | |
| 12 #include "testing/gtest/include/gtest/gtest.h" | |
| 13 | |
| 14 namespace cc { | |
| 15 namespace { | |
| 16 | |
| 17 class TestNotifier : public DelayedUniqueNotifier { | |
| 18 public: | |
| 19 TestNotifier(base::SequencedTaskRunner* task_runner, | |
| 20 const base::Closure& closure, | |
| 21 const base::TimeDelta& delay) | |
| 22 : DelayedUniqueNotifier(task_runner, closure, delay) {} | |
| 23 ~TestNotifier() override {} | |
| 24 | |
| 25 // Overridden from DelayedUniqueNotifier: | |
| 26 base::TimeTicks Now() const override { return now_; } | |
| 27 | |
| 28 void SetNow(base::TimeTicks now) { now_ = now; } | |
| 29 | |
| 30 private: | |
| 31 base::TimeTicks now_; | |
| 32 }; | |
| 33 | |
| 34 class DelayedUniqueNotifierTest : public testing::Test { | |
| 35 public: | |
| 36 DelayedUniqueNotifierTest() : notification_count_(0) {} | |
| 37 | |
| 38 void SetUp() override { | |
| 39 notification_count_ = 0; | |
| 40 task_runner_ = make_scoped_refptr(new base::TestSimpleTaskRunner); | |
| 41 } | |
| 42 | |
| 43 void Notify() { ++notification_count_; } | |
| 44 | |
| 45 int NotificationCount() const { return notification_count_; } | |
| 46 | |
| 47 std::deque<base::TestPendingTask> TakePendingTasks() { | |
| 48 std::deque<base::TestPendingTask> tasks = task_runner_->GetPendingTasks(); | |
| 49 task_runner_->ClearPendingTasks(); | |
| 50 return tasks; | |
| 51 } | |
| 52 | |
| 53 protected: | |
| 54 int notification_count_; | |
| 55 scoped_refptr<base::TestSimpleTaskRunner> task_runner_; | |
| 56 }; | |
| 57 | |
| 58 TEST_F(DelayedUniqueNotifierTest, ZeroDelay) { | |
| 59 base::TimeDelta delay = base::TimeDelta::FromInternalValue(0); | |
| 60 TestNotifier notifier( | |
| 61 task_runner_.get(), | |
| 62 base::Bind(&DelayedUniqueNotifierTest::Notify, base::Unretained(this)), | |
| 63 delay); | |
| 64 | |
| 65 EXPECT_EQ(0, NotificationCount()); | |
| 66 | |
| 67 // Basic schedule for |delay| from now. | |
| 68 base::TimeTicks schedule_time = | |
| 69 base::TimeTicks() + base::TimeDelta::FromInternalValue(10); | |
| 70 | |
| 71 notifier.SetNow(schedule_time); | |
| 72 notifier.Schedule(); | |
| 73 | |
| 74 std::deque<base::TestPendingTask> tasks = TakePendingTasks(); | |
| 75 ASSERT_EQ(1u, tasks.size()); | |
| 76 EXPECT_EQ(base::TimeTicks() + delay, tasks[0].GetTimeToRun()); | |
| 77 | |
| 78 tasks[0].task.Run(); | |
| 79 EXPECT_EQ(1, NotificationCount()); | |
| 80 | |
| 81 // 5 schedules should result in only one run. | |
| 82 for (int i = 0; i < 5; ++i) | |
| 83 notifier.Schedule(); | |
| 84 | |
| 85 tasks = TakePendingTasks(); | |
| 86 ASSERT_EQ(1u, tasks.size()); | |
| 87 EXPECT_EQ(base::TimeTicks() + delay, tasks[0].GetTimeToRun()); | |
| 88 | |
| 89 tasks[0].task.Run(); | |
| 90 EXPECT_EQ(2, NotificationCount()); | |
| 91 } | |
| 92 | |
| 93 TEST_F(DelayedUniqueNotifierTest, SmallDelay) { | |
| 94 base::TimeDelta delay = base::TimeDelta::FromInternalValue(20); | |
| 95 TestNotifier notifier( | |
| 96 task_runner_.get(), | |
| 97 base::Bind(&DelayedUniqueNotifierTest::Notify, base::Unretained(this)), | |
| 98 delay); | |
| 99 | |
| 100 EXPECT_EQ(0, NotificationCount()); | |
| 101 | |
| 102 // Basic schedule for |delay| from now (now: 30, run time: 50). | |
| 103 base::TimeTicks schedule_time = | |
| 104 base::TimeTicks() + base::TimeDelta::FromInternalValue(30); | |
| 105 | |
| 106 notifier.SetNow(schedule_time); | |
| 107 notifier.Schedule(); | |
| 108 | |
| 109 std::deque<base::TestPendingTask> tasks = TakePendingTasks(); | |
| 110 | |
| 111 ASSERT_EQ(1u, tasks.size()); | |
| 112 EXPECT_EQ(base::TimeTicks() + delay, tasks[0].GetTimeToRun()); | |
| 113 | |
| 114 // It's not yet time to run, so we expect no notifications. | |
| 115 tasks[0].task.Run(); | |
| 116 EXPECT_EQ(0, NotificationCount()); | |
| 117 | |
| 118 tasks = TakePendingTasks(); | |
| 119 | |
| 120 ASSERT_EQ(1u, tasks.size()); | |
| 121 // Now the time should be delay minus whatever the value of now happens to be | |
| 122 // (now: 30, run time: 50). | |
| 123 base::TimeTicks scheduled_run_time = notifier.Now() + delay; | |
| 124 base::TimeTicks scheduled_delay = | |
| 125 base::TimeTicks() + (scheduled_run_time - notifier.Now()); | |
| 126 EXPECT_EQ(scheduled_delay, tasks[0].GetTimeToRun()); | |
| 127 | |
| 128 // Move closer to the run time (time: 49, run time: 50). | |
| 129 notifier.SetNow(notifier.Now() + base::TimeDelta::FromInternalValue(19)); | |
| 130 | |
| 131 // It's not yet time to run, so we expect no notifications. | |
| 132 tasks[0].task.Run(); | |
| 133 EXPECT_EQ(0, NotificationCount()); | |
| 134 | |
| 135 tasks = TakePendingTasks(); | |
| 136 ASSERT_EQ(1u, tasks.size()); | |
| 137 | |
| 138 // Now the time should be delay minus whatever the value of now happens to be. | |
| 139 scheduled_delay = base::TimeTicks() + (scheduled_run_time - notifier.Now()); | |
| 140 EXPECT_EQ(scheduled_delay, tasks[0].GetTimeToRun()); | |
| 141 | |
| 142 // Move to exactly the run time (time: 50, run time: 50). | |
| 143 notifier.SetNow(notifier.Now() + base::TimeDelta::FromInternalValue(1)); | |
| 144 | |
| 145 // It's time to run! | |
| 146 tasks[0].task.Run(); | |
| 147 EXPECT_EQ(1, NotificationCount()); | |
| 148 | |
| 149 tasks = TakePendingTasks(); | |
| 150 EXPECT_EQ(0u, tasks.size()); | |
| 151 } | |
| 152 | |
| 153 TEST_F(DelayedUniqueNotifierTest, RescheduleDelay) { | |
| 154 base::TimeDelta delay = base::TimeDelta::FromInternalValue(20); | |
| 155 TestNotifier notifier( | |
| 156 task_runner_.get(), | |
| 157 base::Bind(&DelayedUniqueNotifierTest::Notify, base::Unretained(this)), | |
| 158 delay); | |
| 159 | |
| 160 base::TimeTicks schedule_time; | |
| 161 // Move time 19 units forward and reschedule, expecting that we still need to | |
| 162 // run in |delay| time and we don't get a notification. | |
| 163 for (int i = 0; i < 10; ++i) { | |
| 164 EXPECT_EQ(0, NotificationCount()); | |
| 165 | |
| 166 // Move time forward 19 units. | |
| 167 schedule_time = notifier.Now() + base::TimeDelta::FromInternalValue(19); | |
| 168 notifier.SetNow(schedule_time); | |
| 169 notifier.Schedule(); | |
| 170 | |
| 171 std::deque<base::TestPendingTask> tasks = TakePendingTasks(); | |
| 172 | |
| 173 ASSERT_EQ(1u, tasks.size()); | |
| 174 EXPECT_EQ(base::TimeTicks() + delay, tasks[0].GetTimeToRun()); | |
| 175 | |
| 176 // It's not yet time to run, so we expect no notifications. | |
| 177 tasks[0].task.Run(); | |
| 178 EXPECT_EQ(0, NotificationCount()); | |
| 179 } | |
| 180 | |
| 181 // Move time forward 20 units, expecting a notification. | |
| 182 schedule_time = notifier.Now() + base::TimeDelta::FromInternalValue(20); | |
| 183 notifier.SetNow(schedule_time); | |
| 184 | |
| 185 std::deque<base::TestPendingTask> tasks = TakePendingTasks(); | |
| 186 | |
| 187 ASSERT_EQ(1u, tasks.size()); | |
| 188 EXPECT_EQ(base::TimeTicks() + delay, tasks[0].GetTimeToRun()); | |
| 189 | |
| 190 // Time to run! | |
| 191 tasks[0].task.Run(); | |
| 192 EXPECT_EQ(1, NotificationCount()); | |
| 193 } | |
| 194 | |
| 195 TEST_F(DelayedUniqueNotifierTest, CancelAndHasPendingNotification) { | |
| 196 base::TimeDelta delay = base::TimeDelta::FromInternalValue(20); | |
| 197 TestNotifier notifier( | |
| 198 task_runner_.get(), | |
| 199 base::Bind(&DelayedUniqueNotifierTest::Notify, base::Unretained(this)), | |
| 200 delay); | |
| 201 | |
| 202 EXPECT_EQ(0, NotificationCount()); | |
| 203 | |
| 204 // Schedule for |delay| seconds from now. | |
| 205 base::TimeTicks schedule_time = | |
| 206 notifier.Now() + base::TimeDelta::FromInternalValue(10); | |
| 207 notifier.SetNow(schedule_time); | |
| 208 notifier.Schedule(); | |
| 209 EXPECT_TRUE(notifier.HasPendingNotification()); | |
| 210 | |
| 211 // Cancel the run. | |
| 212 notifier.Cancel(); | |
| 213 EXPECT_FALSE(notifier.HasPendingNotification()); | |
| 214 | |
| 215 std::deque<base::TestPendingTask> tasks = TakePendingTasks(); | |
| 216 | |
| 217 ASSERT_EQ(1u, tasks.size()); | |
| 218 EXPECT_EQ(base::TimeTicks() + delay, tasks[0].GetTimeToRun()); | |
| 219 | |
| 220 // Time to run, but a canceled task! | |
| 221 tasks[0].task.Run(); | |
| 222 EXPECT_EQ(0, NotificationCount()); | |
| 223 EXPECT_FALSE(notifier.HasPendingNotification()); | |
| 224 | |
| 225 tasks = TakePendingTasks(); | |
| 226 EXPECT_EQ(0u, tasks.size()); | |
| 227 | |
| 228 notifier.Schedule(); | |
| 229 EXPECT_TRUE(notifier.HasPendingNotification()); | |
| 230 tasks = TakePendingTasks(); | |
| 231 | |
| 232 ASSERT_EQ(1u, tasks.size()); | |
| 233 EXPECT_EQ(base::TimeTicks() + delay, tasks[0].GetTimeToRun()); | |
| 234 | |
| 235 // Advance the time. | |
| 236 notifier.SetNow(notifier.Now() + delay); | |
| 237 | |
| 238 // This should run since it wasn't canceled. | |
| 239 tasks[0].task.Run(); | |
| 240 EXPECT_EQ(1, NotificationCount()); | |
| 241 EXPECT_FALSE(notifier.HasPendingNotification()); | |
| 242 | |
| 243 for (int i = 0; i < 10; ++i) { | |
| 244 notifier.Schedule(); | |
| 245 EXPECT_TRUE(notifier.HasPendingNotification()); | |
| 246 notifier.Cancel(); | |
| 247 EXPECT_FALSE(notifier.HasPendingNotification()); | |
| 248 } | |
| 249 | |
| 250 tasks = TakePendingTasks(); | |
| 251 | |
| 252 ASSERT_EQ(1u, tasks.size()); | |
| 253 EXPECT_EQ(base::TimeTicks() + delay, tasks[0].GetTimeToRun()); | |
| 254 | |
| 255 // Time to run, but a canceled task! | |
| 256 notifier.SetNow(notifier.Now() + delay); | |
| 257 tasks[0].task.Run(); | |
| 258 EXPECT_EQ(1, NotificationCount()); | |
| 259 | |
| 260 tasks = TakePendingTasks(); | |
| 261 EXPECT_EQ(0u, tasks.size()); | |
| 262 EXPECT_FALSE(notifier.HasPendingNotification()); | |
| 263 } | |
| 264 | |
| 265 TEST_F(DelayedUniqueNotifierTest, ShutdownWithScheduledTask) { | |
| 266 base::TimeDelta delay = base::TimeDelta::FromInternalValue(20); | |
| 267 TestNotifier notifier( | |
| 268 task_runner_.get(), | |
| 269 base::Bind(&DelayedUniqueNotifierTest::Notify, base::Unretained(this)), | |
| 270 delay); | |
| 271 | |
| 272 EXPECT_EQ(0, NotificationCount()); | |
| 273 | |
| 274 // Schedule for |delay| seconds from now. | |
| 275 base::TimeTicks schedule_time = | |
| 276 notifier.Now() + base::TimeDelta::FromInternalValue(10); | |
| 277 notifier.SetNow(schedule_time); | |
| 278 notifier.Schedule(); | |
| 279 EXPECT_TRUE(notifier.HasPendingNotification()); | |
| 280 | |
| 281 // Shutdown the notifier. | |
| 282 notifier.Shutdown(); | |
| 283 | |
| 284 // The task is still there, but... | |
| 285 std::deque<base::TestPendingTask> tasks = TakePendingTasks(); | |
| 286 ASSERT_EQ(1u, tasks.size()); | |
| 287 | |
| 288 // Running the task after shutdown does nothing since it's cancelled. | |
| 289 tasks[0].task.Run(); | |
| 290 EXPECT_EQ(0, NotificationCount()); | |
| 291 | |
| 292 tasks = TakePendingTasks(); | |
| 293 EXPECT_EQ(0u, tasks.size()); | |
| 294 | |
| 295 // We are no longer able to schedule tasks. | |
| 296 notifier.Schedule(); | |
| 297 tasks = TakePendingTasks(); | |
| 298 ASSERT_EQ(0u, tasks.size()); | |
| 299 | |
| 300 // Verify after the scheduled time happens there is still no task. | |
| 301 notifier.SetNow(notifier.Now() + delay); | |
| 302 tasks = TakePendingTasks(); | |
| 303 ASSERT_EQ(0u, tasks.size()); | |
| 304 } | |
| 305 | |
| 306 TEST_F(DelayedUniqueNotifierTest, ShutdownPreventsSchedule) { | |
| 307 base::TimeDelta delay = base::TimeDelta::FromInternalValue(20); | |
| 308 TestNotifier notifier( | |
| 309 task_runner_.get(), | |
| 310 base::Bind(&DelayedUniqueNotifierTest::Notify, base::Unretained(this)), | |
| 311 delay); | |
| 312 | |
| 313 EXPECT_EQ(0, NotificationCount()); | |
| 314 | |
| 315 // Schedule for |delay| seconds from now. | |
| 316 base::TimeTicks schedule_time = | |
| 317 notifier.Now() + base::TimeDelta::FromInternalValue(10); | |
| 318 notifier.SetNow(schedule_time); | |
| 319 | |
| 320 // Shutdown the notifier. | |
| 321 notifier.Shutdown(); | |
| 322 | |
| 323 // Scheduling a task no longer does anything. | |
| 324 notifier.Schedule(); | |
| 325 std::deque<base::TestPendingTask> tasks = TakePendingTasks(); | |
| 326 ASSERT_EQ(0u, tasks.size()); | |
| 327 | |
| 328 // Verify after the scheduled time happens there is still no task. | |
| 329 notifier.SetNow(notifier.Now() + delay); | |
| 330 tasks = TakePendingTasks(); | |
| 331 ASSERT_EQ(0u, tasks.size()); | |
| 332 } | |
| 333 | |
| 334 } // namespace | |
| 335 } // namespace cc | |
| OLD | NEW |