scheduler: Move the Blink scheduler into Blink

media/blink/run_all_unittests.cc

 // Copyright 2014 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 <stddef.h>
 
 #include "base/bind.h"
 #include "base/message_loop/message_loop.h"
 #include "base/test/launcher/unit_test_launcher.h"
 #include "base/test/test_suite.h"
 #include "build/build_config.h"
-#include "components/scheduler/child/scheduler_tqm_delegate_impl.h"
-#include "components/scheduler/child/web_task_runner_impl.h"
-#include "components/scheduler/renderer/renderer_scheduler_impl.h"
-#include "components/scheduler/renderer/renderer_web_scheduler_impl.h"
-#include "components/scheduler/test/lazy_scheduler_message_loop_delegate_for_tests.h"
 #include "media/base/media.h"
 #include "testing/gtest/include/gtest/gtest.h"
-#include "third_party/WebKit/public/platform/WebScheduler.h"
-#include "third_party/WebKit/public/platform/WebTaskRunner.h"
-#include "third_party/WebKit/public/platform/WebThread.h"
+#include "third_party/WebKit/public/platform/scheduler/renderer/renderer_scheduler.h"
+#include "third_party/WebKit/public/platform/scheduler/test/renderer_scheduler_test_support.h"
 #include "third_party/WebKit/public/web/WebKit.h"
 
 #if defined(OS_ANDROID)
 #include "base/android/jni_android.h"
 #include "media/base/android/media_jni_registrar.h"
 #endif
 
 #ifdef V8_USE_EXTERNAL_STARTUP_DATA
 #include "gin/v8_initializer.h"
 #endif
 
-class CurrentThreadMock : public blink::WebThread {
+class TestBlinkPlatformSupport : NON_EXPORTED_BASE(public blink::Platform) {
  public:
-  CurrentThreadMock()
-      : task_runner_delegate_(
-            scheduler::LazySchedulerMessageLoopDelegateForTests::Create()),
-        scheduler_(
-            new scheduler::RendererSchedulerImpl(task_runner_delegate_.get())),
-        web_scheduler_(
-            new scheduler::RendererWebSchedulerImpl(scheduler_.get())),
-        web_task_runner_(
-            new scheduler::WebTaskRunnerImpl(scheduler_->DefaultTaskRunner())) {
-  }
+  TestBlinkPlatformSupport()
+      : renderer_scheduler_(
+            blink::scheduler::CreateRendererSchedulerForTests()),
+        main_thread_(renderer_scheduler_->CreateMainThread()) {}
+  ~TestBlinkPlatformSupport() override;
 
-  ~CurrentThreadMock() override {
-    scheduler_->Shutdown();
-  }
-
-  blink::WebTaskRunner* getWebTaskRunner() override {
-    return web_task_runner_.get();
-  }
-
-  bool isCurrentThread() const override { return true; }
-
-  blink::PlatformThreadId threadId() const override { return 17; }
-
-  blink::WebScheduler* scheduler() const override {
-    return web_scheduler_.get();
+  blink::WebThread* currentThread() override {
+    EXPECT_TRUE(main_thread_->isCurrentThread());
+    return main_thread_.get();
   }
 
  private:
-  scoped_refptr<scheduler::SchedulerTqmDelegate> task_runner_delegate_;
-  std::unique_ptr<scheduler::RendererSchedulerImpl> scheduler_;
-  std::unique_ptr<blink::WebScheduler> web_scheduler_;
-  std::unique_ptr<blink::WebTaskRunner> web_task_runner_;
+  std::unique_ptr<blink::scheduler::RendererScheduler> renderer_scheduler_;
+  std::unique_ptr<blink::WebThread> main_thread_;
 };
 
-class TestBlinkPlatformSupport : NON_EXPORTED_BASE(public blink::Platform) {
- public:
-  ~TestBlinkPlatformSupport() override;
-
-  blink::WebThread* currentThread() override { return &m_currentThread; }
-
- private:
-  CurrentThreadMock m_currentThread;
-};
-
-TestBlinkPlatformSupport::~TestBlinkPlatformSupport() {}
+TestBlinkPlatformSupport::~TestBlinkPlatformSupport() {
+  renderer_scheduler_->Shutdown();
+}
 
 class BlinkMediaTestSuite : public base::TestSuite {
  public:
   BlinkMediaTestSuite(int argc, char** argv);
   ~BlinkMediaTestSuite() override;
 
  protected:
   void Initialize() override;
 
  private:
@@ skipping 33 matching lines @@
   blink::initialize(blink_platform_support_.get());
 }
 
 int main(int argc, char** argv) {
   BlinkMediaTestSuite test_suite(argc, argv);
 
   return base::LaunchUnitTests(
       argc, argv, base::Bind(&BlinkMediaTestSuite::Run,
                              base::Unretained(&test_suite)));
 }