OLD | NEW |
(Empty) | |
| 1 // Copyright 2015 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 "components/scheduler/base/time_domain.h" |
| 6 |
| 7 #include "base/test/simple_test_tick_clock.h" |
| 8 #include "cc/test/ordered_simple_task_runner.h" |
| 9 #include "components/scheduler/base/task_queue_impl.h" |
| 10 #include "components/scheduler/base/task_queue_manager.h" |
| 11 #include "components/scheduler/base/task_queue_manager_delegate_for_test.h" |
| 12 #include "components/scheduler/base/test_time_source.h" |
| 13 #include "testing/gmock/include/gmock/gmock.h" |
| 14 |
| 15 using testing::_; |
| 16 |
| 17 namespace scheduler { |
| 18 |
| 19 class MockTimeDomain : public TimeDomain { |
| 20 public: |
| 21 MockTimeDomain() |
| 22 : now_(base::TimeTicks() + base::TimeDelta::FromSeconds(1)) {} |
| 23 |
| 24 using TimeDomain::NextScheduledRunTime; |
| 25 using TimeDomain::NextScheduledTaskQueue; |
| 26 using TimeDomain::ScheduleDelayedWork; |
| 27 using TimeDomain::UnregisterQueue; |
| 28 using TimeDomain::UpdateWorkQueues; |
| 29 |
| 30 // TimeSource implementation: |
| 31 LazyNow CreateLazyNow() override { return LazyNow(now_); } |
| 32 |
| 33 void AsValueIntoInternal( |
| 34 base::trace_event::TracedValue* state) const override {} |
| 35 |
| 36 bool MaybeAdvanceTime() override { return false; } |
| 37 |
| 38 const char* GetName() const override { return "Test"; } |
| 39 |
| 40 MOCK_METHOD1(RequestWakeup, void(base::TimeDelta delay)); |
| 41 |
| 42 void SetNow(base::TimeTicks now) { now_ = now; } |
| 43 |
| 44 base::TimeTicks Now() const { return now_; } |
| 45 |
| 46 private: |
| 47 base::TimeTicks now_; |
| 48 |
| 49 ~MockTimeDomain() override {} |
| 50 |
| 51 DISALLOW_COPY_AND_ASSIGN(MockTimeDomain); |
| 52 }; |
| 53 |
| 54 class TimeDomainTest : public testing::Test { |
| 55 public: |
| 56 void SetUp() final { |
| 57 time_domain_ = make_scoped_refptr(new MockTimeDomain()); |
| 58 task_queue_ = make_scoped_refptr(new internal::TaskQueueImpl( |
| 59 nullptr, time_domain_, TaskQueue::Spec("test_queue"), "test.category", |
| 60 "test.category")); |
| 61 } |
| 62 |
| 63 scoped_refptr<MockTimeDomain> time_domain_; |
| 64 scoped_refptr<internal::TaskQueueImpl> task_queue_; |
| 65 }; |
| 66 |
| 67 TEST_F(TimeDomainTest, ScheduleDelayedWork) { |
| 68 base::TimeDelta delay = base::TimeDelta::FromMilliseconds(10); |
| 69 base::TimeTicks delayed_runtime = time_domain_->Now() + delay; |
| 70 EXPECT_CALL(*time_domain_.get(), RequestWakeup(delay)); |
| 71 LazyNow lazy_now = time_domain_->CreateLazyNow(); |
| 72 time_domain_->ScheduleDelayedWork(task_queue_.get(), |
| 73 time_domain_->Now() + delay, &lazy_now); |
| 74 |
| 75 base::TimeTicks next_scheduled_runtime; |
| 76 EXPECT_TRUE(time_domain_->NextScheduledRunTime(&next_scheduled_runtime)); |
| 77 EXPECT_EQ(delayed_runtime, next_scheduled_runtime); |
| 78 |
| 79 TaskQueue* next_task_queue; |
| 80 EXPECT_TRUE(time_domain_->NextScheduledTaskQueue(&next_task_queue)); |
| 81 EXPECT_EQ(task_queue_.get(), next_task_queue); |
| 82 } |
| 83 |
| 84 TEST_F(TimeDomainTest, UnregisterQueue) { |
| 85 scoped_refptr<internal::TaskQueueImpl> task_queue2_ = |
| 86 make_scoped_refptr(new internal::TaskQueueImpl( |
| 87 nullptr, time_domain_, TaskQueue::Spec("test_queue2"), |
| 88 "test.category", "test.category")); |
| 89 |
| 90 EXPECT_CALL(*time_domain_.get(), RequestWakeup(_)).Times(2); |
| 91 LazyNow lazy_now = time_domain_->CreateLazyNow(); |
| 92 time_domain_->ScheduleDelayedWork( |
| 93 task_queue_.get(), |
| 94 time_domain_->Now() + base::TimeDelta::FromMilliseconds(10), &lazy_now); |
| 95 time_domain_->ScheduleDelayedWork( |
| 96 task_queue2_.get(), |
| 97 time_domain_->Now() + base::TimeDelta::FromMilliseconds(100), &lazy_now); |
| 98 |
| 99 TaskQueue* next_task_queue; |
| 100 EXPECT_TRUE(time_domain_->NextScheduledTaskQueue(&next_task_queue)); |
| 101 EXPECT_EQ(task_queue_.get(), next_task_queue); |
| 102 |
| 103 time_domain_->UnregisterQueue(task_queue_.get()); |
| 104 EXPECT_TRUE(time_domain_->NextScheduledTaskQueue(&next_task_queue)); |
| 105 EXPECT_EQ(task_queue2_.get(), next_task_queue); |
| 106 |
| 107 time_domain_->UnregisterQueue(task_queue2_.get()); |
| 108 EXPECT_FALSE(time_domain_->NextScheduledTaskQueue(&next_task_queue)); |
| 109 } |
| 110 |
| 111 TEST_F(TimeDomainTest, UpdateWorkQueues) { |
| 112 scoped_refptr<MockTimeDomain> dummy_delegate(new MockTimeDomain()); |
| 113 base::SimpleTestTickClock dummy_time_source; |
| 114 scoped_refptr<cc::OrderedSimpleTaskRunner> dummy_task_runner( |
| 115 new cc::OrderedSimpleTaskRunner(&dummy_time_source, false)); |
| 116 TaskQueueManager task_queue_manager( |
| 117 TaskQueueManagerDelegateForTest::Create( |
| 118 dummy_task_runner, |
| 119 make_scoped_ptr(new TestTimeSource(&dummy_time_source))), |
| 120 "test.scheduler", "test.scheduler", "scheduler.debug"); |
| 121 scoped_refptr<internal::TaskQueueImpl> dummy_queue = |
| 122 task_queue_manager.NewTaskQueue(TaskQueue::Spec("test_queue")); |
| 123 |
| 124 // Post a delayed task on |dummy_queue| and advance the queue's clock so that |
| 125 // next time MoveReadyDelayedTasksToIncomingQueue is called, the task will |
| 126 // get moved onto the incomming queue. |
| 127 base::TimeDelta dummy_delay = base::TimeDelta::FromMilliseconds(10); |
| 128 dummy_queue->PostDelayedTask(FROM_HERE, base::Closure(), dummy_delay); |
| 129 dummy_time_source.Advance(dummy_delay); |
| 130 |
| 131 // Now we can test that ScheduleDelayedWork triggers calls to |
| 132 // MoveReadyDelayedTasksToIncomingQueue as expected. |
| 133 base::TimeDelta delay = base::TimeDelta::FromMilliseconds(50); |
| 134 base::TimeTicks delayed_runtime = time_domain_->Now() + delay; |
| 135 EXPECT_CALL(*time_domain_.get(), RequestWakeup(delay)); |
| 136 LazyNow lazy_now = time_domain_->CreateLazyNow(); |
| 137 time_domain_->ScheduleDelayedWork(dummy_queue.get(), delayed_runtime, |
| 138 &lazy_now); |
| 139 |
| 140 time_domain_->UpdateWorkQueues(false, nullptr); |
| 141 EXPECT_EQ(0UL, dummy_queue->IncomingQueueSizeForTest()); |
| 142 |
| 143 time_domain_->SetNow(delayed_runtime); |
| 144 time_domain_->UpdateWorkQueues(false, nullptr); |
| 145 EXPECT_EQ(1UL, dummy_queue->IncomingQueueSizeForTest()); |
| 146 } |
| 147 |
| 148 } // namespace scheduler |
OLD | NEW |