Add a WorkerScheduler and a WebThreadImplForWorker

content/child/scheduler/worker_scheduler_impl_unittest.cc

+// 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 "content/child/scheduler/worker_scheduler_impl.h"
+
+#include "base/callback.h"
+#include "cc/test/ordered_simple_task_runner.h"
+#include "content/child/scheduler/nestable_task_runner_for_test.h"
+#include "content/child/scheduler/scheduler_message_loop_delegate.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace content {
+
+namespace {
+void NopTask() {
+}
+
+void AppendToVectorTestTask(std::vector<std::string>* vector,
+                            std::string value) {
+  vector->push_back(value);
+}
+
+void AppendToVectorIdleTestTask(std::vector<std::string>* vector,
+                                std::string value,
+                                base::TimeTicks deadline) {
+  AppendToVectorTestTask(vector, value);
+}
+
+void IdleTestTask(base::TimeTicks* deadline_out, base::TimeTicks deadline) {
+  *deadline_out = deadline;
+}
+
+void CountingIdleTestTask(int* run_count,
+                          base::TimeTicks* deadline_out,
+                          base::TimeTicks deadline) {
+  (*run_count)++;
+  *deadline_out = deadline;
+}
+
+};  // namespace
+
+class WorkerSchedulerImplTest : public testing::Test {
+ public:
+  WorkerSchedulerImplTest()
+      : clock_(cc::TestNowSource::Create(5000)),
+        mock_task_runner_(new cc::OrderedSimpleTaskRunner(clock_, true)),
+        nestable_task_runner_(
+            NestableTaskRunnerForTest::Create(mock_task_runner_)),
+        scheduler_(new WorkerSchedulerImpl(nestable_task_runner_)) {
+    scheduler_->SetTimeSourceForTesting(clock_);
+  }
+
+  ~WorkerSchedulerImplTest() override {}
+
+  void TearDown() override {
+    // Check that all tests stop posting tasks.
+    while (mock_task_runner_->RunUntilIdle()) {
+    }
+  }
+
+  void Init() {
+    scheduler_->Init();
+    default_task_runner_ = scheduler_->DefaultTaskRunner();
+    idle_task_runner_ = scheduler_->IdleTaskRunner();
+  }
+
+  void RunUntilIdle() { mock_task_runner_->RunUntilIdle(); }
+
+  void InitAndPostDelayedWakeupTask() {
+    Init();
+    // WorkerSchedulerImpl::Init causes a delayed task to be posted on the
+    // after wakeup control runner.  We need a task to wake the system up
+    // AFTER the delay for this has expired.
+    default_task_runner_->PostDelayedTask(
+        FROM_HERE, base::Bind(&NopTask),
+        base::TimeDelta::FromMilliseconds(100));
+  }
+
+  // Helper for posting several tasks of specific types. |task_descriptor| is a
+  // string with space delimited task identifiers. The first letter of each
+  // task identifier specifies the task type:
+  // - 'D': Default task
+  // - 'I': Idle task
+  void PostTestTasks(std::vector<std::string>* run_order,
+                     const std::string& task_descriptor) {
+    std::istringstream stream(task_descriptor);
+    while (!stream.eof()) {
+      std::string task;
+      stream >> task;
+      switch (task[0]) {
+        case 'D':
+          default_task_runner_->PostTask(
+              FROM_HERE, base::Bind(&AppendToVectorTestTask, run_order, task));
+          break;
+        case 'I':
+          idle_task_runner_->PostIdleTask(
+              FROM_HERE,
+              base::Bind(&AppendToVectorIdleTestTask, run_order, task));
+          break;
+        default:
+          NOTREACHED();
+      }
+    }
+  }
+
+  static base::TimeDelta maximum_idle_period_duration() {
+    return base::TimeDelta::FromMilliseconds(
+        SchedulerHelper::kMaximumIdlePeriodMillis);
+  }
+
+ protected:
+  scoped_refptr<cc::TestNowSource> clock_;
+  // Only one of mock_task_runner_ or message_loop_ will be set.
+  scoped_refptr<cc::OrderedSimpleTaskRunner> mock_task_runner_;
+
+  scoped_refptr<NestableSingleThreadTaskRunner> nestable_task_runner_;
+  scoped_ptr<WorkerSchedulerImpl> scheduler_;
+  scoped_refptr<base::SingleThreadTaskRunner> default_task_runner_;
+  scoped_refptr<SingleThreadIdleTaskRunner> idle_task_runner_;
+
+  DISALLOW_COPY_AND_ASSIGN(WorkerSchedulerImplTest);
+};
+
+TEST_F(WorkerSchedulerImplTest, TestPostDefaultTask) {
+  InitAndPostDelayedWakeupTask();
+
+  std::vector<std::string> run_order;
+  PostTestTasks(&run_order, "D1 D2 D3 D4");
+
+  RunUntilIdle();
+  EXPECT_THAT(run_order,
+              testing::ElementsAre(std::string("D1"), std::string("D2"),
+                                   std::string("D3"), std::string("D4")));
+}
+
+TEST_F(WorkerSchedulerImplTest, TestPostIdleTask) {
+  InitAndPostDelayedWakeupTask();
+
+  std::vector<std::string> run_order;
+  PostTestTasks(&run_order, "I1");
+
+  RunUntilIdle();
+  EXPECT_THAT(run_order, testing::ElementsAre(std::string("I1")));
+}
+
+TEST_F(WorkerSchedulerImplTest, TestPostIdleTask_NoWakeup) {
+  Init();
+  std::vector<std::string> run_order;
+  PostTestTasks(&run_order, "I1");
+
+  RunUntilIdle();
+  EXPECT_TRUE(run_order.empty());
+}
+
+TEST_F(WorkerSchedulerImplTest, TestPostDefaultAndIdleTasks) {
+  InitAndPostDelayedWakeupTask();
+
+  std::vector<std::string> run_order;
+  PostTestTasks(&run_order, "I1 D2 D3 D4");
+
+  RunUntilIdle();
+  EXPECT_THAT(run_order,
+              testing::ElementsAre(std::string("D2"), std::string("D3"),
+                                   std::string("D4"), std::string("I1")));
+}
+
+TEST_F(WorkerSchedulerImplTest, TestPostIdleTaskWithWakeupNeeded_NoWakeup) {
+  InitAndPostDelayedWakeupTask();
+
+  RunUntilIdle();
+  // The delayed call to InitiateLongIdlePeriod happened and it posted a call to
+  // InitiateLongIdlePeriod on the after wakeup control queue.
+
+  std::vector<std::string> run_order;
+  PostTestTasks(&run_order, "I1");
+
+  RunUntilIdle();
+  EXPECT_TRUE(run_order.empty());
+}
+
+TEST_F(WorkerSchedulerImplTest, TestPostDefaultDelayedAndIdleTasks) {
+  InitAndPostDelayedWakeupTask();
+
+  std::vector<std::string> run_order;
+  PostTestTasks(&run_order, "I1 D2 D3 D4");
+
+  default_task_runner_->PostDelayedTask(
+      FROM_HERE, base::Bind(&AppendToVectorTestTask, &run_order, "DELAYED"),
+      base::TimeDelta::FromMilliseconds(1000));
+
+  RunUntilIdle();
+  EXPECT_THAT(run_order,
+              testing::ElementsAre(std::string("D2"), std::string("D3"),
+                                   std::string("D4"), std::string("I1"),
+                                   std::string("DELAYED")));
+}
+
+TEST_F(WorkerSchedulerImplTest, TestIdleDeadlineWithPendingDelayedTask) {
+  InitAndPostDelayedWakeupTask();
+
+  base::TimeDelta delay = base::TimeDelta::FromMilliseconds(140);
+  default_task_runner_->PostDelayedTask(FROM_HERE, base::Bind(&NopTask), delay);
+  base::TimeTicks deadline;
+  idle_task_runner_->PostIdleTask(FROM_HERE,
+                                  base::Bind(&IdleTestTask, &deadline));
+
+  base::TimeTicks expected_deadline = clock_->Now() + delay;
+  clock_->AdvanceNow(base::TimeDelta::FromMilliseconds(100));
+
+  RunUntilIdle();
+  EXPECT_EQ(expected_deadline, deadline);
+}
+
+TEST_F(WorkerSchedulerImplTest,
+       TestIdleDeadlineWithPendingDelayedTaskFarInTheFuture) {
+  InitAndPostDelayedWakeupTask();
+
+  base::TimeDelta delay = base::TimeDelta::FromMilliseconds(1000);
+  base::TimeTicks task_run_time = clock_->Now() + delay;
+  default_task_runner_->PostDelayedTask(FROM_HERE, base::Bind(&NopTask), delay);
+  base::TimeTicks deadline;
+  idle_task_runner_->PostIdleTask(FROM_HERE,
+                                  base::Bind(&IdleTestTask, &deadline));
+
+  clock_->AdvanceNow(base::TimeDelta::FromMilliseconds(100));
+
+  RunUntilIdle();
+  EXPECT_LT(deadline, task_run_time);
+}
+
+TEST_F(WorkerSchedulerImplTest,
+       TestPostIdleTaskAfterRunningUntilIdle_NoWakeUp) {
+  InitAndPostDelayedWakeupTask();
+
+  default_task_runner_->PostDelayedTask(
+      FROM_HERE, base::Bind(&NopTask), base::TimeDelta::FromMilliseconds(1000));
+  RunUntilIdle();
+
+  // The delayed call to InitiateLongIdlePeriod happened and it posted a call to
+  // InitiateLongIdlePeriod on the after wakeup control queue. Without an other
+  // non-idle task posted, the idle tasks won't run.
+  std::vector<std::string> run_order;
+  PostTestTasks(&run_order, "I1 I2");
+
+  RunUntilIdle();
+  EXPECT_TRUE(run_order.empty());
+}
+
+TEST_F(WorkerSchedulerImplTest,
+       TestPostIdleTaskAfterRunningUntilIdle_WithWakeUp) {
+  InitAndPostDelayedWakeupTask();
+
+  default_task_runner_->PostDelayedTask(
+      FROM_HERE, base::Bind(&NopTask), base::TimeDelta::FromMilliseconds(1000));
+  RunUntilIdle();
+  // The delayed call to InitiateLongIdlePeriod happened and it posted a call to
+  // InitiateLongIdlePeriod on the after wakeup control queue. Without an other
+  // non-idle task posted, the idle tasks won't run.
+
+  std::vector<std::string> run_order;
+  PostTestTasks(&run_order, "I1 I2 D3");
+
+  RunUntilIdle();
+  EXPECT_THAT(run_order,
+              testing::ElementsAre(std::string("D3"), std::string("I1"),
+                                   std::string("I2")));
+}
+
+TEST_F(WorkerSchedulerImplTest, TestLongIdlePeriodDoesNotWakeScheduler) {
+  mock_task_runner_->SetAutoAdvanceNowToPendingTasks(false);
+  Init();
+
+  base::TimeTicks deadline_in_task;
+  int run_count = 0;
+
+  // The scheduler should not run the initiate_next_long_idle_period task if
+  // there are no idle tasks and no other task woke up the scheduler, thus
+  // the idle period deadline shouldn't update at the end of the current long
+  // idle period.
+  base::TimeTicks idle_period_deadline =
+      scheduler_->CurrentIdleTaskDeadlineForTesting();
+  clock_->AdvanceNow(maximum_idle_period_duration());
+  RunUntilIdle();
+
+  base::TimeTicks new_idle_period_deadline =
+      scheduler_->CurrentIdleTaskDeadlineForTesting();
+  EXPECT_EQ(idle_period_deadline, new_idle_period_deadline);
+
+  // Posting a after-wakeup idle task also shouldn't wake the scheduler or
+  // initiate the next long idle period.
+  idle_task_runner_->PostIdleTaskAfterWakeup(
+      FROM_HERE,
+      base::Bind(&CountingIdleTestTask, &run_count, &deadline_in_task));
+  RunUntilIdle();
+  new_idle_period_deadline = scheduler_->CurrentIdleTaskDeadlineForTesting();
+  EXPECT_EQ(idle_period_deadline, new_idle_period_deadline);
+  EXPECT_EQ(0, run_count);
+
+  // Running a normal task should initiate a new long idle period though.
+  default_task_runner_->PostTask(FROM_HERE, base::Bind(&NopTask));
+  RunUntilIdle();
+  new_idle_period_deadline = scheduler_->CurrentIdleTaskDeadlineForTesting();
+  EXPECT_EQ(idle_period_deadline + maximum_idle_period_duration(),
+            new_idle_period_deadline);
+
+  EXPECT_EQ(1, run_count);
+  mock_task_runner_->SetAutoAdvanceNowToPendingTasks(true);
+}
+
+}  // namespace content