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