Redirect the MessageLoop's task runner to the renderer scheduler

components/scheduler/child/scheduler_task_runner_delegate_for_test.h

 // 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.
 
-#ifndef CONTENT_RENDERER_SCHEDULER_NESTABLE_TASK_RUNNER_FOR_TEST_H_
-#define CONTENT_RENDERER_SCHEDULER_NESTABLE_TASK_RUNNER_FOR_TEST_H_
+#ifndef CONTENT_RENDERER_SCHEDULER_TASK_RUNNER_DELEGATE_FOR_TEST_H_
+#define CONTENT_RENDERER_SCHEDULER_TASK_RUNNER_DELEGATE_FOR_TEST_H_
 
-#include "components/scheduler/child/nestable_single_thread_task_runner.h"
+#include "components/scheduler/child/scheduler_task_runner_delegate.h"
 
 namespace scheduler {
 
-class NestableTaskRunnerForTest : public NestableSingleThreadTaskRunner {
+class NestableTaskRunnerForTest;
+
+class SchedulerTaskRunnerDelegateForTest : public SchedulerTaskRunnerDelegate {
  public:
-  static scoped_refptr<NestableTaskRunnerForTest> Create(
+  static scoped_refptr<SchedulerTaskRunnerDelegateForTest> Create(
       scoped_refptr<base::SingleThreadTaskRunner> task_runner);
 
-  void SetNested(bool is_nested);
-
-  // NestableSingleThreadTaskRunner implementation
+  // SchedulerTaskRunnerDelegate implementation
+  void SetDefaultTaskRunner(
+      scoped_refptr<base::SingleThreadTaskRunner> task_runner) override;
   bool PostDelayedTask(const tracked_objects::Location& from_here,
                        const base::Closure& task,
                        base::TimeDelta delay) override;
   bool PostNonNestableDelayedTask(const tracked_objects::Location& from_here,
                                   const base::Closure& task,
                                   base::TimeDelta delay) override;
   bool RunsTasksOnCurrentThread() const override;
   bool IsNested() const override;
   void AddTaskObserver(base::MessageLoop::TaskObserver* task_observer) override;
   void RemoveTaskObserver(
       base::MessageLoop::TaskObserver* task_observer) override;
 
+  base::SingleThreadTaskRunner* default_task_runner() const {
+    return default_task_runner_.get();
+  }
+
  protected:
-  ~NestableTaskRunnerForTest() override;
+  ~SchedulerTaskRunnerDelegateForTest() override;
 
  private:
-  NestableTaskRunnerForTest(
+  explicit SchedulerTaskRunnerDelegateForTest(
       scoped_refptr<base::SingleThreadTaskRunner> task_runner);
 
-  void WrapTask(const base::PendingTask* wrapped_task);
+  scoped_refptr<base::SingleThreadTaskRunner> default_task_runner_;
 
-  scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
-  bool is_nested_;
+  scoped_refptr<NestableTaskRunnerForTest> task_runner_;
 
-  base::ObserverList<base::MessageLoop::TaskObserver> task_observers_;
-
-  base::WeakPtr<NestableTaskRunnerForTest> weak_nestable_task_runner_ptr_;
-  base::WeakPtrFactory<NestableTaskRunnerForTest> weak_factory_;
-
-  DISALLOW_COPY_AND_ASSIGN(NestableTaskRunnerForTest);
+  DISALLOW_COPY_AND_ASSIGN(SchedulerTaskRunnerDelegateForTest);
 };
 
 }  // namespace scheduler
 
-#endif  // CONTENT_RENDERER_SCHEDULER_NESTABLE_TASK_RUNNER_FOR_TEST_H_
+#endif  // CONTENT_RENDERER_SCHEDULER_SCHEDULER_TASK_RUNNER_DELEGATE_FOR_TEST_H_