Redirect the MessageLoop's task runner to the renderer scheduler

components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.h

Index: components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.h
diff --git a/components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.h b/components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.h
index a727a6dd8be1b53ccafd397aa475629b448b346b..5f2eaec2e78db34f816378193c7d75498a25eb62 100644
--- a/components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.h
+++ b/components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.h
@@ -6,7 +6,7 @@
 #define COMPONENTS_SCHEDULER_TEST_LAZY_SCHEDULER_MESSAGE_LOOP_DELEGATE_FOR_TESTS_H_
 
 #include "base/message_loop/message_loop.h"
-#include "components/scheduler/child/nestable_single_thread_task_runner.h"
+#include "components/scheduler/child/scheduler_task_runner_delegate.h"
 
 namespace scheduler {
 
@@ -18,11 +18,14 @@ namespace scheduler {
 // TODO(skyostil): Fix the relevant test suites and remove this class
 // (crbug.com/495659).
 class LazySchedulerMessageLoopDelegateForTests
-    : public NestableSingleThreadTaskRunner {
+    : public SchedulerTaskRunnerDelegate {
  public:
   static scoped_refptr<LazySchedulerMessageLoopDelegateForTests> Create();
 
-  // NestableSingleThreadTaskRunner implementation
+  // SchedulerTaskRunnerDelegate implementation
+  void SetDefaultTaskRunner(
+      scoped_refptr<base::SingleThreadTaskRunner> task_runner) override;
+  void RestoreDefaultTaskRunner() override;
   bool PostDelayedTask(const tracked_objects::Location& from_here,
                        const base::Closure& task,
                        base::TimeDelta delay) override;
@@ -49,6 +52,10 @@ class LazySchedulerMessageLoopDelegateForTests
   // owned.
   mutable base::hash_set<base::MessageLoop::TaskObserver*> pending_observers_;
 
+  // A task runner which hasn't yet been overridden in the message loop.
+  mutable scoped_refptr<base::SingleThreadTaskRunner> pending_task_runner_;
+  mutable scoped_refptr<base::SingleThreadTaskRunner> original_task_runner_;
+
   DISALLOW_COPY_AND_ASSIGN(LazySchedulerMessageLoopDelegateForTests);
 };