OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "components/scheduler/renderer/webthread_impl_for_renderer_scheduler.h" | 5 #include "components/scheduler/renderer/webthread_impl_for_renderer_scheduler.h" |
6 | 6 |
7 #include "base/location.h" | 7 #include "base/location.h" |
8 #include "base/run_loop.h" | 8 #include "base/run_loop.h" |
9 #include "base/single_thread_task_runner.h" | 9 #include "base/single_thread_task_runner.h" |
10 #include "components/scheduler/child/scheduler_task_runner_delegate_impl.h" | 10 #include "base/test/simple_test_tick_clock.h" |
| 11 #include "components/scheduler/base/test_time_source.h" |
| 12 #include "components/scheduler/child/scheduler_tqm_delegate_impl.h" |
11 #include "components/scheduler/renderer/renderer_scheduler_impl.h" | 13 #include "components/scheduler/renderer/renderer_scheduler_impl.h" |
12 #include "testing/gmock/include/gmock/gmock.h" | 14 #include "testing/gmock/include/gmock/gmock.h" |
13 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
14 #include "third_party/WebKit/public/platform/WebTaskRunner.h" | 16 #include "third_party/WebKit/public/platform/WebTaskRunner.h" |
15 #include "third_party/WebKit/public/platform/WebTraceLocation.h" | 17 #include "third_party/WebKit/public/platform/WebTraceLocation.h" |
16 | 18 |
17 namespace scheduler { | 19 namespace scheduler { |
18 namespace { | 20 namespace { |
19 | 21 |
20 const int kWorkBatchSize = 2; | 22 const int kWorkBatchSize = 2; |
21 | 23 |
22 class MockTask : public blink::WebTaskRunner::Task { | 24 class MockTask : public blink::WebTaskRunner::Task { |
23 public: | 25 public: |
24 MOCK_METHOD0(run, void()); | 26 MOCK_METHOD0(run, void()); |
25 }; | 27 }; |
26 | 28 |
27 class MockTaskObserver : public blink::WebThread::TaskObserver { | 29 class MockTaskObserver : public blink::WebThread::TaskObserver { |
28 public: | 30 public: |
29 MOCK_METHOD0(willProcessTask, void()); | 31 MOCK_METHOD0(willProcessTask, void()); |
30 MOCK_METHOD0(didProcessTask, void()); | 32 MOCK_METHOD0(didProcessTask, void()); |
31 }; | 33 }; |
32 } // namespace | 34 } // namespace |
33 | 35 |
34 class WebThreadImplForRendererSchedulerTest : public testing::Test { | 36 class WebThreadImplForRendererSchedulerTest : public testing::Test { |
35 public: | 37 public: |
36 WebThreadImplForRendererSchedulerTest() | 38 WebThreadImplForRendererSchedulerTest() |
37 : scheduler_(SchedulerTaskRunnerDelegateImpl::Create(&message_loop_)), | 39 : clock_(new base::SimpleTestTickClock()), |
| 40 scheduler_(SchedulerTqmDelegateImpl::Create( |
| 41 &message_loop_, |
| 42 make_scoped_ptr(new TestTimeSource(clock_.get())))), |
38 default_task_runner_(scheduler_.DefaultTaskRunner()), | 43 default_task_runner_(scheduler_.DefaultTaskRunner()), |
39 thread_(&scheduler_) {} | 44 thread_(&scheduler_) {} |
40 | 45 |
41 ~WebThreadImplForRendererSchedulerTest() override {} | 46 ~WebThreadImplForRendererSchedulerTest() override {} |
42 | 47 |
43 void SetWorkBatchSizeForTesting(size_t work_batch_size) { | 48 void SetWorkBatchSizeForTesting(size_t work_batch_size) { |
44 scheduler_.GetSchedulerHelperForTesting()->SetWorkBatchSizeForTesting( | 49 scheduler_.GetSchedulerHelperForTesting()->SetWorkBatchSizeForTesting( |
45 work_batch_size); | 50 work_batch_size); |
46 } | 51 } |
47 | 52 |
48 void TearDown() override { scheduler_.Shutdown(); } | 53 void TearDown() override { scheduler_.Shutdown(); } |
49 | 54 |
50 protected: | 55 protected: |
51 base::MessageLoop message_loop_; | 56 base::MessageLoop message_loop_; |
| 57 scoped_ptr<base::SimpleTestTickClock> clock_; |
52 RendererSchedulerImpl scheduler_; | 58 RendererSchedulerImpl scheduler_; |
53 scoped_refptr<base::SingleThreadTaskRunner> default_task_runner_; | 59 scoped_refptr<base::SingleThreadTaskRunner> default_task_runner_; |
54 WebThreadImplForRendererScheduler thread_; | 60 WebThreadImplForRendererScheduler thread_; |
55 | 61 |
56 DISALLOW_COPY_AND_ASSIGN(WebThreadImplForRendererSchedulerTest); | 62 DISALLOW_COPY_AND_ASSIGN(WebThreadImplForRendererSchedulerTest); |
57 }; | 63 }; |
58 | 64 |
59 TEST_F(WebThreadImplForRendererSchedulerTest, TestTaskObserver) { | 65 TEST_F(WebThreadImplForRendererSchedulerTest, TestTaskObserver) { |
60 MockTaskObserver observer; | 66 MockTaskObserver observer; |
61 thread_.addTaskObserver(&observer); | 67 thread_.addTaskObserver(&observer); |
(...skipping 122 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
184 } | 190 } |
185 | 191 |
186 message_loop_.task_runner()->PostTask( | 192 message_loop_.task_runner()->PostTask( |
187 FROM_HERE, base::Bind(&EnterRunLoop, base::Unretained(&message_loop_), | 193 FROM_HERE, base::Bind(&EnterRunLoop, base::Unretained(&message_loop_), |
188 base::Unretained(&thread_))); | 194 base::Unretained(&thread_))); |
189 message_loop_.RunUntilIdle(); | 195 message_loop_.RunUntilIdle(); |
190 thread_.removeTaskObserver(&observer); | 196 thread_.removeTaskObserver(&observer); |
191 } | 197 } |
192 | 198 |
193 } // namespace scheduler | 199 } // namespace scheduler |
OLD | NEW |