Replaced TestNowSource with SimpleTestTickClock.

components/scheduler/child/task_queue_manager_unittest.cc

 // Copyright 2014 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/child/task_queue_manager.h"
 
+#include "base/test/simple_test_tick_clock.h"
 #include "base/threading/thread.h"
 #include "cc/test/ordered_simple_task_runner.h"
-#include "cc/test/test_now_source.h"
 #include "components/scheduler/child/nestable_task_runner_for_test.h"
 #include "components/scheduler/child/scheduler_message_loop_delegate.h"
 #include "components/scheduler/child/task_queue_selector.h"
+#include "components/scheduler/child/test_always_fail_time_source.h"
 #include "components/scheduler/child/test_time_source.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 using testing::ElementsAre;
 using testing::_;
 
 namespace scheduler {
 namespace {
 
 class SelectorForTest : public TaskQueueSelector {
@@ skipping 85 matching lines @@
  public:
   void DeleteTaskQueueManager() { manager_.reset(); }
 
  protected:
   enum class SelectorType {
     Automatic,
     Explicit,
   };
 
   void Initialize(size_t num_queues, SelectorType type) {
-    now_src_ = cc::TestNowSource::Create(1000);
-    test_task_runner_ =
-        make_scoped_refptr(new cc::OrderedSimpleTaskRunner(now_src_, false));
+    now_src_ = make_scoped_ptr(new base::SimpleTestTickClock());
+    now_src_->Advance(base::TimeDelta::FromInternalValue(1000));
+    test_task_runner_ = make_scoped_refptr(
+        new cc::OrderedSimpleTaskRunner(now_src_.get(), false));
     selector_ = make_scoped_ptr(createSelectorForTest(type));
     manager_ = make_scoped_ptr(new TaskQueueManager(
         num_queues, NestableTaskRunnerForTest::Create(test_task_runner_.get()),
         selector_.get(), "test.scheduler"));
     manager_->SetTimeSourceForTesting(
-        make_scoped_ptr(new TestTimeSource(now_src_)));
+        make_scoped_ptr(new TestTimeSource(now_src_.get())));
 
     EXPECT_EQ(num_queues, selector_->work_queues().size());
   }
 
   void InitializeWithRealMessageLoop(size_t num_queues, SelectorType type) {
     message_loop_.reset(new base::MessageLoop());
     selector_ = make_scoped_ptr(createSelectorForTest(type));
     manager_ = make_scoped_ptr(new TaskQueueManager(
         num_queues, SchedulerMessageLoopDelegate::Create(message_loop_.get()),
         selector_.get(), "test.scheduler"));
@@ skipping 29 matching lines @@
         &TaskQueueManager::PumpPolicyToString);
   }
 
   static void CheckAllWakeupPolicyToString() {
     CallForEachEnumValue<TaskQueueManager::WakeupPolicy>(
         TaskQueueManager::WakeupPolicy::FIRST_WAKEUP_POLICY,
         TaskQueueManager::WakeupPolicy::WAKEUP_POLICY_COUNT,
         &TaskQueueManager::WakeupPolicyToString);
   }
 
-  scoped_refptr<cc::TestNowSource> now_src_;
+  scoped_ptr<base::SimpleTestTickClock> now_src_;
   scoped_refptr<cc::OrderedSimpleTaskRunner> test_task_runner_;
   scoped_ptr<SelectorForTest> selector_;
   scoped_ptr<TaskQueueManager> manager_;
   scoped_ptr<base::MessageLoop> message_loop_;
 };
 
 void PostFromNestedRunloop(base::MessageLoop* message_loop,
                            base::SingleThreadTaskRunner* runner,
                            std::vector<std::pair<base::Closure, bool>>* tasks) {
   base::MessageLoop::ScopedNestableTaskAllower allow(message_loop);
@@ skipping 55 matching lines @@
   test_task_runner_->RunUntilIdle();
   EXPECT_THAT(run_order, ElementsAre(1, 3, 5, 2, 4, 6));
 }
 
 void NopTask() {
 }
 
 TEST_F(TaskQueueManagerTest, NowNotCalledWhenThereAreNoDelayedTasks) {
   Initialize(3u, SelectorType::Explicit);
 
-  scoped_refptr<cc::TestNowSource> now_src = cc::TestNowSource::Create(1000);
+  scoped_ptr<base::SimpleTestTickClock> now_src(
+      new base::SimpleTestTickClock());
+  now_src->Advance(base::TimeDelta::FromInternalValue(1000));
   manager_->SetTimeSourceForTesting(
-      make_scoped_ptr(new TestTimeSource(now_src)));
+      make_scoped_ptr(new TestAlwaysFailTimeSource(now_src.get())));
 
   scoped_refptr<base::SingleThreadTaskRunner> runners[3] = {
       manager_->TaskRunnerForQueue(0),
       manager_->TaskRunnerForQueue(1),
       manager_->TaskRunnerForQueue(2)};
 
   selector_->AppendQueueToService(0);
   selector_->AppendQueueToService(1);
   selector_->AppendQueueToService(2);
   selector_->AppendQueueToService(0);
   selector_->AppendQueueToService(1);
   selector_->AppendQueueToService(2);
 
   runners[0]->PostTask(FROM_HERE, base::Bind(&NopTask));
   runners[0]->PostTask(FROM_HERE, base::Bind(&NopTask));
   runners[1]->PostTask(FROM_HERE, base::Bind(&NopTask));
   runners[1]->PostTask(FROM_HERE, base::Bind(&NopTask));
   runners[2]->PostTask(FROM_HERE, base::Bind(&NopTask));
   runners[2]->PostTask(FROM_HERE, base::Bind(&NopTask));
 
   test_task_runner_->RunUntilIdle();
