Move blink scheduler implementation into a component

content/renderer/scheduler/webthread_impl_for_renderer_scheduler_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/renderer/scheduler/webthread_impl_for_renderer_scheduler.h"
 
 #include "base/run_loop.h"
-#include "content/child/scheduler/scheduler_message_loop_delegate.h"
-#include "content/renderer/scheduler/renderer_scheduler_impl.h"
+#include "components/scheduler/child/scheduler_message_loop_delegate.h"
+#include "components/scheduler/renderer/renderer_scheduler_impl.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/WebKit/public/platform/WebTraceLocation.h"
 
 namespace content {
 namespace {
 
 const int kWorkBatchSize = 2;
 
 class MockTask : public blink::WebThread::Task {
  public:
   MOCK_METHOD0(run, void());
 };
 
 class MockTaskObserver : public blink::WebThread::TaskObserver {
  public:
   MOCK_METHOD0(willProcessTask, void());
   MOCK_METHOD0(didProcessTask, void());
 };
 }  // namespace
 
 class WebThreadImplForRendererSchedulerTest : public testing::Test {
  public:
   WebThreadImplForRendererSchedulerTest()
-      : scheduler_(SchedulerMessageLoopDelegate::Create(&message_loop_)),
+      : scheduler_(
+            scheduler::SchedulerMessageLoopDelegate::Create(&message_loop_)),
         default_task_runner_(scheduler_.DefaultTaskRunner()),
         thread_(&scheduler_) {}
 
   ~WebThreadImplForRendererSchedulerTest() override {}
 
   void SetWorkBatchSizeForTesting(size_t work_batch_size) {
-    scheduler_.SetWorkBatchSizeForTesting(work_batch_size);
+    scheduler_.GetSchedulerHelperForTesting()->SetWorkBatchSizeForTesting(
+        work_batch_size);
   }
 
  protected:
   void EatDefaultTask(MockTaskObserver* observer) {
     // The scheduler posts one extra DoWork() task automatically.
     EXPECT_CALL(*observer, willProcessTask());
     EXPECT_CALL(*observer, didProcessTask());
   }
 
   base::MessageLoop message_loop_;
-  RendererSchedulerImpl scheduler_;
+  scheduler::RendererSchedulerImpl scheduler_;
   scoped_refptr<base::SingleThreadTaskRunner> default_task_runner_;
   WebThreadImplForRendererScheduler thread_;
 
   DISALLOW_COPY_AND_ASSIGN(WebThreadImplForRendererSchedulerTest);
 };
 
 TEST_F(WebThreadImplForRendererSchedulerTest, TestTaskObserver) {
   MockTaskObserver observer;
   thread_.addTaskObserver(&observer);
   scoped_ptr<MockTask> task(new MockTask());
@@ skipping 10 matching lines @@
   thread_.postTask(blink::WebTraceLocation(), task.release());
   message_loop_.RunUntilIdle();
   thread_.removeTaskObserver(&observer);
 }
 
 TEST_F(WebThreadImplForRendererSchedulerTest, TestWorkBatchWithOneTask) {
   MockTaskObserver observer;
   thread_.addTaskObserver(&observer);
   scoped_ptr<MockTask> task(new MockTask());
 
-  scheduler_.SetWorkBatchSizeForTesting(kWorkBatchSize);
+  SetWorkBatchSizeForTesting(kWorkBatchSize);
   {
     testing::InSequence sequence;
     EXPECT_CALL(observer, willProcessTask());
     EXPECT_CALL(*task, run());
     EXPECT_CALL(observer, didProcessTask());
 
     EatDefaultTask(&observer);
   }
 
   thread_.postTask(blink::WebTraceLocation(), task.release());
   message_loop_.RunUntilIdle();
   thread_.removeTaskObserver(&observer);
 }
 
 TEST_F(WebThreadImplForRendererSchedulerTest, TestWorkBatchWithTwoTasks) {
   MockTaskObserver observer;
   thread_.addTaskObserver(&observer);
   scoped_ptr<MockTask> task1(new MockTask());
   scoped_ptr<MockTask> task2(new MockTask());
 
-  scheduler_.SetWorkBatchSizeForTesting(kWorkBatchSize);
+  SetWorkBatchSizeForTesting(kWorkBatchSize);
   {
     testing::InSequence sequence;
     EXPECT_CALL(observer, willProcessTask());
     EXPECT_CALL(*task1, run());
     EXPECT_CALL(observer, didProcessTask());
 
     EXPECT_CALL(observer, willProcessTask());
     EXPECT_CALL(*task2, run());
     EXPECT_CALL(observer, didProcessTask());
 
     EatDefaultTask(&observer);
   }
 
   thread_.postTask(blink::WebTraceLocation(), task1.release());
   thread_.postTask(blink::WebTraceLocation(), task2.release());
   message_loop_.RunUntilIdle();
   thread_.removeTaskObserver(&observer);
 }
 
 TEST_F(WebThreadImplForRendererSchedulerTest, TestWorkBatchWithThreeTasks) {
   MockTaskObserver observer;
   thread_.addTaskObserver(&observer);
   scoped_ptr<MockTask> task1(new MockTask());
   scoped_ptr<MockTask> task2(new MockTask());
   scoped_ptr<MockTask> task3(new MockTask());
 
-  scheduler_.SetWorkBatchSizeForTesting(kWorkBatchSize);
+  SetWorkBatchSizeForTesting(kWorkBatchSize);
   {
     testing::InSequence sequence;
     EXPECT_CALL(observer, willProcessTask());
     EXPECT_CALL(*task1, run());
     EXPECT_CALL(observer, didProcessTask());
 
     EXPECT_CALL(observer, willProcessTask());
     EXPECT_CALL(*task2, run());
     EXPECT_CALL(observer, didProcessTask());
 
@@ skipping 51 matching lines @@
   }
 
   message_loop_.PostTask(
       FROM_HERE, base::Bind(&EnterRunLoop, base::Unretained(&message_loop_),
                             base::Unretained(&thread_)));
   message_loop_.RunUntilIdle();
   thread_.removeTaskObserver(&observer);
 }
 
 }  // namespace content