1 // Copyright 2016 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/sequenced_task_runner.h"
6
7 #include <utility>
8
9 #include "base/bind.h"
10 #include "base/gtest_prod_util.h"
11 #include "base/message_loop/message_loop.h"
12 #include "base/run_loop.h"
13 #include "base/threading/thread.h"
14 #include "testing/gtest/include/gtest/gtest.h"
15
16 namespace base {
17 namespace {
18
19 class FlagOnDelete {
20 public:
FlagOnDelete(bool * deleted,scoped_refptr<SequencedTaskRunner> expected_deletion_sequence)21 FlagOnDelete(bool* deleted,
22 scoped_refptr<SequencedTaskRunner> expected_deletion_sequence)
23 : deleted_(deleted),
24 expected_deletion_sequence_(std::move(expected_deletion_sequence)) {}
25
26 private:
27 friend class DeleteHelper<FlagOnDelete>;
28 FRIEND_TEST_ALL_PREFIXES(SequencedTaskRunnerTest,
29 OnTaskRunnerDeleterTargetStoppedEarly);
30
~FlagOnDelete()31 ~FlagOnDelete() {
32 EXPECT_FALSE(*deleted_);
33 *deleted_ = true;
34 if (expected_deletion_sequence_)
35 EXPECT_TRUE(expected_deletion_sequence_->RunsTasksInCurrentSequence());
36 }
37
38 bool* deleted_;
39 const scoped_refptr<SequencedTaskRunner> expected_deletion_sequence_;
40
41 DISALLOW_COPY_AND_ASSIGN(FlagOnDelete);
42 };
43
44 class SequencedTaskRunnerTest : public testing::Test {
45 protected:
SequencedTaskRunnerTest()46 SequencedTaskRunnerTest() : foreign_thread_("foreign") {}
47
SetUp()48 void SetUp() override {
49 main_runner_ = message_loop_.task_runner();
50
51 foreign_thread_.Start();
52 foreign_runner_ = foreign_thread_.task_runner();
53 }
54
55 scoped_refptr<SequencedTaskRunner> main_runner_;
56 scoped_refptr<SequencedTaskRunner> foreign_runner_;
57
58 Thread foreign_thread_;
59
60 private:
61 MessageLoop message_loop_;
62
63 DISALLOW_COPY_AND_ASSIGN(SequencedTaskRunnerTest);
64 };
65
66 using SequenceBoundUniquePtr =
67 std::unique_ptr<FlagOnDelete, OnTaskRunnerDeleter>;
68
TEST_F(SequencedTaskRunnerTest,OnTaskRunnerDeleterOnMainThread)69 TEST_F(SequencedTaskRunnerTest, OnTaskRunnerDeleterOnMainThread) {
70 bool deleted_on_main_thread = false;
71 SequenceBoundUniquePtr ptr(
72 new FlagOnDelete(&deleted_on_main_thread, main_runner_),
73 OnTaskRunnerDeleter(main_runner_));
74 EXPECT_FALSE(deleted_on_main_thread);
75 foreign_runner_->PostTask(
76 FROM_HERE, BindOnce([](SequenceBoundUniquePtr) {}, std::move(ptr)));
77
78 {
79 RunLoop run_loop;
80 foreign_runner_->PostTaskAndReply(FROM_HERE, BindOnce([] {}),
81 run_loop.QuitClosure());
82 run_loop.Run();
83 }
84 EXPECT_TRUE(deleted_on_main_thread);
85 }
86
TEST_F(SequencedTaskRunnerTest,OnTaskRunnerDeleterTargetStoppedEarly)87 TEST_F(SequencedTaskRunnerTest, OnTaskRunnerDeleterTargetStoppedEarly) {
88 bool deleted_on_main_thread = false;
89 FlagOnDelete* raw = new FlagOnDelete(&deleted_on_main_thread, main_runner_);
90 SequenceBoundUniquePtr ptr(raw, OnTaskRunnerDeleter(foreign_runner_));
91 EXPECT_FALSE(deleted_on_main_thread);
92
93 // Stopping the target ahead of deleting |ptr| should make its
94 // OnTaskRunnerDeleter no-op.
95 foreign_thread_.Stop();
96 ptr = nullptr;
97 EXPECT_FALSE(deleted_on_main_thread);
98
99 delete raw;
100 EXPECT_TRUE(deleted_on_main_thread);
101 }
102
103 } // namespace
104 } // namespace base
105