Add ScopedTaskEnvironment::GetMainThreadTaskRunner().

base/test/scoped_task_environment.h

 // Copyright 2017 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 BASE_TEST_SCOPED_TASK_ENVIRONMENT_H_
 #define BASE_TEST_SCOPED_TASK_ENVIRONMENT_H_
 
 #include "base/macros.h"
 #include "base/message_loop/message_loop.h"
+#include "base/single_thread_task_runner.h"
 
 namespace base {
 
 class TaskScheduler;
 
 namespace test {
 
 // ScopedTaskEnvironment allows usage of these APIs within its scope:
 // - (Thread|Sequenced)TaskRunnerHandle, on the thread where it lives
 // - base/task_scheduler/post_task.h, on any thread
@@ skipping 40 matching lines @@
     IO,
   };
 
   ScopedTaskEnvironment(
       MainThreadType main_thread_type = MainThreadType::DEFAULT);
 
   // Waits until no undelayed TaskScheduler tasks remain. Then, unregisters the
   // TaskScheduler and the (Thread|Sequenced)TaskRunnerHandle.
   ~ScopedTaskEnvironment();
 
+  // Returns a TaskRunner that schedules tasks on the main thread.
+  scoped_refptr<base::SingleThreadTaskRunner> GetMainThreadTaskRunner();
+
   // Synchronously runs (Thread|Sequenced)TaskRunnerHandle tasks until no
   // undelayed (Thread|Sequenced)TaskRunnerHandle or TaskScheduler tasks remain.
   void RunUntilIdle();
 
  private:
   // Note: |message_loop_| is an implementation detail and will be replaced in
   // the future, do NOT rely on the presence of a MessageLoop beyond
   // (Thread|Sequenced)TaskRunnerHandle and RunLoop.
   MessageLoop message_loop_;
 
   const TaskScheduler* task_scheduler_ = nullptr;
 
   DISALLOW_COPY_AND_ASSIGN(ScopedTaskEnvironment);
 };
 
 }  // namespace test
 }  // namespace base
 
 #endif  // BASE_TEST_SCOPED_ASYNC_TASK_SCHEDULER_H_