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 "base/test/simple_test_tick_clock.h" |
| 11 #include "components/scheduler/base/test_time_source.h" |
10 #include "components/scheduler/child/scheduler_task_runner_delegate_impl.h" | 12 #include "components/scheduler/child/scheduler_task_runner_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_(SchedulerTaskRunnerDelegateImpl::Create( |
| 41 &message_loop_, |
| 42 make_scoped_ptr(new TestTimeSource(clock_.get()))), |
| 43 RendererSchedulerImpl::TimeSource::NORMAL), |
38 default_task_runner_(scheduler_.DefaultTaskRunner()), | 44 default_task_runner_(scheduler_.DefaultTaskRunner()), |
39 thread_(&scheduler_) {} | 45 thread_(&scheduler_) {} |
40 | 46 |
41 ~WebThreadImplForRendererSchedulerTest() override {} | 47 ~WebThreadImplForRendererSchedulerTest() override {} |
42 | 48 |
43 void SetWorkBatchSizeForTesting(size_t work_batch_size) { | 49 void SetWorkBatchSizeForTesting(size_t work_batch_size) { |
44 scheduler_.GetSchedulerHelperForTesting()->SetWorkBatchSizeForTesting( | 50 scheduler_.GetSchedulerHelperForTesting()->SetWorkBatchSizeForTesting( |
45 work_batch_size); | 51 work_batch_size); |
46 } | 52 } |
47 | 53 |
48 void TearDown() override { scheduler_.Shutdown(); } | 54 void TearDown() override { scheduler_.Shutdown(); } |
49 | 55 |
50 protected: | 56 protected: |
51 base::MessageLoop message_loop_; | 57 base::MessageLoop message_loop_; |
| 58 scoped_ptr<base::SimpleTestTickClock> clock_; |
52 RendererSchedulerImpl scheduler_; | 59 RendererSchedulerImpl scheduler_; |
53 scoped_refptr<base::SingleThreadTaskRunner> default_task_runner_; | 60 scoped_refptr<base::SingleThreadTaskRunner> default_task_runner_; |
54 WebThreadImplForRendererScheduler thread_; | 61 WebThreadImplForRendererScheduler thread_; |
55 | 62 |
56 DISALLOW_COPY_AND_ASSIGN(WebThreadImplForRendererSchedulerTest); | 63 DISALLOW_COPY_AND_ASSIGN(WebThreadImplForRendererSchedulerTest); |
57 }; | 64 }; |
58 | 65 |
59 TEST_F(WebThreadImplForRendererSchedulerTest, TestTaskObserver) { | 66 TEST_F(WebThreadImplForRendererSchedulerTest, TestTaskObserver) { |
60 MockTaskObserver observer; | 67 MockTaskObserver observer; |
61 thread_.addTaskObserver(&observer); | 68 thread_.addTaskObserver(&observer); |
(...skipping 122 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
184 } | 191 } |
185 | 192 |
186 message_loop_.task_runner()->PostTask( | 193 message_loop_.task_runner()->PostTask( |
187 FROM_HERE, base::Bind(&EnterRunLoop, base::Unretained(&message_loop_), | 194 FROM_HERE, base::Bind(&EnterRunLoop, base::Unretained(&message_loop_), |
188 base::Unretained(&thread_))); | 195 base::Unretained(&thread_))); |
189 message_loop_.RunUntilIdle(); | 196 message_loop_.RunUntilIdle(); |
190 thread_.removeTaskObserver(&observer); | 197 thread_.removeTaskObserver(&observer); |
191 } | 198 } |
192 | 199 |
193 } // namespace scheduler | 200 } // namespace scheduler |
OLD | NEW |