-
-  EXPECT_EQ(0, now_src->NumNowCalls());
 }
 
 TEST_F(TaskQueueManagerTest, NonNestableTaskPosting) {
   InitializeWithRealMessageLoop(1u, SelectorType::Automatic);
 
   std::vector<int> run_order;
   scoped_refptr<base::SingleThreadTaskRunner> runner =
       manager_->TaskRunnerForQueue(0);
 
   runner->PostNonNestableTask(FROM_HERE, base::Bind(&TestTask, 1, &run_order));
@@ skipping 275 matching lines @@
   base::TimeDelta delay(base::TimeDelta::FromMilliseconds(10));
   runner->PostDelayedTask(FROM_HERE, base::Bind(&TestTask, 1, &run_order),
                           delay);
 
   // After pumping but before the delay period has expired, task does not run.
   manager_->PumpQueue(0);
   test_task_runner_->RunForPeriod(base::TimeDelta::FromMilliseconds(5));
   EXPECT_TRUE(run_order.empty());
 
   // Once the delay has expired, pumping causes the task to run.
-  now_src_->AdvanceNow(base::TimeDelta::FromMilliseconds(5));
+  now_src_->Advance(base::TimeDelta::FromMilliseconds(5));
   manager_->PumpQueue(0);
   EXPECT_TRUE(test_task_runner_->HasPendingTasks());
   test_task_runner_->RunPendingTasks();
   EXPECT_THAT(run_order, ElementsAre(1));
 }
 
 TEST_F(TaskQueueManagerTest, ManualPumpingWithMultipleDelayedTasks) {
   Initialize(1u, SelectorType::Automatic);
   manager_->SetPumpPolicy(0, TaskQueueManager::PumpPolicy::MANUAL);
 
   std::vector<int> run_order;
   scoped_refptr<base::SingleThreadTaskRunner> runner =
       manager_->TaskRunnerForQueue(0);
 
   // Posting a delayed task when pumping will apply the delay, but won't cause
   // work to executed afterwards.
   base::TimeDelta delay1(base::TimeDelta::FromMilliseconds(1));
   base::TimeDelta delay2(base::TimeDelta::FromMilliseconds(10));
   base::TimeDelta delay3(base::TimeDelta::FromMilliseconds(20));
   runner->PostDelayedTask(FROM_HERE, base::Bind(&TestTask, 1, &run_order),
                           delay1);
   runner->PostDelayedTask(FROM_HERE, base::Bind(&TestTask, 2, &run_order),
                           delay2);
   runner->PostDelayedTask(FROM_HERE, base::Bind(&TestTask, 3, &run_order),
                           delay3);
 
-  now_src_->AdvanceNow(base::TimeDelta::FromMilliseconds(15));
+  now_src_->Advance(base::TimeDelta::FromMilliseconds(15));
   test_task_runner_->RunUntilIdle();
   EXPECT_TRUE(run_order.empty());
 
   // Once the delay has expired, pumping causes the task to run.
   manager_->PumpQueue(0);
   test_task_runner_->RunUntilIdle();
   EXPECT_THAT(run_order, ElementsAre(1, 2));
 }
 
 TEST_F(TaskQueueManagerTest, DelayedTasksDontAutoRunWithManualPumping) {
@@ skipping 424 matching lines @@
 TEST_F(TaskQueueManagerTest, ThreadCheckAfterTermination) {
   Initialize(1u, SelectorType::Automatic);
   scoped_refptr<base::SingleThreadTaskRunner> runner =
       manager_->TaskRunnerForQueue(0);
   EXPECT_TRUE(runner->RunsTasksOnCurrentThread());
   manager_.reset();
   EXPECT_TRUE(runner->RunsTasksOnCurrentThread());
 }
 
 TEST_F(TaskQueueManagerTest, NextPendingDelayedTaskRunTime) {
-  scoped_refptr<cc::TestNowSource> clock(cc::TestNowSource::Create());
+  scoped_ptr<base::SimpleTestTickClock> clock(new base::SimpleTestTickClock());
+  clock->Advance(base::TimeDelta::FromInternalValue(10000));
   Initialize(2u, SelectorType::Explicit);
-  manager_->SetTimeSourceForTesting(make_scoped_ptr(new TestTimeSource(clock)));
+  manager_->SetTimeSourceForTesting(
+      make_scoped_ptr(new TestTimeSource(clock.get())));
 
   scoped_refptr<base::SingleThreadTaskRunner> runners[2] = {
       manager_->TaskRunnerForQueue(0), manager_->TaskRunnerForQueue(1)};
 
   // With no delayed tasks.
   EXPECT_TRUE(manager_->NextPendingDelayedTaskRunTime().is_null());
 
   // With a non-delayed task.
   runners[0]->PostTask(FROM_HERE, base::Bind(&NopTask));
   EXPECT_TRUE(manager_->NextPendingDelayedTaskRunTime().is_null());
 
   // With a delayed task.
   base::TimeDelta expected_delay = base::TimeDelta::FromMilliseconds(50);
   runners[0]->PostDelayedTask(FROM_HERE, base::Bind(&NopTask), expected_delay);
-  EXPECT_EQ(clock->Now() + expected_delay,
+  EXPECT_EQ(clock->NowTicks() + expected_delay,
             manager_->NextPendingDelayedTaskRunTime());
 
   // With another delayed task in the same queue with a longer delay.
   runners[0]->PostDelayedTask(FROM_HERE, base::Bind(&NopTask),
                               base::TimeDelta::FromMilliseconds(100));
-  EXPECT_EQ(clock->Now() + expected_delay,
+  EXPECT_EQ(clock->NowTicks() + expected_delay,
             manager_->NextPendingDelayedTaskRunTime());
 
   // With another delayed task in the same queue with a shorter delay.
   expected_delay = base::TimeDelta::FromMilliseconds(20);
   runners[0]->PostDelayedTask(FROM_HERE, base::Bind(&NopTask), expected_delay);
-  EXPECT_EQ(clock->Now() + expected_delay,
+  EXPECT_EQ(clock->NowTicks() + expected_delay,
             manager_->NextPendingDelayedTaskRunTime());
 
   // With another delayed task in a different queue with a shorter delay.
   expected_delay = base::TimeDelta::FromMilliseconds(10);
   runners[1]->PostDelayedTask(FROM_HERE, base::Bind(&NopTask), expected_delay);
-  EXPECT_EQ(clock->Now() + expected_delay,
+  EXPECT_EQ(clock->NowTicks() + expected_delay,
             manager_->NextPendingDelayedTaskRunTime());
 
   // Test it updates as time progresses
-  clock->AdvanceNow(expected_delay);
-  EXPECT_EQ(clock->Now(), manager_->NextPendingDelayedTaskRunTime());
+  clock->Advance(expected_delay);
+  EXPECT_EQ(clock->NowTicks(), manager_->NextPendingDelayedTaskRunTime());
 }
 
 TEST_F(TaskQueueManagerTest, NextPendingDelayedTaskRunTime_MultipleQueues) {
   Initialize(3u, SelectorType::Automatic);
 
   scoped_refptr<base::SingleThreadTaskRunner> runners[3] = {
       manager_->TaskRunnerForQueue(0),
       manager_->TaskRunnerForQueue(1),
       manager_->TaskRunnerForQueue(2)};
 
   base::TimeDelta delay1 = base::TimeDelta::FromMilliseconds(50);
   base::TimeDelta delay2 = base::TimeDelta::FromMilliseconds(5);
   base::TimeDelta delay3 = base::TimeDelta::FromMilliseconds(10);
   runners[0]->PostDelayedTask(FROM_HERE, base::Bind(&NopTask), delay1);
   runners[1]->PostDelayedTask(FROM_HERE, base::Bind(&NopTask), delay2);
   runners[2]->PostDelayedTask(FROM_HERE, base::Bind(&NopTask), delay3);
 
-  EXPECT_EQ(now_src_->Now() + delay2,
+  EXPECT_EQ(now_src_->NowTicks() + delay2,
             manager_->NextPendingDelayedTaskRunTime());
 }
 
 TEST_F(TaskQueueManagerTest, DeleteTaskQueueManagerInsideATask) {
   Initialize(1u, SelectorType::Automatic);
 
   scoped_refptr<base::SingleThreadTaskRunner> runner =
       manager_->TaskRunnerForQueue(0);
   runner->PostTask(FROM_HERE,
                    base::Bind(&TaskQueueManagerTest::DeleteTaskQueueManager,
@@ skipping 119 matching lines @@
             manager_->GetQueueState(1));
 
   test_task_runner_->RunUntilIdle();
   EXPECT_EQ(TaskQueueManager::QueueState::EMPTY,
             manager_->GetQueueState(0));
   EXPECT_EQ(TaskQueueManager::QueueState::EMPTY,
             manager_->GetQueueState(1));
 }
 
 }  // namespace scheduler