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

Unified Diff: components/scheduler/renderer/webthread_impl_for_renderer_scheduler_unittest.cc

Issue 2118903002: scheduler: Move the Blink scheduler into Blink (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Rebased Created 4 years, 4 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 side-by-side diff with in-line comments
Download patch
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
« no previous file with comments | « components/scheduler/renderer/webthread_impl_for_renderer_scheduler.cc ('k') | components/scheduler/scheduler.gyp » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698