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/task_scheduler/delayed_task_manager.h" | 5 #include "base/task_scheduler/delayed_task_manager.h" |
6 | 6 |
7 #include <memory> | 7 #include <memory> |
8 #include <utility> | 8 #include <utility> |
9 | 9 |
10 #include "base/bind.h" | 10 #include "base/bind.h" |
11 #include "base/logging.h" | |
12 #include "base/memory/ref_counted.h" | 11 #include "base/memory/ref_counted.h" |
| 12 #include "base/sequenced_task_runner.h" |
| 13 #include "base/single_thread_task_runner.h" |
| 14 #include "base/task_runner.h" |
13 #include "base/task_scheduler/scheduler_worker_pool.h" | 15 #include "base/task_scheduler/scheduler_worker_pool.h" |
14 #include "base/task_scheduler/sequence.h" | 16 #include "base/task_scheduler/sequence.h" |
15 #include "base/task_scheduler/task.h" | 17 #include "base/task_scheduler/task.h" |
16 #include "base/test/test_mock_time_task_runner.h" | 18 #include "base/test/test_mock_time_task_runner.h" |
17 #include "base/time/time.h" | 19 #include "base/time/time.h" |
18 #include "testing/gmock/include/gmock/gmock.h" | 20 #include "testing/gmock/include/gmock/gmock.h" |
19 #include "testing/gtest/include/gtest/gtest.h" | 21 #include "testing/gtest/include/gtest/gtest.h" |
20 | 22 |
21 namespace base { | 23 namespace base { |
22 namespace internal { | 24 namespace internal { |
23 namespace { | 25 namespace { |
24 | 26 |
25 constexpr TimeDelta kLongDelay = TimeDelta::FromHours(1); | 27 constexpr TimeDelta kLongDelay = TimeDelta::FromHours(1); |
26 | 28 |
27 class MockSchedulerWorkerPool : public SchedulerWorkerPool { | 29 class MockSchedulerWorkerPool : public SchedulerWorkerPool { |
28 public: | 30 public: |
29 // SchedulerWorkerPool: | 31 // SchedulerWorkerPool: |
30 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits( | 32 scoped_refptr<TaskRunner> CreateTaskRunnerWithTraits( |
31 const TaskTraits& traits, | 33 const TaskTraits& traits) override { |
32 ExecutionMode execution_mode) override { | 34 ADD_FAILURE() << "Call to unimplemented method."; |
33 NOTREACHED(); | 35 return nullptr; |
| 36 } |
| 37 |
| 38 scoped_refptr<SequencedTaskRunner> CreateSequencedTaskRunnerWithTraits( |
| 39 const TaskTraits& traits) override { |
| 40 ADD_FAILURE() << "Call to unimplemented method."; |
| 41 return nullptr; |
| 42 } |
| 43 |
| 44 scoped_refptr<SingleThreadTaskRunner> CreateSingleThreadTaskRunnerWithTraits( |
| 45 const TaskTraits& traits) override { |
| 46 ADD_FAILURE() << "Call to unimplemented method."; |
34 return nullptr; | 47 return nullptr; |
35 } | 48 } |
36 | 49 |
37 void ReEnqueueSequence(scoped_refptr<Sequence> sequence, | 50 void ReEnqueueSequence(scoped_refptr<Sequence> sequence, |
38 const SequenceSortKey& sequence_sort_key) override { | 51 const SequenceSortKey& sequence_sort_key) override { |
39 NOTREACHED(); | 52 ADD_FAILURE() << "Call to unimplemented method."; |
40 } | 53 } |
41 | 54 |
42 bool PostTaskWithSequence(std::unique_ptr<Task> task, | 55 bool PostTaskWithSequence(std::unique_ptr<Task> task, |
43 scoped_refptr<Sequence> sequence, | 56 scoped_refptr<Sequence> sequence, |
44 SchedulerWorker* worker) override { | 57 SchedulerWorker* worker) override { |
45 NOTREACHED(); | 58 ADD_FAILURE() << "Call to unimplemented method."; |
46 return true; | 59 return true; |
47 } | 60 } |
48 | 61 |
49 void PostTaskWithSequenceNow(std::unique_ptr<Task> task, | 62 void PostTaskWithSequenceNow(std::unique_ptr<Task> task, |
50 scoped_refptr<Sequence> sequence, | 63 scoped_refptr<Sequence> sequence, |
51 SchedulerWorker* worker) override { | 64 SchedulerWorker* worker) override { |
52 PostTaskWithSequenceNowMock(task.get(), sequence.get(), worker); | 65 PostTaskWithSequenceNowMock(task.get(), sequence.get(), worker); |
53 } | 66 } |
54 | 67 |
55 MOCK_METHOD3(PostTaskWithSequenceNowMock, | 68 MOCK_METHOD3(PostTaskWithSequenceNowMock, |
(...skipping 88 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
144 | 157 |
145 // Fast-forward time. Expect |task_b_raw| to be forwarded to the worker pool. | 158 // Fast-forward time. Expect |task_b_raw| to be forwarded to the worker pool. |
146 EXPECT_CALL(worker_pool, | 159 EXPECT_CALL(worker_pool, |
147 PostTaskWithSequenceNowMock(task_b_raw, sequence.get(), nullptr)); | 160 PostTaskWithSequenceNowMock(task_b_raw, sequence.get(), nullptr)); |
148 service_thread_task_runner->FastForwardBy(TimeDelta::FromHours(1)); | 161 service_thread_task_runner->FastForwardBy(TimeDelta::FromHours(1)); |
149 testing::Mock::VerifyAndClear(&worker_pool); | 162 testing::Mock::VerifyAndClear(&worker_pool); |
150 } | 163 } |
151 | 164 |
152 } // namespace internal | 165 } // namespace internal |
153 } // namespace base | 166 } // namespace base |
OLD | NEW |