Index: components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.cc |
diff --git a/components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.cc b/components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.cc |
index 999d8567a1f03be79f526b42b6aa6340254a80df..1ad2b172236c3c7e84eb8215dda6b8e58a78af66 100644 |
--- a/components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.cc |
+++ b/components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.cc |
@@ -33,9 +33,20 @@ base::MessageLoop* LazySchedulerMessageLoopDelegateForTests::EnsureMessageLoop() |
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; |
} |