TaskScheduler: Add TaskRunnerHandle support to TaskScheduler tasks

base/threading/sequenced_task_runner_handle_unittest.cc

 // 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 "base/threading/sequenced_task_runner_handle.h"
 
 #include <memory>
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/location.h"
 #include "base/memory/ref_counted.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/sequence_checker_impl.h"
 #include "base/sequenced_task_runner.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/test/sequenced_worker_pool_owner.h"
+#include "base/test/test_simple_task_runner.h"
+#include "base/thread_task_runner_handle.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "base/threading/simple_thread.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace base {
 namespace {
 
 class SequencedTaskRunnerHandleTest : public ::testing::Test {
  protected:
   static void VerifyCurrentSequencedTaskRunner(const Closure& callback) {
@@ skipping 48 matching lines @@
   SequencedWorkerPoolOwner owner(3, "Test");
   WaitableEvent event(false, false);
   owner.pool()->PostWorkerTask(
       FROM_HERE,
       base::Bind(
           &SequencedTaskRunnerHandleTest::VerifyCurrentSequencedTaskRunner,
           base::Bind(&WaitableEvent::Signal, base::Unretained(&event))));
   event.Wait();
 }
 
-class ThreadRunner : public DelegateSimpleThread::Delegate {
- public:
-  void Run() override {
-    ASSERT_FALSE(SequencedTaskRunnerHandle::IsSet());
+TEST(SequencedTaskRunnerHandleTestWithoutMessageLoop, FromHandleInScope) {
+  scoped_refptr<SequencedTaskRunner> test_task_runner(new TestSimpleTaskRunner);
+  EXPECT_FALSE(SequencedTaskRunnerHandle::IsSet());
+  EXPECT_FALSE(ThreadTaskRunnerHandle::IsSet());
+  {
+    SequencedTaskRunnerHandle handle(test_task_runner);
+    EXPECT_TRUE(SequencedTaskRunnerHandle::IsSet());
+    EXPECT_FALSE(ThreadTaskRunnerHandle::IsSet());
+    EXPECT_EQ(test_task_runner, SequencedTaskRunnerHandle::Get());
   }
-
- private:
-  Closure callback_;
-};
-
-TEST_F(SequencedTaskRunnerHandleTest, FromSimpleThread) {
-  ThreadRunner thread_runner;
-  DelegateSimpleThread thread(&thread_runner, "Background thread");
-  thread.Start();
-  thread.Join();
+  EXPECT_FALSE(SequencedTaskRunnerHandle::IsSet());
+  EXPECT_FALSE(ThreadTaskRunnerHandle::IsSet());
 }
 
 }  // namespace
 }  // namespace base