Task Scheduler.

base/task_scheduler/task_scheduler_impl_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/task_scheduler_impl.h"
+
+#include "base/bind.h"
+#include "base/callback_forward.h"
+#include "base/synchronization/condition_variable.h"
+#include "base/task_scheduler/scheduler_lock.h"
+#include "base/threading/platform_thread.h"
+#include "build/build_config.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace base {
+namespace task_scheduler {
+
+class TaskSchedulerImplTest : public testing::Test {
+ protected:
+  TaskSchedulerImplTest()
+      : cv_(lock_.RawLockForConditionVariable()),
+        num_posted_tasks_(0),
+        num_executed_tasks_(0),
+        ran_all_tasks_with_expected_priority_(true) {}
+
+  Closure GetTaskClosure(ThreadPriority expected_priority) {
+    ++num_posted_tasks_;
+    return Bind(&TaskSchedulerImplTest::RunTask, Unretained(this),
+                expected_priority);
+  }
+
+  void WaitUntilLastPostedTaskHasRun() {
+    AutoSchedulerLock auto_lock(lock_);
+    while (num_posted_tasks_ != num_executed_tasks_)
+      cv_.Wait();
+  }
+
+  void Shutdown() {
+    scheduler_.Shutdown();
+    scheduler_.JoinAllThreadsForTesting();
+  }
+
+  bool ran_all_tasks_with_expected_priority() const {
+    return ran_all_tasks_with_expected_priority_;
+  }
+
+  TaskSchedulerImpl scheduler_;
+
+ private:
+  void RunTask(ThreadPriority expected_priority) {
+#if !defined(OS_MACOSX)
+    // PlatformThread::GetCurrentThreadPriority() isn't implemented on Mac.
+    // crbug.com/554651
+    if (PlatformThread::GetCurrentThreadPriority() != expected_priority)
+      ran_all_tasks_with_expected_priority_ = false;
+#endif  // defined(OS_MACOSX)
+
+    AutoSchedulerLock auto_lock(lock_);
+    ++num_executed_tasks_;
+    cv_.Signal();
+  }
+
+  // Lock protecting |cv_|.
+  SchedulerLock lock_;
+
+  // Condition variable signaled each time a task completes its execution.
+  ConditionVariable cv_;
+
+  // Number of posted tasks.
+  size_t num_posted_tasks_;
+
+  // Number of tasks that were executed.
+  size_t num_executed_tasks_;
+
+  // True if all tasks ran on a thread with the expected priority.
+  bool ran_all_tasks_with_expected_priority_;
+};
+
+TEST_F(TaskSchedulerImplTest, PostTaskWithTraits) {
+  scheduler_.PostTaskWithTraits(
+      FROM_HERE, TaskTraits().WithPriority(TaskPriority::BACKGROUND),
+      GetTaskClosure(ThreadPriority::BACKGROUND));
+  scheduler_.PostTaskWithTraits(
+      FROM_HERE, TaskTraits().WithPriority(TaskPriority::USER_VISIBLE),
+      GetTaskClosure(ThreadPriority::NORMAL));
+  scheduler_.PostTaskWithTraits(
+      FROM_HERE, TaskTraits().WithPriority(TaskPriority::USER_BLOCKING),
+      GetTaskClosure(ThreadPriority::NORMAL));
+
+  scheduler_.PostTaskWithTraits(
+      FROM_HERE,
+      TaskTraits().WithPriority(TaskPriority::BACKGROUND).WithFileIO(),
+      GetTaskClosure(ThreadPriority::BACKGROUND));
+  scheduler_.PostTaskWithTraits(
+      FROM_HERE,
+      TaskTraits().WithPriority(TaskPriority::USER_VISIBLE).WithFileIO(),
+      GetTaskClosure(ThreadPriority::NORMAL));
+  scheduler_.PostTaskWithTraits(
+      FROM_HERE,
+      TaskTraits().WithPriority(TaskPriority::USER_BLOCKING).WithFileIO(),
+      GetTaskClosure(ThreadPriority::NORMAL));
+
+  WaitUntilLastPostedTaskHasRun();
+  Shutdown();
+
+  EXPECT_TRUE(ran_all_tasks_with_expected_priority());
+}
+
+TEST_F(TaskSchedulerImplTest, CreateTaskRunnerWithTraits) {
+  scheduler_.CreateTaskRunnerWithTraits(
+                TaskTraits().WithPriority(TaskPriority::BACKGROUND),
+                ExecutionMode::PARALLEL)
+      ->PostTask(FROM_HERE, GetTaskClosure(ThreadPriority::BACKGROUND));
+  scheduler_.CreateTaskRunnerWithTraits(
+                TaskTraits().WithPriority(TaskPriority::USER_VISIBLE),
+                ExecutionMode::PARALLEL)
+      ->PostTask(FROM_HERE, GetTaskClosure(ThreadPriority::NORMAL));
+  scheduler_.CreateTaskRunnerWithTraits(
+                TaskTraits().WithPriority(TaskPriority::USER_BLOCKING),
+                ExecutionMode::PARALLEL)
+      ->PostTask(FROM_HERE, GetTaskClosure(ThreadPriority::NORMAL));
+
+  scheduler_
+      .CreateTaskRunnerWithTraits(
+          TaskTraits().WithPriority(TaskPriority::BACKGROUND).WithFileIO(),
+          ExecutionMode::PARALLEL)
+      ->PostTask(FROM_HERE, GetTaskClosure(ThreadPriority::BACKGROUND));
+  scheduler_
+      .CreateTaskRunnerWithTraits(
+          TaskTraits().WithPriority(TaskPriority::USER_VISIBLE).WithFileIO(),
+          ExecutionMode::PARALLEL)
+      ->PostTask(FROM_HERE, GetTaskClosure(ThreadPriority::NORMAL));
+  scheduler_
+      .CreateTaskRunnerWithTraits(
+          TaskTraits().WithPriority(TaskPriority::USER_BLOCKING).WithFileIO(),
+          ExecutionMode::PARALLEL)
+      ->PostTask(FROM_HERE, GetTaskClosure(ThreadPriority::NORMAL));
+
+  WaitUntilLastPostedTaskHasRun();
+  Shutdown();
+
+  EXPECT_TRUE(ran_all_tasks_with_expected_priority());
+}
+
+}  // namespace task_scheduler
+}  // namespace base