Introduce a SchedulerHelper in content/child/scheduler

content/child/scheduler/task_queue_selector_impl_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 "content/renderer/scheduler/renderer_task_queue_selector.h"
+#include "content/child/scheduler/task_queue_selector_impl.h"
 
 #include "base/bind.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/scoped_vector.h"
 #include "base/pending_task.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace content {
 
-class RendererTaskQueueSelectorTest : public testing::Test {
+class TaskQueueSelectorImplTest : public testing::Test {
  public:
-  RendererTaskQueueSelectorTest()
-      : test_closure_(
-            base::Bind(&RendererTaskQueueSelectorTest::TestFunction)) {}
-  ~RendererTaskQueueSelectorTest() override {}
+  TaskQueueSelectorImplTest()
+      : test_closure_(base::Bind(&TaskQueueSelectorImplTest::TestFunction)) {}
+  ~TaskQueueSelectorImplTest() override {}
 
   std::vector<base::PendingTask> GetTasks(int count) {
     std::vector<base::PendingTask> tasks;
     for (int i = 0; i < count; i++) {
       base::PendingTask task = base::PendingTask(FROM_HERE, test_closure_);
       task.sequence_num = i;
       tasks.push_back(task);
     }
     return tasks;
   }
@@ skipping 25 matching lines @@
       const_task_queues.push_back(task_queue.get());
       task_queues_.push_back(task_queue.release());
     }
     selector_.RegisterWorkQueues(const_task_queues);
     for (size_t i = 0; i < kTaskQueueCount; i++)
       EXPECT_TRUE(selector_.IsQueueEnabled(i)) << i;
   }
 
   const size_t kTaskQueueCount = 5;
   base::Closure test_closure_;
-  RendererTaskQueueSelector selector_;
+  TaskQueueSelectorImpl selector_;
   ScopedVector<base::TaskQueue> task_queues_;
 };
 
