Index: base/test/scoped_task_scheduler.cc |
diff --git a/base/test/scoped_task_scheduler.cc b/base/test/scoped_task_scheduler.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..76965378e7eadb032236d7f09d53085ceb302f68 |
--- /dev/null |
+++ b/base/test/scoped_task_scheduler.cc |
@@ -0,0 +1,29 @@ |
+// 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 "base/task_scheduler/task_scheduler.h" |
+ |
+namespace base { |
+namespace test { |
+ |
+ScopedTaskScheduler::ScopedTaskScheduler() { |
+ DCHECK(!TaskScheduler::GetInstance()); |
+ |
+ // Create a TaskScheduler with a single thread to make tests deterministic. |
+ constexpr int kMaxThreads = 1; |
+ TaskScheduler::CreateAndSetSimpleTaskScheduler(kMaxThreads); |
+ 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 |