| 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" | 10 #include "base/test/simple_test_tick_clock.h" |
| (...skipping 17 matching lines...) Expand all Loading... |
| 28 | 28 |
| 29 class MockTaskObserver : public blink::WebThread::TaskObserver { | 29 class MockTaskObserver : public blink::WebThread::TaskObserver { |
| 30 public: | 30 public: |
| 31 MOCK_METHOD0(willProcessTask, void()); | 31 MOCK_METHOD0(willProcessTask, void()); |
| 32 MOCK_METHOD0(didProcessTask, void()); | 32 MOCK_METHOD0(didProcessTask, void()); |
| 33 }; | 33 }; |
| 34 } // namespace | 34 } // namespace |
| 35 | 35 |
| 36 class WebThreadImplForRendererSchedulerTest : public testing::Test { | 36 class WebThreadImplForRendererSchedulerTest : public testing::Test { |
| 37 public: | 37 public: |
| 38 WebThreadImplForRendererSchedulerTest() {} | 38 WebThreadImplForRendererSchedulerTest() |
| 39 | 39 : clock_(new base::SimpleTestTickClock()), |
| 40 void SetUp() override { | 40 scheduler_(SchedulerTqmDelegateImpl::Create( |
| 41 clock_.reset(new base::SimpleTestTickClock()); | 41 &message_loop_, |
| 42 clock_->Advance(base::TimeDelta::FromMicroseconds(5000)); | 42 make_scoped_ptr(new TestTimeSource(clock_.get())))), |
| 43 scheduler_.reset(new RendererSchedulerImpl(SchedulerTqmDelegateImpl::Create( | 43 default_task_runner_(scheduler_.DefaultTaskRunner()), |
| 44 &message_loop_, make_scoped_ptr(new TestTimeSource(clock_.get()))))); | 44 thread_(scheduler_.CreateMainThread()) {} |
| 45 default_task_runner_ = scheduler_->DefaultTaskRunner(); | |
| 46 thread_ = scheduler_->CreateMainThread(); | |
| 47 } | |
| 48 | 45 |
| 49 ~WebThreadImplForRendererSchedulerTest() override {} | 46 ~WebThreadImplForRendererSchedulerTest() override {} |
| 50 | 47 |
| 51 void SetWorkBatchSizeForTesting(size_t work_batch_size) { | 48 void SetWorkBatchSizeForTesting(size_t work_batch_size) { |
| 52 scheduler_->GetSchedulerHelperForTesting()->SetWorkBatchSizeForTesting( | 49 scheduler_.GetSchedulerHelperForTesting()->SetWorkBatchSizeForTesting( |
| 53 work_batch_size); | 50 work_batch_size); |
| 54 } | 51 } |
| 55 | 52 |
| 56 void TearDown() override { scheduler_->Shutdown(); } | 53 void TearDown() override { scheduler_.Shutdown(); } |
| 57 | 54 |
| 58 protected: | 55 protected: |
| 59 base::MessageLoop message_loop_; | 56 base::MessageLoop message_loop_; |
| 60 scoped_ptr<base::SimpleTestTickClock> clock_; | 57 scoped_ptr<base::SimpleTestTickClock> clock_; |
| 61 scoped_ptr<RendererSchedulerImpl> scheduler_; | 58 RendererSchedulerImpl scheduler_; |
| 62 scoped_refptr<base::SingleThreadTaskRunner> default_task_runner_; | 59 scoped_refptr<base::SingleThreadTaskRunner> default_task_runner_; |
| 63 scoped_ptr<blink::WebThread> thread_; | 60 scoped_ptr<blink::WebThread> thread_; |
| 64 | 61 |
| 65 DISALLOW_COPY_AND_ASSIGN(WebThreadImplForRendererSchedulerTest); | 62 DISALLOW_COPY_AND_ASSIGN(WebThreadImplForRendererSchedulerTest); |
| 66 }; | 63 }; |
| 67 | 64 |
| 68 TEST_F(WebThreadImplForRendererSchedulerTest, TestTaskObserver) { | 65 TEST_F(WebThreadImplForRendererSchedulerTest, TestTaskObserver) { |
| 69 MockTaskObserver observer; | 66 MockTaskObserver observer; |
| 70 thread_->addTaskObserver(&observer); | 67 thread_->addTaskObserver(&observer); |
| 71 scoped_ptr<MockTask> task(new MockTask()); | 68 scoped_ptr<MockTask> task(new MockTask()); |
| (...skipping 121 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 193 } | 190 } |
| 194 | 191 |
| 195 message_loop_.task_runner()->PostTask( | 192 message_loop_.task_runner()->PostTask( |
| 196 FROM_HERE, base::Bind(&EnterRunLoop, base::Unretained(&message_loop_), | 193 FROM_HERE, base::Bind(&EnterRunLoop, base::Unretained(&message_loop_), |
| 197 base::Unretained(thread_.get()))); | 194 base::Unretained(thread_.get()))); |
| 198 message_loop_.RunUntilIdle(); | 195 message_loop_.RunUntilIdle(); |
| 199 thread_->removeTaskObserver(&observer); | 196 thread_->removeTaskObserver(&observer); |
| 200 } | 197 } |
| 201 | 198 |
| 202 } // namespace scheduler | 199 } // namespace scheduler |
| OLD | NEW |