| OLD | NEW | 
|   1 // Copyright 2016 The Chromium Authors. All rights reserved. |   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 |   2 // Use of this source code is governed by a BSD-style license that can be | 
|   3 // found in the LICENSE file. |   3 // found in the LICENSE file. | 
|   4  |   4  | 
|   5 #include "base/test/scoped_task_scheduler.h" |   5 #include "base/test/scoped_task_scheduler.h" | 
|   6  |   6  | 
|   7 #include <vector> |   7 #include <vector> | 
|   8  |   8  | 
|   9 #include "base/bind.h" |   9 #include "base/bind.h" | 
|  10 #include "base/task_scheduler/scheduler_worker_pool_params.h" |  10 #include "base/task_scheduler/scheduler_worker_pool_params.h" | 
|  11 #include "base/task_scheduler/task_scheduler.h" |  11 #include "base/task_scheduler/task_scheduler.h" | 
|  12 #include "base/threading/platform_thread.h" |  12 #include "base/threading/platform_thread.h" | 
|  13 #include "base/time/time.h" |  13 #include "base/time/time.h" | 
|  14  |  14  | 
|  15 namespace base { |  15 namespace base { | 
|  16 namespace test { |  16 namespace test { | 
|  17  |  17  | 
|  18 ScopedTaskScheduler::ScopedTaskScheduler() { |  18 ScopedTaskScheduler::ScopedTaskScheduler() { | 
|  19   DCHECK(!TaskScheduler::GetInstance()); |  19   DCHECK(!TaskScheduler::GetInstance()); | 
|  20  |  20  | 
|  21   // Create a TaskScheduler with a single thread to make tests deterministic. |  21   // Create a TaskScheduler with a single thread to make tests deterministic. | 
|  22   constexpr int kMaxThreads = 1; |  22   constexpr int kMaxThreads = 1; | 
|  23   std::vector<SchedulerWorkerPoolParams> worker_pool_params_vector; |  23   std::vector<SchedulerWorkerPoolParams> worker_pool_params_vector; | 
|  24   worker_pool_params_vector.emplace_back( |  24   worker_pool_params_vector.emplace_back( | 
|  25       "Simple", ThreadPriority::NORMAL, |  25       "Simple", ThreadPriority::NORMAL, | 
|  26       SchedulerWorkerPoolParams::IORestriction::ALLOWED, |  | 
|  27       SchedulerWorkerPoolParams::StandbyThreadPolicy::LAZY, kMaxThreads, |  26       SchedulerWorkerPoolParams::StandbyThreadPolicy::LAZY, kMaxThreads, | 
|  28       TimeDelta::Max()); |  27       TimeDelta::Max()); | 
|  29   TaskScheduler::CreateAndSetDefaultTaskScheduler( |  28   TaskScheduler::CreateAndSetDefaultTaskScheduler( | 
|  30       worker_pool_params_vector, |  29       worker_pool_params_vector, | 
|  31       Bind([](const TaskTraits&) -> size_t { return 0; })); |  30       Bind([](const TaskTraits&) -> size_t { return 0; })); | 
|  32   task_scheduler_ = TaskScheduler::GetInstance(); |  31   task_scheduler_ = TaskScheduler::GetInstance(); | 
|  33 } |  32 } | 
|  34  |  33  | 
|  35 ScopedTaskScheduler::~ScopedTaskScheduler() { |  34 ScopedTaskScheduler::~ScopedTaskScheduler() { | 
|  36   DCHECK_EQ(task_scheduler_, TaskScheduler::GetInstance()); |  35   DCHECK_EQ(task_scheduler_, TaskScheduler::GetInstance()); | 
|  37   TaskScheduler::GetInstance()->Shutdown(); |  36   TaskScheduler::GetInstance()->Shutdown(); | 
|  38   TaskScheduler::GetInstance()->JoinForTesting(); |  37   TaskScheduler::GetInstance()->JoinForTesting(); | 
|  39   TaskScheduler::SetInstance(nullptr); |  38   TaskScheduler::SetInstance(nullptr); | 
|  40 } |  39 } | 
|  41  |  40  | 
|  42 }  // namespace test |  41 }  // namespace test | 
|  43 }  // namespace base |  42 }  // namespace base | 
| OLD | NEW |