Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(76)

Side by Side Diff: components/scheduler/renderer/webthread_impl_for_renderer_scheduler_unittest.cc

Issue 1058873010: Move blink scheduler implementation into a component (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: updates Created 5 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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 "content/renderer/scheduler/webthread_impl_for_renderer_scheduler.h" 5 #include "components/scheduler/renderer/webthread_impl_for_renderer_scheduler.h"
6 6
7 #include "base/run_loop.h" 7 #include "base/run_loop.h"
8 #include "content/child/scheduler/scheduler_message_loop_delegate.h" 8 #include "components/scheduler/child/scheduler_message_loop_delegate.h"
9 #include "content/renderer/scheduler/renderer_scheduler_impl.h" 9 #include "components/scheduler/renderer/renderer_scheduler_impl.h"
10 #include "testing/gmock/include/gmock/gmock.h" 10 #include "testing/gmock/include/gmock/gmock.h"
11 #include "testing/gtest/include/gtest/gtest.h" 11 #include "testing/gtest/include/gtest/gtest.h"
12 #include "third_party/WebKit/public/platform/WebTraceLocation.h" 12 #include "third_party/WebKit/public/platform/WebTraceLocation.h"
13 13
14 namespace content { 14 namespace scheduler {
15 namespace { 15 namespace {
16 16
17 const int kWorkBatchSize = 2; 17 const int kWorkBatchSize = 2;
18 18
19 class MockTask : public blink::WebThread::Task { 19 class MockTask : public blink::WebThread::Task {
20 public: 20 public:
21 MOCK_METHOD0(run, void()); 21 MOCK_METHOD0(run, void());
22 }; 22 };
23 23
24 class MockTaskObserver : public blink::WebThread::TaskObserver { 24 class MockTaskObserver : public blink::WebThread::TaskObserver {
25 public: 25 public:
26 MOCK_METHOD0(willProcessTask, void()); 26 MOCK_METHOD0(willProcessTask, void());
27 MOCK_METHOD0(didProcessTask, void()); 27 MOCK_METHOD0(didProcessTask, void());
28 }; 28 };
29 } // namespace 29 } // namespace
30 30
31 class WebThreadImplForRendererSchedulerTest : public testing::Test { 31 class WebThreadImplForRendererSchedulerTest : public testing::Test {
32 public: 32 public:
33 WebThreadImplForRendererSchedulerTest() 33 WebThreadImplForRendererSchedulerTest()
34 : scheduler_(SchedulerMessageLoopDelegate::Create(&message_loop_)), 34 : scheduler_(SchedulerMessageLoopDelegate::Create(&message_loop_)),
35 default_task_runner_(scheduler_.DefaultTaskRunner()), 35 default_task_runner_(scheduler_.DefaultTaskRunner()),
36 thread_(&scheduler_) {} 36 thread_(&scheduler_) {}
37 37
38 ~WebThreadImplForRendererSchedulerTest() override {} 38 ~WebThreadImplForRendererSchedulerTest() override {}
39 39
40 void SetWorkBatchSizeForTesting(size_t work_batch_size) { 40 void SetWorkBatchSizeForTesting(size_t work_batch_size) {
41 scheduler_.SetWorkBatchSizeForTesting(work_batch_size); 41 scheduler_.GetSchedulerHelperForTesting()->SetWorkBatchSizeForTesting(
42 work_batch_size);
42 } 43 }
43 44
44 protected: 45 protected:
45 void EatDefaultTask(MockTaskObserver* observer) { 46 void EatDefaultTask(MockTaskObserver* observer) {
46 // The scheduler posts one extra DoWork() task automatically. 47 // The scheduler posts one extra DoWork() task automatically.
47 EXPECT_CALL(*observer, willProcessTask()); 48 EXPECT_CALL(*observer, willProcessTask());
48 EXPECT_CALL(*observer, didProcessTask()); 49 EXPECT_CALL(*observer, didProcessTask());
49 } 50 }
50 51
51 base::MessageLoop message_loop_; 52 base::MessageLoop message_loop_;
(...skipping 21 matching lines...) Expand all
73 thread_.postTask(blink::WebTraceLocation(), task.release()); 74 thread_.postTask(blink::WebTraceLocation(), task.release());
74 message_loop_.RunUntilIdle(); 75 message_loop_.RunUntilIdle();
75 thread_.removeTaskObserver(&observer); 76 thread_.removeTaskObserver(&observer);
76 } 77 }
77 78
78 TEST_F(WebThreadImplForRendererSchedulerTest, TestWorkBatchWithOneTask) { 79 TEST_F(WebThreadImplForRendererSchedulerTest, TestWorkBatchWithOneTask) {
79 MockTaskObserver observer; 80 MockTaskObserver observer;
80 thread_.addTaskObserver(&observer); 81 thread_.addTaskObserver(&observer);
81 scoped_ptr<MockTask> task(new MockTask()); 82 scoped_ptr<MockTask> task(new MockTask());
82 83
83 scheduler_.SetWorkBatchSizeForTesting(kWorkBatchSize); 84 SetWorkBatchSizeForTesting(kWorkBatchSize);
84 { 85 {
85 testing::InSequence sequence; 86 testing::InSequence sequence;
86 EXPECT_CALL(observer, willProcessTask()); 87 EXPECT_CALL(observer, willProcessTask());
87 EXPECT_CALL(*task, run()); 88 EXPECT_CALL(*task, run());
88 EXPECT_CALL(observer, didProcessTask()); 89 EXPECT_CALL(observer, didProcessTask());
89 90
90 EatDefaultTask(&observer); 91 EatDefaultTask(&observer);
91 } 92 }
92 93
93 thread_.postTask(blink::WebTraceLocation(), task.release()); 94 thread_.postTask(blink::WebTraceLocation(), task.release());
94 message_loop_.RunUntilIdle(); 95 message_loop_.RunUntilIdle();
95 thread_.removeTaskObserver(&observer); 96 thread_.removeTaskObserver(&observer);
96 } 97 }
97 98
98 TEST_F(WebThreadImplForRendererSchedulerTest, TestWorkBatchWithTwoTasks) { 99 TEST_F(WebThreadImplForRendererSchedulerTest, TestWorkBatchWithTwoTasks) {
99 MockTaskObserver observer; 100 MockTaskObserver observer;
100 thread_.addTaskObserver(&observer); 101 thread_.addTaskObserver(&observer);
101 scoped_ptr<MockTask> task1(new MockTask()); 102 scoped_ptr<MockTask> task1(new MockTask());
102 scoped_ptr<MockTask> task2(new MockTask()); 103 scoped_ptr<MockTask> task2(new MockTask());
103 104
104 scheduler_.SetWorkBatchSizeForTesting(kWorkBatchSize); 105 SetWorkBatchSizeForTesting(kWorkBatchSize);
105 { 106 {
106 testing::InSequence sequence; 107 testing::InSequence sequence;
107 EXPECT_CALL(observer, willProcessTask()); 108 EXPECT_CALL(observer, willProcessTask());
108 EXPECT_CALL(*task1, run()); 109 EXPECT_CALL(*task1, run());
109 EXPECT_CALL(observer, didProcessTask()); 110 EXPECT_CALL(observer, didProcessTask());
110 111
111 EXPECT_CALL(observer, willProcessTask()); 112 EXPECT_CALL(observer, willProcessTask());
112 EXPECT_CALL(*task2, run()); 113 EXPECT_CALL(*task2, run());
113 EXPECT_CALL(observer, didProcessTask()); 114 EXPECT_CALL(observer, didProcessTask());
114 115
115 EatDefaultTask(&observer); 116 EatDefaultTask(&observer);
116 } 117 }
117 118
118 thread_.postTask(blink::WebTraceLocation(), task1.release()); 119 thread_.postTask(blink::WebTraceLocation(), task1.release());
119 thread_.postTask(blink::WebTraceLocation(), task2.release()); 120 thread_.postTask(blink::WebTraceLocation(), task2.release());
120 message_loop_.RunUntilIdle(); 121 message_loop_.RunUntilIdle();
121 thread_.removeTaskObserver(&observer); 122 thread_.removeTaskObserver(&observer);
122 } 123 }
123 124
124 TEST_F(WebThreadImplForRendererSchedulerTest, TestWorkBatchWithThreeTasks) { 125 TEST_F(WebThreadImplForRendererSchedulerTest, TestWorkBatchWithThreeTasks) {
125 MockTaskObserver observer; 126 MockTaskObserver observer;
126 thread_.addTaskObserver(&observer); 127 thread_.addTaskObserver(&observer);
127 scoped_ptr<MockTask> task1(new MockTask()); 128 scoped_ptr<MockTask> task1(new MockTask());
128 scoped_ptr<MockTask> task2(new MockTask()); 129 scoped_ptr<MockTask> task2(new MockTask());
129 scoped_ptr<MockTask> task3(new MockTask()); 130 scoped_ptr<MockTask> task3(new MockTask());
130 131
131 scheduler_.SetWorkBatchSizeForTesting(kWorkBatchSize); 132 SetWorkBatchSizeForTesting(kWorkBatchSize);
132 { 133 {
133 testing::InSequence sequence; 134 testing::InSequence sequence;
134 EXPECT_CALL(observer, willProcessTask()); 135 EXPECT_CALL(observer, willProcessTask());
135 EXPECT_CALL(*task1, run()); 136 EXPECT_CALL(*task1, run());
136 EXPECT_CALL(observer, didProcessTask()); 137 EXPECT_CALL(observer, didProcessTask());
137 138
138 EXPECT_CALL(observer, willProcessTask()); 139 EXPECT_CALL(observer, willProcessTask());
139 EXPECT_CALL(*task2, run()); 140 EXPECT_CALL(*task2, run());
140 EXPECT_CALL(observer, didProcessTask()); 141 EXPECT_CALL(observer, didProcessTask());
141 142
(...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after
192 EatDefaultTask(&observer); 193 EatDefaultTask(&observer);
193 } 194 }
194 195
195 message_loop_.PostTask( 196 message_loop_.PostTask(
196 FROM_HERE, base::Bind(&EnterRunLoop, base::Unretained(&message_loop_), 197 FROM_HERE, base::Bind(&EnterRunLoop, base::Unretained(&message_loop_),
197 base::Unretained(&thread_))); 198 base::Unretained(&thread_)));
198 message_loop_.RunUntilIdle(); 199 message_loop_.RunUntilIdle();
199 thread_.removeTaskObserver(&observer); 200 thread_.removeTaskObserver(&observer);
200 } 201 }
201 202
202 } // namespace content 203 } // namespace scheduler
OLDNEW
« no previous file with comments | « components/scheduler/renderer/webthread_impl_for_renderer_scheduler.cc ('k') | components/scheduler/scheduler.gni » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698