Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(230)

Side by Side Diff: base/test/scoped_task_scheduler.cc

Issue 2511473004: Create base::test::ScopedTaskScheduler. (Closed)
Patch Set: CR robliao/gab #4-6 Created 4 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 // Copyright 2016 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/test/scoped_task_scheduler.h"
6
7 #include <vector>
8
9 #include "base/bind.h"
10 #include "base/task_scheduler/scheduler_worker_pool_params.h"
11 #include "base/task_scheduler/task_scheduler.h"
12 #include "base/threading/platform_thread.h"
13 #include "base/time/time.h"
14
15 namespace base {
16 namespace test {
17
18 namespace {
19
20 size_t GetWorkerPoolIndexForTraits(const TaskTraits&) {
21 // All TaskTraits are mapped to the single worker pool.
22 return 0;
23 }
24
25 } // namespace
26
27 ScopedTaskScheduler::ScopedTaskScheduler() {
28 DCHECK(!TaskScheduler::GetInstance());
29
30 // Create a TaskScheduler with a single thread to make tests deterministic.
31 std::vector<SchedulerWorkerPoolParams> worker_pool_params_vector;
32 worker_pool_params_vector.emplace_back(
33 "Test", ThreadPriority::NORMAL,
34 SchedulerWorkerPoolParams::IORestriction::ALLOWED, 1, TimeDelta::Max());
35 TaskScheduler::CreateAndSetDefaultTaskScheduler(
36 worker_pool_params_vector, Bind(&GetWorkerPoolIndexForTraits));
37 task_scheduler_ = TaskScheduler::GetInstance();
38 }
39
40 ScopedTaskScheduler::~ScopedTaskScheduler() {
41 DCHECK_EQ(task_scheduler_, TaskScheduler::GetInstance());
42 TaskScheduler::GetInstance()->Shutdown();
43 TaskScheduler::GetInstance()->JoinForTesting();
44 TaskScheduler::SetInstance(nullptr);
45 }
46
47 } // namespace test
48 } // namespace base
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698