scheduler: Add a base directory

components/scheduler/base/task_queue_manager_unittest.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 "components/scheduler/child/task_queue_manager.h"
+#include "components/scheduler/base/task_queue_manager.h"
 
 #include "base/location.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/test/simple_test_tick_clock.h"
 #include "base/threading/thread.h"
 #include "cc/test/ordered_simple_task_runner.h"
-#include "components/scheduler/child/nestable_task_runner_for_test.h"
-#include "components/scheduler/child/scheduler_task_runner_delegate_impl.h"
-#include "components/scheduler/child/task_queue_impl.h"
-#include "components/scheduler/child/task_queue_selector.h"
-#include "components/scheduler/child/task_queue_sets.h"
-#include "components/scheduler/child/test_time_source.h"
-#include "components/scheduler/test/test_always_fail_time_source.h"
+#include "components/scheduler/base/nestable_task_runner_for_test.h"
+#include "components/scheduler/base/task_queue_impl.h"
+#include "components/scheduler/base/task_queue_selector.h"
+#include "components/scheduler/base/task_queue_sets.h"
+#include "components/scheduler/base/test_always_fail_time_source.h"
+#include "components/scheduler/base/test_time_source.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 using testing::ElementsAre;
 using testing::_;
 
 namespace scheduler {
 
+class MessageLoopTaskRunner : public NestableTaskRunnerForTest {
+ public:
+  static scoped_refptr<MessageLoopTaskRunner> Create() {
+    return make_scoped_refptr(new MessageLoopTaskRunner());
+  }
+
+  // NestableTaskRunner implementation.
+  bool IsNested() const override {
+    return base::MessageLoop::current()->IsNested();
+  }
+
+ private:
+  MessageLoopTaskRunner()
+      : NestableTaskRunnerForTest(base::MessageLoop::current()->task_runner()) {
+  }
+  ~MessageLoopTaskRunner() override {}
+};
+
 class TaskQueueManagerTest : public testing::Test {
  public:
   void DeleteTaskQueueManager() { manager_.reset(); }
 
  protected:
   void Initialize(size_t num_queues) {
     now_src_.reset(new base::SimpleTestTickClock());
     now_src_->Advance(base::TimeDelta::FromMicroseconds(1000));
     test_task_runner_ = make_scoped_refptr(
         new cc::OrderedSimpleTaskRunner(now_src_.get(), false));
     main_task_runner_ =
         NestableTaskRunnerForTest::Create(test_task_runner_.get());
     manager_ = make_scoped_ptr(new TaskQueueManager(
         main_task_runner_, "test.scheduler", "test.scheduler.debug"));
     manager_->SetTimeSourceForTesting(
         make_scoped_ptr(new TestTimeSource(now_src_.get())));
 
     for (size_t i = 0; i < num_queues; i++)
       runners_.push_back(manager_->NewTaskQueue(TaskQueue::Spec("test_queue")));
   }
 
   void InitializeWithRealMessageLoop(size_t num_queues) {
     message_loop_.reset(new base::MessageLoop());
-    manager_ = make_scoped_ptr(new TaskQueueManager(
-        SchedulerTaskRunnerDelegateImpl::Create(message_loop_.get()),
-        "test.scheduler", "test.scheduler.debug"));
+    manager_ = make_scoped_ptr(
+        new TaskQueueManager(MessageLoopTaskRunner::Create(), "test.scheduler",
+                             "test.scheduler.debug"));
 
     for (size_t i = 0; i < num_queues; i++)
       runners_.push_back(manager_->NewTaskQueue(TaskQueue::Spec("test_queue")));
   }
 
   scoped_ptr<base::MessageLoop> message_loop_;
   scoped_ptr<base::SimpleTestTickClock> now_src_;
   scoped_refptr<NestableTaskRunnerForTest> main_task_runner_;
   scoped_refptr<cc::OrderedSimpleTaskRunner> test_task_runner_;
   scoped_ptr<TaskQueueManager> manager_;
   std::vector<scoped_refptr<internal::TaskQueueImpl>> runners_;
 };
 
 void PostFromNestedRunloop(base::MessageLoop* message_loop,
                            base::SingleThreadTaskRunner* runner,
                            std::vector<std::pair<base::Closure, bool>>* tasks) {
   base::MessageLoop::ScopedNestableTaskAllower allow(message_loop);
   for (std::pair<base::Closure, bool>& pair : *tasks) {
     if (pair.second) {
       runner->PostTask(FROM_HERE, pair.first);
     } else {
       runner->PostNonNestableTask(FROM_HERE, pair.first);
     }
   }
   message_loop->RunUntilIdle();
 }
 
-void NullTask() {
-}
+void NullTask() {}
 
 void TestTask(int value, std::vector<int>* out_result) {
   out_result->push_back(value);
 }
 
 TEST_F(TaskQueueManagerTest, SingleQueuePosting) {
   Initialize(1u);
 
   std::vector<int> run_order;
   runners_[0]->PostTask(FROM_HERE, base::Bind(&TestTask, 1, &run_order));
@@ skipping 12 matching lines @@
   runners_[0]->PostTask(FROM_HERE, base::Bind(&TestTask, 2, &run_order));
   runners_[1]->PostTask(FROM_HERE, base::Bind(&TestTask, 3, &run_order));
   runners_[1]->PostTask(FROM_HERE, base::Bind(&TestTask, 4, &run_order));
   runners_[2]->PostTask(FROM_HERE, base::Bind(&TestTask, 5, &run_order));
   runners_[2]->PostTask(FROM_HERE, base::Bind(&TestTask, 6, &run_order));
 
   test_task_runner_->RunUntilIdle();
   EXPECT_THAT(run_order, ElementsAre(1, 2, 3, 4, 5, 6));
 }
 
-void NopTask() {
-}
+void NopTask() {}
 
 TEST_F(TaskQueueManagerTest, NowNotCalledWhenThereAreNoDelayedTasks) {
   Initialize(3u);
 
   manager_->SetTimeSourceForTesting(
       make_scoped_ptr(new TestAlwaysFailTimeSource()));
 
   runners_[0]->PostTask(FROM_HERE, base::Bind(&NopTask));
   runners_[0]->PostTask(FROM_HERE, base::Bind(&NopTask));
   runners_[1]->PostTask(FROM_HERE, base::Bind(&NopTask));
@@ skipping 1159 matching lines @@
   scoped_refptr<internal::TaskQueueImpl> task_queue =
       manager_->NewTaskQueue(TaskQueue::Spec("test_queue"));
 
   EXPECT_CALL(observer, OnUnregisterTaskQueue(_)).Times(1);
   task_queue->UnregisterTaskQueue();
 
   manager_->SetObserver(nullptr);
 }
 
 }  // namespace scheduler