| Index: components/scheduler/renderer/webthread_impl_for_renderer_scheduler_unittest.cc
|
| diff --git a/components/scheduler/renderer/webthread_impl_for_renderer_scheduler_unittest.cc b/components/scheduler/renderer/webthread_impl_for_renderer_scheduler_unittest.cc
|
| deleted file mode 100644
|
| index 1d29f244f2f98b024a1848927e0e8ae87136f314..0000000000000000000000000000000000000000
|
| --- a/components/scheduler/renderer/webthread_impl_for_renderer_scheduler_unittest.cc
|
| +++ /dev/null
|
| @@ -1,213 +0,0 @@
|
| -// 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 "components/scheduler/renderer/webthread_impl_for_renderer_scheduler.h"
|
| -
|
| -#include <stddef.h>
|
| -
|
| -#include "base/location.h"
|
| -#include "base/macros.h"
|
| -#include "base/memory/ptr_util.h"
|
| -#include "base/run_loop.h"
|
| -#include "base/single_thread_task_runner.h"
|
| -#include "base/test/simple_test_tick_clock.h"
|
| -#include "components/scheduler/base/test_time_source.h"
|
| -#include "components/scheduler/child/scheduler_tqm_delegate_impl.h"
|
| -#include "components/scheduler/renderer/renderer_scheduler_impl.h"
|
| -#include "testing/gmock/include/gmock/gmock.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -#include "third_party/WebKit/public/platform/WebTaskRunner.h"
|
| -#include "third_party/WebKit/public/platform/WebTraceLocation.h"
|
| -
|
| -namespace scheduler {
|
| -namespace {
|
| -
|
| -const int kWorkBatchSize = 2;
|
| -
|
| -class MockTask : public blink::WebTaskRunner::Task {
|
| - public:
|
| - MOCK_METHOD0(run, void());
|
| -};
|
| -
|
| -class MockTaskObserver : public blink::WebThread::TaskObserver {
|
| - public:
|
| - MOCK_METHOD0(willProcessTask, void());
|
| - MOCK_METHOD0(didProcessTask, void());
|
| -};
|
| -} // namespace
|
| -
|
| -class WebThreadImplForRendererSchedulerTest : public testing::Test {
|
| - public:
|
| - WebThreadImplForRendererSchedulerTest() {}
|
| -
|
| - void SetUp() override {
|
| - clock_.reset(new base::SimpleTestTickClock());
|
| - clock_->Advance(base::TimeDelta::FromMicroseconds(5000));
|
| - scheduler_.reset(new RendererSchedulerImpl(SchedulerTqmDelegateImpl::Create(
|
| - &message_loop_, base::WrapUnique(new TestTimeSource(clock_.get())))));
|
| - default_task_runner_ = scheduler_->DefaultTaskRunner();
|
| - thread_ = scheduler_->CreateMainThread();
|
| - }
|
| -
|
| - ~WebThreadImplForRendererSchedulerTest() override {}
|
| -
|
| - void SetWorkBatchSizeForTesting(size_t work_batch_size) {
|
| - scheduler_->GetSchedulerHelperForTesting()->SetWorkBatchSizeForTesting(
|
| - work_batch_size);
|
| - }
|
| -
|
| - void TearDown() override { scheduler_->Shutdown(); }
|
| -
|
| - protected:
|
| - base::MessageLoop message_loop_;
|
| - std::unique_ptr<base::SimpleTestTickClock> clock_;
|
| - std::unique_ptr<RendererSchedulerImpl> scheduler_;
|
| - scoped_refptr<base::SingleThreadTaskRunner> default_task_runner_;
|
| - std::unique_ptr<blink::WebThread> thread_;
|
| -
|
| - DISALLOW_COPY_AND_ASSIGN(WebThreadImplForRendererSchedulerTest);
|
| -};
|
| -
|
| -TEST_F(WebThreadImplForRendererSchedulerTest, TestTaskObserver) {
|
| - MockTaskObserver observer;
|
| - thread_->addTaskObserver(&observer);
|
| - std::unique_ptr<MockTask> task(new MockTask());
|
| -
|
| - {
|
| - testing::InSequence sequence;
|
| - EXPECT_CALL(observer, willProcessTask());
|
| - EXPECT_CALL(*task, run());
|
| - EXPECT_CALL(observer, didProcessTask());
|
| - }
|
| -
|
| - thread_->getWebTaskRunner()->postTask(blink::WebTraceLocation(),
|
| - task.release());
|
| - base::RunLoop().RunUntilIdle();
|
| - thread_->removeTaskObserver(&observer);
|
| -}
|
| -
|
| -TEST_F(WebThreadImplForRendererSchedulerTest, TestWorkBatchWithOneTask) {
|
| - MockTaskObserver observer;
|
| - thread_->addTaskObserver(&observer);
|
| - std::unique_ptr<MockTask> task(new MockTask());
|
| -
|
| - SetWorkBatchSizeForTesting(kWorkBatchSize);
|
| - {
|
| - testing::InSequence sequence;
|
| - EXPECT_CALL(observer, willProcessTask());
|
| - EXPECT_CALL(*task, run());
|
| - EXPECT_CALL(observer, didProcessTask());
|
| - }
|
| -
|
| - thread_->getWebTaskRunner()->postTask(blink::WebTraceLocation(),
|
| - task.release());
|
| - base::RunLoop().RunUntilIdle();
|
| - thread_->removeTaskObserver(&observer);
|
| -}
|
| -
|
| -TEST_F(WebThreadImplForRendererSchedulerTest, TestWorkBatchWithTwoTasks) {
|
| - MockTaskObserver observer;
|
| - thread_->addTaskObserver(&observer);
|
| - std::unique_ptr<MockTask> task1(new MockTask());
|
| - std::unique_ptr<MockTask> task2(new MockTask());
|
| -
|
| - 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());
|
| - }
|
| -
|
| - thread_->getWebTaskRunner()->postTask(blink::WebTraceLocation(),
|
| - task1.release());
|
| - thread_->getWebTaskRunner()->postTask(blink::WebTraceLocation(),
|
| - task2.release());
|
| - base::RunLoop().RunUntilIdle();
|
| - thread_->removeTaskObserver(&observer);
|
| -}
|
| -
|
| -TEST_F(WebThreadImplForRendererSchedulerTest, TestWorkBatchWithThreeTasks) {
|
| - MockTaskObserver observer;
|
| - thread_->addTaskObserver(&observer);
|
| - std::unique_ptr<MockTask> task1(new MockTask());
|
| - std::unique_ptr<MockTask> task2(new MockTask());
|
| - std::unique_ptr<MockTask> task3(new MockTask());
|
| -
|
| - 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());
|
| - }
|
| -
|
| - thread_->getWebTaskRunner()->postTask(blink::WebTraceLocation(),
|
| - task1.release());
|
| - thread_->getWebTaskRunner()->postTask(blink::WebTraceLocation(),
|
| - task2.release());
|
| - thread_->getWebTaskRunner()->postTask(blink::WebTraceLocation(),
|
| - task3.release());
|
| - base::RunLoop().RunUntilIdle();
|
| - thread_->removeTaskObserver(&observer);
|
| -}
|
| -
|
| -class ExitRunLoopTask : public blink::WebTaskRunner::Task {
|
| - public:
|
| - ExitRunLoopTask(base::RunLoop* run_loop) : run_loop_(run_loop) {}
|
| -
|
| - void run() override { run_loop_->Quit(); }
|
| -
|
| - private:
|
| - base::RunLoop* run_loop_;
|
| -};
|
| -
|
| -void EnterRunLoop(base::MessageLoop* message_loop, blink::WebThread* thread) {
|
| - // Note: WebThreads do not support nested run loops, which is why we use a
|
| - // run loop directly.
|
| - base::RunLoop run_loop;
|
| - thread->getWebTaskRunner()->postTask(blink::WebTraceLocation(),
|
| - new ExitRunLoopTask(&run_loop));
|
| - message_loop->SetNestableTasksAllowed(true);
|
| - run_loop.Run();
|
| -}
|
| -
|
| -TEST_F(WebThreadImplForRendererSchedulerTest, TestNestedRunLoop) {
|
| - MockTaskObserver observer;
|
| - thread_->addTaskObserver(&observer);
|
| -
|
| - {
|
| - testing::InSequence sequence;
|
| -
|
| - // One callback for EnterRunLoop.
|
| - EXPECT_CALL(observer, willProcessTask());
|
| -
|
| - // A pair for ExitRunLoopTask.
|
| - EXPECT_CALL(observer, willProcessTask());
|
| - EXPECT_CALL(observer, didProcessTask());
|
| -
|
| - // A final callback for EnterRunLoop.
|
| - EXPECT_CALL(observer, didProcessTask());
|
| - }
|
| -
|
| - message_loop_.task_runner()->PostTask(
|
| - FROM_HERE, base::Bind(&EnterRunLoop, base::Unretained(&message_loop_),
|
| - base::Unretained(thread_.get())));
|
| - base::RunLoop().RunUntilIdle();
|
| - thread_->removeTaskObserver(&observer);
|
| -}
|
| -
|
| -} // namespace scheduler
|
|
|