| // Copyright 2015 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "components/scheduler/base/time_domain.h" |
| |
| #include "base/macros.h" |
| #include "base/memory/ptr_util.h" |
| #include "base/test/simple_test_tick_clock.h" |
| #include "cc/test/ordered_simple_task_runner.h" |
| #include "components/scheduler/base/task_queue_impl.h" |
| #include "components/scheduler/base/task_queue_manager.h" |
| #include "components/scheduler/base/task_queue_manager_delegate_for_test.h" |
| #include "components/scheduler/base/test_time_source.h" |
| #include "components/scheduler/base/work_queue.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| |
| using testing::_; |
| using testing::AnyNumber; |
| using testing::Mock; |
| |
| namespace scheduler { |
| |
| class MockTimeDomain : public TimeDomain { |
| public: |
| explicit MockTimeDomain(TimeDomain::Observer* observer) |
| : TimeDomain(observer), |
| now_(base::TimeTicks() + base::TimeDelta::FromSeconds(1)) {} |
| |
| ~MockTimeDomain() override {} |
| |
| using TimeDomain::ClearExpiredWakeups; |
| using TimeDomain::NextScheduledRunTime; |
| using TimeDomain::NextScheduledTaskQueue; |
| using TimeDomain::ScheduleDelayedWork; |
| using TimeDomain::UnregisterQueue; |
| using TimeDomain::UpdateWorkQueues; |
| using TimeDomain::RegisterAsUpdatableTaskQueue; |
| |
| // TimeSource implementation: |
| LazyNow CreateLazyNow() const override { return LazyNow(now_); } |
| base::TimeTicks Now() const override { return now_; } |
| |
| base::TimeTicks ComputeDelayedRunTime(base::TimeTicks time_domain_now, |
| base::TimeDelta delay) const override { |
| return time_domain_now + delay; |
| } |
| |
| void AsValueIntoInternal( |
| base::trace_event::TracedValue* state) const override {} |
| |
| bool MaybeAdvanceTime() override { return false; } |
| const char* GetName() const override { return "Test"; } |
| void OnRegisterWithTaskQueueManager( |
| TaskQueueManager* task_queue_manager) override {} |
| |
| MOCK_METHOD2(RequestWakeup, void(base::TimeTicks now, base::TimeDelta delay)); |
| |
| void SetNow(base::TimeTicks now) { now_ = now; } |
| |
| |
| private: |
| base::TimeTicks now_; |
| |
| DISALLOW_COPY_AND_ASSIGN(MockTimeDomain); |
| }; |
| |
| class TimeDomainTest : public testing::Test { |
| public: |
| void SetUp() final { |
| time_domain_ = base::WrapUnique(CreateMockTimeDomain()); |
| task_queue_ = make_scoped_refptr(new internal::TaskQueueImpl( |
| nullptr, time_domain_.get(), TaskQueue::Spec("test_queue"), |
| "test.category", "test.category")); |
| } |
| |
| void TearDown() final { |
| if (task_queue_) |
| task_queue_->UnregisterTaskQueue(); |
| } |
| |
| virtual MockTimeDomain* CreateMockTimeDomain() { |
| return new MockTimeDomain(nullptr); |
| } |
| |
| std::unique_ptr<MockTimeDomain> time_domain_; |
| scoped_refptr<internal::TaskQueueImpl> task_queue_; |
| }; |
| |
| TEST_F(TimeDomainTest, ScheduleDelayedWork) { |
| base::TimeDelta delay = base::TimeDelta::FromMilliseconds(10); |
| base::TimeTicks delayed_runtime = time_domain_->Now() + delay; |
| EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, delay)); |
| base::TimeTicks now = time_domain_->Now(); |
| time_domain_->ScheduleDelayedWork(task_queue_.get(), now + delay, now); |
| |
| base::TimeTicks next_scheduled_runtime; |
| EXPECT_TRUE(time_domain_->NextScheduledRunTime(&next_scheduled_runtime)); |
| EXPECT_EQ(delayed_runtime, next_scheduled_runtime); |
| |
| TaskQueue* next_task_queue; |
| EXPECT_TRUE(time_domain_->NextScheduledTaskQueue(&next_task_queue)); |
| EXPECT_EQ(task_queue_.get(), next_task_queue); |
| } |
| |
| TEST_F(TimeDomainTest, RequestWakeup_OnlyCalledForEarlierTasks) { |
| base::TimeDelta delay1 = base::TimeDelta::FromMilliseconds(10); |
| base::TimeDelta delay2 = base::TimeDelta::FromMilliseconds(20); |
| base::TimeDelta delay3 = base::TimeDelta::FromMilliseconds(30); |
| base::TimeDelta delay4 = base::TimeDelta::FromMilliseconds(1); |
| |
| // RequestWakeup should always be called if there are no other wakeups. |
| EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, delay1)); |
| base::TimeTicks now = time_domain_->Now(); |
| time_domain_->ScheduleDelayedWork(task_queue_.get(), now + delay1, now); |
| |
| Mock::VerifyAndClearExpectations(time_domain_.get()); |
| |
| // RequestWakeup should not be called when scheduling later tasks. |
| EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, _)).Times(0); |
| time_domain_->ScheduleDelayedWork(task_queue_.get(), now + delay2, now); |
| time_domain_->ScheduleDelayedWork(task_queue_.get(), now + delay3, now); |
| |
| // RequestWakeup should be called when scheduling earlier tasks. |
| Mock::VerifyAndClearExpectations(time_domain_.get()); |
| EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, delay4)); |
| time_domain_->ScheduleDelayedWork(task_queue_.get(), now + delay4, now); |
| } |
| |
| TEST_F(TimeDomainTest, UnregisterQueue) { |
| scoped_refptr<internal::TaskQueueImpl> task_queue2_ = |
| make_scoped_refptr(new internal::TaskQueueImpl( |
| nullptr, time_domain_.get(), TaskQueue::Spec("test_queue2"), |
| "test.category", "test.category")); |
| |
| EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, _)).Times(1); |
| base::TimeTicks now = time_domain_->Now(); |
| time_domain_->ScheduleDelayedWork( |
| task_queue_.get(), now + base::TimeDelta::FromMilliseconds(10), now); |
| time_domain_->ScheduleDelayedWork( |
| task_queue2_.get(), now + base::TimeDelta::FromMilliseconds(100), now); |
| |
| TaskQueue* next_task_queue; |
| EXPECT_TRUE(time_domain_->NextScheduledTaskQueue(&next_task_queue)); |
| EXPECT_EQ(task_queue_.get(), next_task_queue); |
| |
| time_domain_->UnregisterQueue(task_queue_.get()); |
| task_queue_ = scoped_refptr<internal::TaskQueueImpl>(); |
| EXPECT_TRUE(time_domain_->NextScheduledTaskQueue(&next_task_queue)); |
| EXPECT_EQ(task_queue2_.get(), next_task_queue); |
| |
| time_domain_->UnregisterQueue(task_queue2_.get()); |
| EXPECT_FALSE(time_domain_->NextScheduledTaskQueue(&next_task_queue)); |
| } |
| |
| TEST_F(TimeDomainTest, UpdateWorkQueues) { |
| base::TimeDelta delay = base::TimeDelta::FromMilliseconds(50); |
| EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, delay)); |
| base::TimeTicks now = time_domain_->Now(); |
| base::TimeTicks delayed_runtime = now + delay; |
| time_domain_->ScheduleDelayedWork(task_queue_.get(), delayed_runtime, now); |
| |
| base::TimeTicks next_run_time; |
| ASSERT_TRUE(time_domain_->NextScheduledRunTime(&next_run_time)); |
| EXPECT_EQ(delayed_runtime, next_run_time); |
| |
| time_domain_->UpdateWorkQueues(false, nullptr); |
| ASSERT_TRUE(time_domain_->NextScheduledRunTime(&next_run_time)); |
| EXPECT_EQ(delayed_runtime, next_run_time); |
| |
| time_domain_->SetNow(delayed_runtime); |
| time_domain_->UpdateWorkQueues(false, nullptr); |
| ASSERT_FALSE(time_domain_->NextScheduledRunTime(&next_run_time)); |
| } |
| |
| TEST_F(TimeDomainTest, ClearExpiredWakeups) { |
| base::TimeTicks now = time_domain_->Now(); |
| base::TimeDelta delay1 = base::TimeDelta::FromMilliseconds(10); |
| base::TimeDelta delay2 = base::TimeDelta::FromMilliseconds(20); |
| base::TimeTicks run_time1 = now + delay1; |
| base::TimeTicks run_time2 = now + delay2; |
| |
| EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, _)).Times(AnyNumber()); |
| time_domain_->ScheduleDelayedWork(task_queue_.get(), run_time1, now); |
| time_domain_->ScheduleDelayedWork(task_queue_.get(), run_time2, now); |
| |
| base::TimeTicks next_run_time; |
| ASSERT_TRUE(time_domain_->NextScheduledRunTime(&next_run_time)); |
| EXPECT_EQ(run_time1, next_run_time); |
| |
| time_domain_->SetNow(run_time1); |
| time_domain_->ClearExpiredWakeups(); |
| |
| ASSERT_TRUE(time_domain_->NextScheduledRunTime(&next_run_time)); |
| EXPECT_EQ(run_time2, next_run_time); |
| |
| time_domain_->SetNow(run_time2); |
| time_domain_->ClearExpiredWakeups(); |
| ASSERT_FALSE(time_domain_->NextScheduledRunTime(&next_run_time)); |
| } |
| |
| namespace { |
| class MockObserver : public TimeDomain::Observer { |
| public: |
| ~MockObserver() override {} |
| |
| MOCK_METHOD0(OnTimeDomainHasImmediateWork, void()); |
| MOCK_METHOD0(OnTimeDomainHasDelayedWork, void()); |
| }; |
| } // namespace |
| |
| class TimeDomainWithObserverTest : public TimeDomainTest { |
| public: |
| MockTimeDomain* CreateMockTimeDomain() override { |
| observer_.reset(new MockObserver()); |
| return new MockTimeDomain(observer_.get()); |
| } |
| |
| std::unique_ptr<MockObserver> observer_; |
| }; |
| |
| TEST_F(TimeDomainWithObserverTest, OnTimeDomainHasImmediateWork) { |
| EXPECT_CALL(*observer_, OnTimeDomainHasImmediateWork()); |
| time_domain_->RegisterAsUpdatableTaskQueue(task_queue_.get()); |
| } |
| |
| TEST_F(TimeDomainWithObserverTest, OnTimeDomainHasDelayedWork) { |
| EXPECT_CALL(*observer_, OnTimeDomainHasDelayedWork()); |
| EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, _)); |
| base::TimeTicks now = time_domain_->Now(); |
| time_domain_->ScheduleDelayedWork( |
| task_queue_.get(), now + base::TimeDelta::FromMilliseconds(10), now); |
| } |
| |
| } // namespace scheduler |