| 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 |