Redirect the MessageLoop's task runner to the renderer scheduler

components/scheduler/child/scheduler_helper_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/scheduler_helper.h"
 
 #include "base/callback.h"
 #include "base/test/simple_test_tick_clock.h"
 #include "cc/test/ordered_simple_task_runner.h"
-#include "components/scheduler/child/nestable_task_runner_for_test.h"
-#include "components/scheduler/child/scheduler_message_loop_delegate.h"
+#include "components/scheduler/child/scheduler_task_runner_delegate_for_test.h"
 #include "components/scheduler/child/task_queue.h"
 #include "components/scheduler/child/test_time_source.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::_;
 using testing::AnyNumber;
 using testing::Invoke;
 using testing::Return;
 
@@ skipping 18 matching lines @@
   }
 }
 
 };  // namespace
 
 class SchedulerHelperTest : public testing::Test {
  public:
   SchedulerHelperTest()
       : clock_(new base::SimpleTestTickClock()),
         mock_task_runner_(new cc::OrderedSimpleTaskRunner(clock_.get(), false)),
-        nestable_task_runner_(
-            NestableTaskRunnerForTest::Create(mock_task_runner_)),
+        main_task_runner_(
+            SchedulerTaskRunnerDelegateForTest::Create(mock_task_runner_)),
         scheduler_helper_(
-            new SchedulerHelper(nestable_task_runner_,
+            new SchedulerHelper(main_task_runner_,
                                 "test.scheduler",
                                 TRACE_DISABLED_BY_DEFAULT("test.scheduler"),
                                 TRACE_DISABLED_BY_DEFAULT("test.scheduler.dbg"),
                                 SchedulerHelper::TASK_QUEUE_COUNT)),
         default_task_runner_(scheduler_helper_->DefaultTaskRunner()) {
     clock_->Advance(base::TimeDelta::FromMicroseconds(5000));
     scheduler_helper_->SetTimeSourceForTesting(
         make_scoped_ptr(new TestTimeSource(clock_.get())));
     scheduler_helper_->GetTaskQueueManagerForTesting()->SetTimeSourceForTesting(
         make_scoped_ptr(new TestTimeSource(clock_.get())));
@@ skipping 23 matching lines @@
   static void CheckAllTaskQueueIdToString() {
     CallForEachEnumValue<SchedulerHelper::QueueId>(
         SchedulerHelper::FIRST_QUEUE_ID, SchedulerHelper::TASK_QUEUE_COUNT,
         &SchedulerHelper::TaskQueueIdToString);
   }
 
  protected:
   scoped_ptr<base::SimpleTestTickClock> clock_;
   scoped_refptr<cc::OrderedSimpleTaskRunner> mock_task_runner_;
 
-  scoped_refptr<NestableSingleThreadTaskRunner> nestable_task_runner_;
+  scoped_refptr<SchedulerTaskRunnerDelegateForTest> main_task_runner_;
   scoped_ptr<SchedulerHelper> scheduler_helper_;
   scoped_refptr<base::SingleThreadTaskRunner> default_task_runner_;
 
   DISALLOW_COPY_AND_ASSIGN(SchedulerHelperTest);
 };
 
 TEST_F(SchedulerHelperTest, TestPostDefaultTask) {
   std::vector<std::string> run_order;
   default_task_runner_->PostTask(
       FROM_HERE, base::Bind(&AppendToVectorTestTask, &run_order, "D1"));
@@ skipping 25 matching lines @@
   EXPECT_FALSE(scheduler_helper_->IsShutdown());
 
   scheduler_helper_->Shutdown();
   EXPECT_TRUE(scheduler_helper_->IsShutdown());
 }
 
 TEST_F(SchedulerHelperTest, TaskQueueIdToString) {
   CheckAllTaskQueueIdToString();
 }
 
+TEST_F(SchedulerHelperTest, DefaultTaskRunnerRegistration) {
+  EXPECT_EQ(main_task_runner_->default_task_runner(),
+            scheduler_helper_->DefaultTaskRunner());
+  scheduler_helper_->Shutdown();
+  EXPECT_EQ(nullptr, main_task_runner_->default_task_runner());
+}
+
 }  // namespace scheduler