-TEST_F(RendererTaskQueueSelectorTest, TestDefaultPriority) {
+TEST_F(TaskQueueSelectorImplTest, TestDefaultPriority) {
   std::vector<base::PendingTask> tasks = GetTasks(5);
   size_t queue_order[] = {4, 3, 2, 1, 0};
   PushTasks(tasks, queue_order);
   EXPECT_THAT(PopTasks(), testing::ElementsAre(4, 3, 2, 1, 0));
 }
 
-TEST_F(RendererTaskQueueSelectorTest, TestHighPriority) {
+TEST_F(TaskQueueSelectorImplTest, TestHighPriority) {
   std::vector<base::PendingTask> tasks = GetTasks(5);
   size_t queue_order[] = {0, 1, 2, 3, 4};
   PushTasks(tasks, queue_order);
-  selector_.SetQueuePriority(2, RendererTaskQueueSelector::HIGH_PRIORITY);
+  selector_.SetQueuePriority(2, TaskQueueSelectorImpl::HIGH_PRIORITY);
   EXPECT_THAT(PopTasks(), testing::ElementsAre(2, 0, 1, 3, 4));
 }
 
-TEST_F(RendererTaskQueueSelectorTest, TestBestEffortPriority) {
+TEST_F(TaskQueueSelectorImplTest, TestBestEffortPriority) {
   std::vector<base::PendingTask> tasks = GetTasks(5);
   size_t queue_order[] = {0, 1, 2, 3, 4};
   PushTasks(tasks, queue_order);
-  selector_.SetQueuePriority(0,
-                             RendererTaskQueueSelector::BEST_EFFORT_PRIORITY);
-  selector_.SetQueuePriority(2, RendererTaskQueueSelector::HIGH_PRIORITY);
+  selector_.SetQueuePriority(0, TaskQueueSelectorImpl::BEST_EFFORT_PRIORITY);
+  selector_.SetQueuePriority(2, TaskQueueSelectorImpl::HIGH_PRIORITY);
   EXPECT_THAT(PopTasks(), testing::ElementsAre(2, 1, 3, 4, 0));
 }
 
-TEST_F(RendererTaskQueueSelectorTest, TestControlPriority) {
+TEST_F(TaskQueueSelectorImplTest, TestControlPriority) {
   std::vector<base::PendingTask> tasks = GetTasks(5);
   size_t queue_order[] = {0, 1, 2, 3, 4};
   PushTasks(tasks, queue_order);
-  selector_.SetQueuePriority(4, RendererTaskQueueSelector::CONTROL_PRIORITY);
+  selector_.SetQueuePriority(4, TaskQueueSelectorImpl::CONTROL_PRIORITY);
   EXPECT_TRUE(selector_.IsQueueEnabled(4));
-  selector_.SetQueuePriority(2, RendererTaskQueueSelector::HIGH_PRIORITY);
+  selector_.SetQueuePriority(2, TaskQueueSelectorImpl::HIGH_PRIORITY);
   EXPECT_TRUE(selector_.IsQueueEnabled(2));
   EXPECT_THAT(PopTasks(), testing::ElementsAre(4, 2, 0, 1, 3));
 }
 
-TEST_F(RendererTaskQueueSelectorTest, TestDisableEnable) {
+TEST_F(TaskQueueSelectorImplTest, TestDisableEnable) {
   std::vector<base::PendingTask> tasks = GetTasks(5);
   size_t queue_order[] = {0, 1, 2, 3, 4};
   PushTasks(tasks, queue_order);
   selector_.DisableQueue(2);
   EXPECT_FALSE(selector_.IsQueueEnabled(2));
   selector_.DisableQueue(4);
   EXPECT_FALSE(selector_.IsQueueEnabled(4));
   EXPECT_THAT(PopTasks(), testing::ElementsAre(0, 1, 3));
-  selector_.EnableQueue(2, RendererTaskQueueSelector::BEST_EFFORT_PRIORITY);
+  selector_.EnableQueue(2, TaskQueueSelectorImpl::BEST_EFFORT_PRIORITY);
   EXPECT_THAT(PopTasks(), testing::ElementsAre(2));
-  selector_.EnableQueue(4, RendererTaskQueueSelector::NORMAL_PRIORITY);
+  selector_.EnableQueue(4, TaskQueueSelectorImpl::NORMAL_PRIORITY);
   EXPECT_THAT(PopTasks(), testing::ElementsAre(4));
 }
 
-TEST_F(RendererTaskQueueSelectorTest, TestEmptyQueues) {
+TEST_F(TaskQueueSelectorImplTest, TestEmptyQueues) {
   size_t chosen_queue_index = 0;
   EXPECT_FALSE(selector_.SelectWorkQueueToService(&chosen_queue_index));
 
   // Test only disabled queues.
   std::vector<base::PendingTask> tasks = GetTasks(1);
   size_t queue_order[] = {0};
   PushTasks(tasks, queue_order);
   selector_.DisableQueue(0);
   EXPECT_FALSE(selector_.IsQueueEnabled(0));
   EXPECT_FALSE(selector_.SelectWorkQueueToService(&chosen_queue_index));
 }
 
-TEST_F(RendererTaskQueueSelectorTest, TestDelay) {
+TEST_F(TaskQueueSelectorImplTest, TestDelay) {
   std::vector<base::PendingTask> tasks = GetTasks(5);
   tasks[0].delayed_run_time =
       base::TimeTicks() + base::TimeDelta::FromMilliseconds(200);
   tasks[3].delayed_run_time =
       base::TimeTicks() + base::TimeDelta::FromMilliseconds(100);
   size_t queue_order[] = {0, 1, 2, 3, 4};
   PushTasks(tasks, queue_order);
   EXPECT_THAT(PopTasks(), testing::ElementsAre(1, 2, 4, 3, 0));
 }
 
-TEST_F(RendererTaskQueueSelectorTest, TestControlStarvesOthers) {
+TEST_F(TaskQueueSelectorImplTest, TestControlStarvesOthers) {
   std::vector<base::PendingTask> tasks = GetTasks(4);
   size_t queue_order[] = {0, 1, 2, 3};
   PushTasks(tasks, queue_order);
-  selector_.SetQueuePriority(3, RendererTaskQueueSelector::CONTROL_PRIORITY);
-  selector_.SetQueuePriority(2, RendererTaskQueueSelector::HIGH_PRIORITY);
-  selector_.SetQueuePriority(1,
-                             RendererTaskQueueSelector::BEST_EFFORT_PRIORITY);
+  selector_.SetQueuePriority(3, TaskQueueSelectorImpl::CONTROL_PRIORITY);
+  selector_.SetQueuePriority(2, TaskQueueSelectorImpl::HIGH_PRIORITY);
+  selector_.SetQueuePriority(1, TaskQueueSelectorImpl::BEST_EFFORT_PRIORITY);
   for (int i = 0; i < 100; i++) {
     size_t chosen_queue_index = 0;
     EXPECT_TRUE(selector_.SelectWorkQueueToService(&chosen_queue_index));
     EXPECT_EQ(3ul, chosen_queue_index);
     // Don't remove task from queue to simulate all queues still being full.
   }
 }
 
-TEST_F(RendererTaskQueueSelectorTest, TestHighPriorityDoesNotStarveNormal) {
+TEST_F(TaskQueueSelectorImplTest, TestHighPriorityDoesNotStarveNormal) {
   std::vector<base::PendingTask> tasks = GetTasks(3);
   size_t queue_order[] = {0, 1, 2};
   PushTasks(tasks, queue_order);
-  selector_.SetQueuePriority(2, RendererTaskQueueSelector::HIGH_PRIORITY);
-  selector_.SetQueuePriority(1,
-                             RendererTaskQueueSelector::BEST_EFFORT_PRIORITY);
+  selector_.SetQueuePriority(2, TaskQueueSelectorImpl::HIGH_PRIORITY);
+  selector_.SetQueuePriority(1, TaskQueueSelectorImpl::BEST_EFFORT_PRIORITY);
   size_t counts[] = {0, 0, 0};
   for (int i = 0; i < 100; i++) {
     size_t chosen_queue_index = 0;
     EXPECT_TRUE(selector_.SelectWorkQueueToService(&chosen_queue_index));
     counts[chosen_queue_index]++;
     // Don't remove task from queue to simulate all queues still being full.
   }
   EXPECT_GT(counts[0], 0ul);        // Check high doesn't starve normal.
   EXPECT_GT(counts[2], counts[0]);  // Check high gets more chance to run.
   EXPECT_EQ(0ul, counts[1]);        // Check best effort is starved.
 }
 
-TEST_F(RendererTaskQueueSelectorTest, TestBestEffortGetsStarved) {
+TEST_F(TaskQueueSelectorImplTest, TestBestEffortGetsStarved) {
   std::vector<base::PendingTask> tasks = GetTasks(2);
   size_t queue_order[] = {0, 1};
   PushTasks(tasks, queue_order);
-  selector_.SetQueuePriority(0,
-                             RendererTaskQueueSelector::BEST_EFFORT_PRIORITY);
-  selector_.SetQueuePriority(1, RendererTaskQueueSelector::NORMAL_PRIORITY);
+  selector_.SetQueuePriority(0, TaskQueueSelectorImpl::BEST_EFFORT_PRIORITY);
+  selector_.SetQueuePriority(1, TaskQueueSelectorImpl::NORMAL_PRIORITY);
   size_t chosen_queue_index = 0;
   for (int i = 0; i < 100; i++) {
     EXPECT_TRUE(selector_.SelectWorkQueueToService(&chosen_queue_index));
     EXPECT_EQ(1ul, chosen_queue_index);
     // Don't remove task from queue to simulate all queues still being full.
   }
-  selector_.SetQueuePriority(1, RendererTaskQueueSelector::HIGH_PRIORITY);
+  selector_.SetQueuePriority(1, TaskQueueSelectorImpl::HIGH_PRIORITY);
   for (int i = 0; i < 100; i++) {
     EXPECT_TRUE(selector_.SelectWorkQueueToService(&chosen_queue_index));
     EXPECT_EQ(1ul, chosen_queue_index);
     // Don't remove task from queue to simulate all queues still being full.
   }
-  selector_.SetQueuePriority(1, RendererTaskQueueSelector::CONTROL_PRIORITY);
+  selector_.SetQueuePriority(1, TaskQueueSelectorImpl::CONTROL_PRIORITY);
   for (int i = 0; i < 100; i++) {
     EXPECT_TRUE(selector_.SelectWorkQueueToService(&chosen_queue_index));
     EXPECT_EQ(1ul, chosen_queue_index);
     // Don't remove task from queue to simulate all queues still being full.
   }
 }
 
 }  // namespace content