TaskScheduler [9] Delayed Task Manager

base/task_scheduler/scheduler_thread_pool_unittest.cc

 // Copyright 2016 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 "base/task_scheduler/scheduler_thread_pool.h"
 
 #include <stddef.h>
 
 #include <memory>
 #include <unordered_set>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/memory/ref_counted.h"
 #include "base/synchronization/condition_variable.h"
 #include "base/synchronization/lock.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/task_runner.h"
+#include "base/task_scheduler/delayed_task_manager.h"
 #include "base/task_scheduler/sequence.h"
 #include "base/task_scheduler/sequence_sort_key.h"
 #include "base/task_scheduler/task_tracker.h"
 #include "base/threading/platform_thread.h"
 #include "base/threading/simple_thread.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace base {
 namespace internal {
 namespace {
 
 const size_t kNumThreadsInThreadPool = 4;
 const size_t kNumThreadsPostingTasks = 4;
 const size_t kNumTasksPostedPerThread = 150;
 
 class TaskSchedulerThreadPoolTest
     : public testing::TestWithParam<ExecutionMode> {
  protected:
-  TaskSchedulerThreadPoolTest() = default;
+  TaskSchedulerThreadPoolTest() : delayed_task_manager_(Bind(&DoNothing)) {}
 
   void SetUp() override {
     thread_pool_ = SchedulerThreadPool::CreateThreadPool(
         ThreadPriority::NORMAL, kNumThreadsInThreadPool,
         Bind(&TaskSchedulerThreadPoolTest::EnqueueSequenceCallback,
              Unretained(this)),
-        &task_tracker_);
+        &task_tracker_, &delayed_task_manager_);
     ASSERT_TRUE(thread_pool_);
   }
 
   void TearDown() override {
     thread_pool_->WaitForAllWorkerThreadsIdleForTesting();
     thread_pool_->JoinForTesting();
   }
 
   std::unique_ptr<SchedulerThreadPool> thread_pool_;
 
  private:
   void EnqueueSequenceCallback(scoped_refptr<Sequence> sequence) {
     // In production code, this callback would be implemented by the
     // TaskScheduler which would first determine which PriorityQueue the
     // sequence must be reinserted.
     const SequenceSortKey sort_key(sequence->GetSortKey());
     thread_pool_->EnqueueSequence(std::move(sequence), sort_key);
   }
 
   TaskTracker task_tracker_;
+  DelayedTaskManager delayed_task_manager_;
 
   DISALLOW_COPY_AND_ASSIGN(TaskSchedulerThreadPoolTest);
 };
 
 class TaskFactory {
  public:
   // Constructs a TaskFactory that posts tasks with |execution_mode| to
   // |thread_pool|.
   TaskFactory(SchedulerThreadPool* thread_pool, ExecutionMode execution_mode)
       : cv_(&lock_),
@@ skipping 248 matching lines @@
 INSTANTIATE_TEST_CASE_P(Parallel,
                         TaskSchedulerThreadPoolTest,
                         ::testing::Values(ExecutionMode::PARALLEL));
 INSTANTIATE_TEST_CASE_P(Sequenced,
                         TaskSchedulerThreadPoolTest,
                         ::testing::Values(ExecutionMode::SEQUENCED));
 
 }  // namespace
 }  // namespace internal
 }  // namespace base