Create base::test::ScopedTaskScheduler.

base/test/scoped_task_scheduler.cc

+// Copyright 2016 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 "base/test/scoped_task_scheduler.h"
+
+#include <vector>
+
+#include "base/bind.h"
+#include "base/task_scheduler/scheduler_worker_pool_params.h"
+#include "base/task_scheduler/task_scheduler.h"
+#include "base/threading/platform_thread.h"
+#include "base/time/time.h"
+
+namespace base {
+namespace test {
+
+namespace {
+
+size_t GetWorkerPoolIndexForTraits(const TaskTraits&) {

[danakj, 2016/11/17 00:48:01]

Not used?

[fdoray, 2016/11/17 13:25:25]

Done.

+  // All TaskTraits are mapped to the single worker pool.
+  return 0;
+}
+
+}  // namespace
+
+ScopedTaskScheduler::ScopedTaskScheduler() {
+  DCHECK(!TaskScheduler::GetInstance());
+
+  // Create a TaskScheduler with a single thread to make tests deterministic.
+  TaskScheduler::CreateAndSetSimpleTaskScheduler(1, base::TimeDelta::Max());
+  task_scheduler_ = TaskScheduler::GetInstance();
+}
+
+ScopedTaskScheduler::~ScopedTaskScheduler() {
+  DCHECK_EQ(task_scheduler_, TaskScheduler::GetInstance());
+  TaskScheduler::GetInstance()->Shutdown();
+  TaskScheduler::GetInstance()->JoinForTesting();
+  TaskScheduler::SetInstance(nullptr);
+}
+
+}  // namespace test
+}  // namespace base