scheduler: Implement task observers

content/renderer/scheduler/webthread_impl_for_scheduler_unittest.cc

Index: content/renderer/scheduler/webthread_impl_for_scheduler_unittest.cc
diff --git a/content/renderer/scheduler/webthread_impl_for_scheduler_unittest.cc b/content/renderer/scheduler/webthread_impl_for_scheduler_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..65f0695353326e382d538837030dfecb767b6426
--- /dev/null
+++ b/content/renderer/scheduler/webthread_impl_for_scheduler_unittest.cc
@@ -0,0 +1,149 @@
+// Copyright 2015 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "content/renderer/scheduler/webthread_impl_for_scheduler.h"
+
+#include "content/renderer/scheduler/renderer_scheduler_impl.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/WebKit/public/platform/WebTraceLocation.h"
+
+namespace content {
+namespace {
+const int kWorkBatchSize = 2;

[rmcilroy, 2015/02/19 11:47:49]

nit - newline above

[Sami, 2015/02/19 16:31:09]

Done.

+
+class MockTask : public blink::WebThread::Task {
+ public:
+  MOCK_METHOD0(run, void());
+};
+
+class MockTaskObserver : public blink::WebThread::TaskObserver {
+ public:
+  MOCK_METHOD0(willProcessTask, void());
+  MOCK_METHOD0(didProcessTask, void());
+};
+}  // namespace
+
+class WebThreadImplForSchedulerTest : public testing::Test {
+ public:
+  WebThreadImplForSchedulerTest()
+      : scheduler_(message_loop_.task_runner()),
+        default_task_runner_(scheduler_.DefaultTaskRunner()),
+        thread_(&scheduler_) {}
+
+  ~WebThreadImplForSchedulerTest() override {}
+
+ protected:
+  void EatDefaultTask(MockTaskObserver* observer) {
+    // The scheduler posts one extra DoWork() task automatically.
+    EXPECT_CALL(*observer, willProcessTask());
+    EXPECT_CALL(*observer, didProcessTask());
+  }
+
+  base::MessageLoop message_loop_;
+  RendererSchedulerImpl scheduler_;
+  scoped_refptr<base::SingleThreadTaskRunner> default_task_runner_;
+  WebThreadImplForScheduler thread_;
+
+  DISALLOW_COPY_AND_ASSIGN(WebThreadImplForSchedulerTest);
+};
+
+TEST_F(WebThreadImplForSchedulerTest, TestTaskObserver) {
+  MockTaskObserver observer;
+  thread_.addTaskObserver(&observer);
+  scoped_ptr<MockTask> task(new MockTask());
+
+  {
+    testing::InSequence sequence;
+    EXPECT_CALL(observer, willProcessTask());
+    EXPECT_CALL(*task, run());
+    EXPECT_CALL(observer, didProcessTask());
+
+    EatDefaultTask(&observer);
+  }
+
+  thread_.postTask(blink::WebTraceLocation(), task.release());
+  message_loop_.RunUntilIdle();
+  thread_.removeTaskObserver(&observer);
+}
+
+TEST_F(WebThreadImplForSchedulerTest, TestWorkBatchWithOneTask) {
+  MockTaskObserver observer;
+  thread_.addTaskObserver(&observer);
+  scoped_ptr<MockTask> task(new MockTask());
+
+  scheduler_.SetWorkBatchSizeForTesting(kWorkBatchSize);
+  {
+    testing::InSequence sequence;
+    EXPECT_CALL(observer, willProcessTask());
+    EXPECT_CALL(*task, run());
+    EXPECT_CALL(observer, didProcessTask());
+
+    EatDefaultTask(&observer);
+  }
+
+  thread_.postTask(blink::WebTraceLocation(), task.release());
+  message_loop_.RunUntilIdle();
+  thread_.removeTaskObserver(&observer);
+}
+
+TEST_F(WebThreadImplForSchedulerTest, TestWorkBatchWithTwoTasks) {
+  MockTaskObserver observer;
+  thread_.addTaskObserver(&observer);
+  scoped_ptr<MockTask> task1(new MockTask());
+  scoped_ptr<MockTask> task2(new MockTask());
+
+  scheduler_.SetWorkBatchSizeForTesting(kWorkBatchSize);
+  {
+    testing::InSequence sequence;
+    EXPECT_CALL(observer, willProcessTask());
+    EXPECT_CALL(*task1, run());
+    EXPECT_CALL(observer, didProcessTask());
+
+    EXPECT_CALL(observer, willProcessTask());
+    EXPECT_CALL(*task2, run());
+    EXPECT_CALL(observer, didProcessTask());
+
+    EatDefaultTask(&observer);
+  }
+
+  thread_.postTask(blink::WebTraceLocation(), task1.release());
+  thread_.postTask(blink::WebTraceLocation(), task2.release());
+  message_loop_.RunUntilIdle();
+  thread_.removeTaskObserver(&observer);
+}
+
+TEST_F(WebThreadImplForSchedulerTest, TestWorkBatchWithThreeTasks) {
+  MockTaskObserver observer;
+  thread_.addTaskObserver(&observer);
+  scoped_ptr<MockTask> task1(new MockTask());
+  scoped_ptr<MockTask> task2(new MockTask());
+  scoped_ptr<MockTask> task3(new MockTask());
+
+  scheduler_.SetWorkBatchSizeForTesting(kWorkBatchSize);
+  {
+    testing::InSequence sequence;
+    EXPECT_CALL(observer, willProcessTask());
+    EXPECT_CALL(*task1, run());
+    EXPECT_CALL(observer, didProcessTask());
+
+    EXPECT_CALL(observer, willProcessTask());
+    EXPECT_CALL(*task2, run());
+    EXPECT_CALL(observer, didProcessTask());
+
+    EXPECT_CALL(observer, willProcessTask());
+    EXPECT_CALL(*task3, run());
+    EXPECT_CALL(observer, didProcessTask());
+
+    EatDefaultTask(&observer);
+  }
+
+  thread_.postTask(blink::WebTraceLocation(), task1.release());
+  thread_.postTask(blink::WebTraceLocation(), task2.release());
+  thread_.postTask(blink::WebTraceLocation(), task3.release());
+  message_loop_.RunUntilIdle();
+  thread_.removeTaskObserver(&observer);
+}
+
+}  // namespace content