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.
5 #include "base/sequenced_task_runner.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"
21 FlagOnDelete(bool* deleted,
22 scoped_refptr<SequencedTaskRunner> expected_deletion_sequence)
24 expected_deletion_sequence_(std::move(expected_deletion_sequence)) {}
27 friend class DeleteHelper<FlagOnDelete>;
28 FRIEND_TEST_ALL_PREFIXES(SequencedTaskRunnerTest,
29 OnTaskRunnerDeleterTargetStoppedEarly);
32 EXPECT_FALSE(*deleted_);
34 if (expected_deletion_sequence_)
35 EXPECT_TRUE(expected_deletion_sequence_->RunsTasksInCurrentSequence());
39 const scoped_refptr<SequencedTaskRunner> expected_deletion_sequence_;
41 DISALLOW_COPY_AND_ASSIGN(FlagOnDelete);
44 class SequencedTaskRunnerTest : public testing::Test {
46 SequencedTaskRunnerTest() : foreign_thread_("foreign") {}
48 void SetUp() override {
49 main_runner_ = message_loop_.task_runner();
51 foreign_thread_.Start();
52 foreign_runner_ = foreign_thread_.task_runner();
55 scoped_refptr<SequencedTaskRunner> main_runner_;
56 scoped_refptr<SequencedTaskRunner> foreign_runner_;
58 Thread foreign_thread_;
61 MessageLoop message_loop_;
63 DISALLOW_COPY_AND_ASSIGN(SequencedTaskRunnerTest);
66 using SequenceBoundUniquePtr =
67 std::unique_ptr<FlagOnDelete, OnTaskRunnerDeleter>;
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)));
80 foreign_runner_->PostTaskAndReply(FROM_HERE, BindOnce([] {}),
81 run_loop.QuitClosure());
84 EXPECT_TRUE(deleted_on_main_thread);
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);
93 // Stopping the target ahead of deleting |ptr| should make its
94 // OnTaskRunnerDeleter no-op.
95 foreign_thread_.Stop();
97 EXPECT_FALSE(deleted_on_main_thread);
100 EXPECT_TRUE(deleted_on_main_thread);