deferred_sequenced_task_runner_unittest.cc 6.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215
  1. // Copyright (c) 2013 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. #include "base/task/deferred_sequenced_task_runner.h"
  5. #include "base/bind.h"
  6. #include "base/callback_helpers.h"
  7. #include "base/location.h"
  8. #include "base/memory/raw_ptr.h"
  9. #include "base/memory/ref_counted.h"
  10. #include "base/run_loop.h"
  11. #include "base/task/single_thread_task_runner.h"
  12. #include "base/test/task_environment.h"
  13. #include "base/threading/thread.h"
  14. #include "base/threading/thread_task_runner_handle.h"
  15. #include "testing/gmock/include/gmock/gmock.h"
  16. #include "testing/gtest/include/gtest/gtest.h"
  17. namespace base {
  18. namespace {
  19. class DeferredSequencedTaskRunnerTest : public testing::Test {
  20. public:
  21. class ExecuteTaskOnDestructor : public RefCounted<ExecuteTaskOnDestructor> {
  22. public:
  23. ExecuteTaskOnDestructor(
  24. DeferredSequencedTaskRunnerTest* executor,
  25. int task_id)
  26. : executor_(executor),
  27. task_id_(task_id) {
  28. }
  29. private:
  30. friend class RefCounted<ExecuteTaskOnDestructor>;
  31. virtual ~ExecuteTaskOnDestructor() { executor_->ExecuteTask(task_id_); }
  32. raw_ptr<DeferredSequencedTaskRunnerTest> executor_;
  33. int task_id_;
  34. };
  35. void ExecuteTask(int task_id) {
  36. AutoLock lock(lock_);
  37. executed_task_ids_.push_back(task_id);
  38. }
  39. void PostExecuteTask(int task_id) {
  40. runner_->PostTask(FROM_HERE,
  41. BindOnce(&DeferredSequencedTaskRunnerTest::ExecuteTask,
  42. Unretained(this), task_id));
  43. }
  44. void StartRunner() {
  45. runner_->Start();
  46. }
  47. void DoNothing(ExecuteTaskOnDestructor* object) {
  48. }
  49. protected:
  50. DeferredSequencedTaskRunnerTest()
  51. : runner_(
  52. new DeferredSequencedTaskRunner(ThreadTaskRunnerHandle::Get())) {}
  53. test::TaskEnvironment task_environment_;
  54. scoped_refptr<DeferredSequencedTaskRunner> runner_;
  55. mutable Lock lock_;
  56. std::vector<int> executed_task_ids_;
  57. };
  58. TEST_F(DeferredSequencedTaskRunnerTest, Stopped) {
  59. PostExecuteTask(1);
  60. RunLoop().RunUntilIdle();
  61. EXPECT_THAT(executed_task_ids_, testing::ElementsAre());
  62. }
  63. TEST_F(DeferredSequencedTaskRunnerTest, Start) {
  64. StartRunner();
  65. PostExecuteTask(1);
  66. RunLoop().RunUntilIdle();
  67. EXPECT_THAT(executed_task_ids_, testing::ElementsAre(1));
  68. }
  69. TEST_F(DeferredSequencedTaskRunnerTest, StartWithMultipleElements) {
  70. StartRunner();
  71. for (int i = 1; i < 5; ++i)
  72. PostExecuteTask(i);
  73. RunLoop().RunUntilIdle();
  74. EXPECT_THAT(executed_task_ids_, testing::ElementsAre(1, 2, 3, 4));
  75. }
  76. TEST_F(DeferredSequencedTaskRunnerTest, DeferredStart) {
  77. PostExecuteTask(1);
  78. RunLoop().RunUntilIdle();
  79. EXPECT_THAT(executed_task_ids_, testing::ElementsAre());
  80. StartRunner();
  81. RunLoop().RunUntilIdle();
  82. EXPECT_THAT(executed_task_ids_, testing::ElementsAre(1));
  83. PostExecuteTask(2);
  84. RunLoop().RunUntilIdle();
  85. EXPECT_THAT(executed_task_ids_, testing::ElementsAre(1, 2));
  86. }
  87. TEST_F(DeferredSequencedTaskRunnerTest, DeferredStartWithMultipleElements) {
  88. for (int i = 1; i < 5; ++i)
  89. PostExecuteTask(i);
  90. RunLoop().RunUntilIdle();
  91. EXPECT_THAT(executed_task_ids_, testing::ElementsAre());
  92. StartRunner();
  93. for (int i = 5; i < 9; ++i)
  94. PostExecuteTask(i);
  95. RunLoop().RunUntilIdle();
  96. EXPECT_THAT(executed_task_ids_, testing::ElementsAre(1, 2, 3, 4, 5, 6, 7, 8));
  97. }
  98. TEST_F(DeferredSequencedTaskRunnerTest, DeferredStartWithMultipleThreads) {
  99. {
  100. Thread thread1("DeferredSequencedTaskRunnerTestThread1");
  101. Thread thread2("DeferredSequencedTaskRunnerTestThread2");
  102. thread1.Start();
  103. thread2.Start();
  104. for (int i = 0; i < 5; ++i) {
  105. thread1.task_runner()->PostTask(
  106. FROM_HERE, BindOnce(&DeferredSequencedTaskRunnerTest::PostExecuteTask,
  107. Unretained(this), 2 * i));
  108. thread2.task_runner()->PostTask(
  109. FROM_HERE, BindOnce(&DeferredSequencedTaskRunnerTest::PostExecuteTask,
  110. Unretained(this), 2 * i + 1));
  111. if (i == 2) {
  112. thread1.task_runner()->PostTask(
  113. FROM_HERE, BindOnce(&DeferredSequencedTaskRunnerTest::StartRunner,
  114. Unretained(this)));
  115. }
  116. }
  117. }
  118. RunLoop().RunUntilIdle();
  119. EXPECT_THAT(executed_task_ids_,
  120. testing::WhenSorted(testing::ElementsAre(0, 1, 2, 3, 4, 5, 6, 7, 8, 9)));
  121. }
  122. TEST_F(DeferredSequencedTaskRunnerTest, ObjectDestructionOrder) {
  123. {
  124. Thread thread("DeferredSequencedTaskRunnerTestThread");
  125. thread.Start();
  126. runner_ = new DeferredSequencedTaskRunner(thread.task_runner());
  127. for (int i = 0; i < 5; ++i) {
  128. {
  129. // Use a block to ensure that no reference to |short_lived_object|
  130. // is kept on the main thread after it is posted to |runner_|.
  131. scoped_refptr<ExecuteTaskOnDestructor> short_lived_object =
  132. new ExecuteTaskOnDestructor(this, 2 * i);
  133. runner_->PostTask(
  134. FROM_HERE,
  135. BindOnce(&DeferredSequencedTaskRunnerTest::DoNothing,
  136. Unretained(this), RetainedRef(short_lived_object)));
  137. }
  138. // |short_lived_object| with id |2 * i| should be destroyed before the
  139. // task |2 * i + 1| is executed.
  140. PostExecuteTask(2 * i + 1);
  141. }
  142. StartRunner();
  143. }
  144. // All |short_lived_object| with id |2 * i| are destroyed before the task
  145. // |2 * i + 1| is executed.
  146. EXPECT_THAT(executed_task_ids_,
  147. testing::ElementsAre(0, 1, 2, 3, 4, 5, 6, 7, 8, 9));
  148. }
  149. void GetRunsTasksInCurrentSequence(bool* result,
  150. scoped_refptr<SequencedTaskRunner> runner,
  151. OnceClosure quit) {
  152. *result = runner->RunsTasksInCurrentSequence();
  153. std::move(quit).Run();
  154. }
  155. TEST_F(DeferredSequencedTaskRunnerTest, RunsTasksInCurrentSequence) {
  156. scoped_refptr<DeferredSequencedTaskRunner> runner =
  157. MakeRefCounted<DeferredSequencedTaskRunner>();
  158. EXPECT_TRUE(runner->RunsTasksInCurrentSequence());
  159. Thread thread1("DeferredSequencedTaskRunnerTestThread1");
  160. thread1.Start();
  161. bool runs_task_in_current_thread = true;
  162. base::RunLoop run_loop;
  163. thread1.task_runner()->PostTask(
  164. FROM_HERE,
  165. BindOnce(&GetRunsTasksInCurrentSequence, &runs_task_in_current_thread,
  166. runner, run_loop.QuitClosure()));
  167. run_loop.Run();
  168. EXPECT_FALSE(runs_task_in_current_thread);
  169. }
  170. TEST_F(DeferredSequencedTaskRunnerTest, StartWithTaskRunner) {
  171. scoped_refptr<DeferredSequencedTaskRunner> runner =
  172. MakeRefCounted<DeferredSequencedTaskRunner>();
  173. bool run_called = false;
  174. base::RunLoop run_loop;
  175. runner->PostTask(FROM_HERE,
  176. BindOnce(
  177. [](bool* run_called, base::OnceClosure quit_closure) {
  178. *run_called = true;
  179. std::move(quit_closure).Run();
  180. },
  181. &run_called, run_loop.QuitClosure()));
  182. runner->StartWithTaskRunner(ThreadTaskRunnerHandle::Get());
  183. run_loop.Run();
  184. EXPECT_TRUE(run_called);
  185. }
  186. } // namespace
  187. } // namespace base