1 // Copyright 2015 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/threading/sequenced_task_runner_handle.h"
6
7 #include <memory>
8 #include <utility>
9
10 #include "base/bind.h"
11 #include "base/callback.h"
12 #include "base/location.h"
13 #include "base/memory/ref_counted.h"
14 #include "base/run_loop.h"
15 #include "base/sequence_checker_impl.h"
16 #include "base/sequenced_task_runner.h"
17 #include "base/synchronization/waitable_event.h"
18 #include "base/task_scheduler/post_task.h"
19 #include "base/test/scoped_task_environment.h"
20 #include "base/test/test_simple_task_runner.h"
21 #include "base/threading/thread_task_runner_handle.h"
22 #include "testing/gtest/include/gtest/gtest.h"
23
24 namespace base {
25 namespace {
26
27 class SequencedTaskRunnerHandleTest : public ::testing::Test {
28 protected:
29 // Verifies that the context it runs on has a SequencedTaskRunnerHandle
30 // and that posting to it results in the posted task running in that same
31 // context (sequence).
VerifyCurrentSequencedTaskRunner()32 static void VerifyCurrentSequencedTaskRunner() {
33 ASSERT_TRUE(SequencedTaskRunnerHandle::IsSet());
34 scoped_refptr<SequencedTaskRunner> task_runner =
35 SequencedTaskRunnerHandle::Get();
36 ASSERT_TRUE(task_runner);
37
38 // Use SequenceCheckerImpl to make sure it's not a no-op in Release builds.
39 std::unique_ptr<SequenceCheckerImpl> sequence_checker(
40 new SequenceCheckerImpl);
41 task_runner->PostTask(
42 FROM_HERE,
43 base::BindOnce(&SequencedTaskRunnerHandleTest::CheckValidSequence,
44 std::move(sequence_checker)));
45 }
46
CheckValidSequence(std::unique_ptr<SequenceCheckerImpl> sequence_checker)47 static void CheckValidSequence(
48 std::unique_ptr<SequenceCheckerImpl> sequence_checker) {
49 EXPECT_TRUE(sequence_checker->CalledOnValidSequence());
50 }
51
52 base::test::ScopedTaskEnvironment scoped_task_environment_;
53 };
54
TEST_F(SequencedTaskRunnerHandleTest,FromMessageLoop)55 TEST_F(SequencedTaskRunnerHandleTest, FromMessageLoop) {
56 VerifyCurrentSequencedTaskRunner();
57 RunLoop().RunUntilIdle();
58 }
59
TEST_F(SequencedTaskRunnerHandleTest,FromTaskSchedulerSequencedTask)60 TEST_F(SequencedTaskRunnerHandleTest, FromTaskSchedulerSequencedTask) {
61 base::CreateSequencedTaskRunnerWithTraits({})->PostTask(
62 FROM_HERE,
63 base::BindOnce(
64 &SequencedTaskRunnerHandleTest::VerifyCurrentSequencedTaskRunner));
65 scoped_task_environment_.RunUntilIdle();
66 }
67
TEST_F(SequencedTaskRunnerHandleTest,NoHandleFromUnsequencedTask)68 TEST_F(SequencedTaskRunnerHandleTest, NoHandleFromUnsequencedTask) {
69 base::PostTask(FROM_HERE, base::BindOnce([]() {
70 EXPECT_FALSE(SequencedTaskRunnerHandle::IsSet());
71 }));
72 scoped_task_environment_.RunUntilIdle();
73 }
74
TEST(SequencedTaskRunnerHandleTestWithoutMessageLoop,FromHandleInScope)75 TEST(SequencedTaskRunnerHandleTestWithoutMessageLoop, FromHandleInScope) {
76 scoped_refptr<SequencedTaskRunner> test_task_runner(new TestSimpleTaskRunner);
77 EXPECT_FALSE(SequencedTaskRunnerHandle::IsSet());
78 EXPECT_FALSE(ThreadTaskRunnerHandle::IsSet());
79 {
80 SequencedTaskRunnerHandle handle(test_task_runner);
81 EXPECT_TRUE(SequencedTaskRunnerHandle::IsSet());
82 EXPECT_FALSE(ThreadTaskRunnerHandle::IsSet());
83 EXPECT_EQ(test_task_runner, SequencedTaskRunnerHandle::Get());
84 }
85 EXPECT_FALSE(SequencedTaskRunnerHandle::IsSet());
86 EXPECT_FALSE(ThreadTaskRunnerHandle::IsSet());
87 }
88
89 } // namespace
90 } // namespace base
91