Redirect the MessageLoop's task runner to the renderer scheduler

components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.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 "components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.h"
 
 namespace scheduler {
 
 // static
 scoped_refptr<LazySchedulerMessageLoopDelegateForTests>
@@ skipping 15 matching lines @@
     const {
   if (message_loop_)
     return message_loop_;
   DCHECK(RunsTasksOnCurrentThread());
   message_loop_ = base::MessageLoop::current();
   DCHECK(message_loop_);
   for (auto& observer : pending_observers_) {
     message_loop_->AddTaskObserver(observer);
   }
   pending_observers_.clear();
+  if (pending_task_runner_)
+    message_loop_->SetTaskRunner(pending_task_runner_.Pass());
   return message_loop_;
 }
 
+void LazySchedulerMessageLoopDelegateForTests::SetDefaultTaskRunner(
+    scoped_refptr<base::SingleThreadTaskRunner> task_runner) {
+  if (!HasMessageLoop()) {
+    pending_task_runner_ = task_runner.Pass();
+    return;
+  }
+  message_loop_->SetTaskRunner(task_runner.Pass());
+}
+
 bool LazySchedulerMessageLoopDelegateForTests::HasMessageLoop() const {
   return message_loop_ != nullptr;
 }
 
 bool LazySchedulerMessageLoopDelegateForTests::PostDelayedTask(
     const tracked_objects::Location& from_here,
     const base::Closure& task,
     base::TimeDelta delay) {
   return EnsureMessageLoop()->task_runner()->PostDelayedTask(from_here, task,
                                                              delay);
@@ skipping 28 matching lines @@
 void LazySchedulerMessageLoopDelegateForTests::RemoveTaskObserver(
     base::MessageLoop::TaskObserver* task_observer) {
   if (!HasMessageLoop()) {
     pending_observers_.erase(task_observer);
     return;
   }
   EnsureMessageLoop()->RemoveTaskObserver(task_observer);
 }
 
 }  // namespace scheduler