sequenced_task_runner_handle_unittest.cc 3.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  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. #include "base/threading/sequenced_task_runner_handle.h"
  5. #include <memory>
  6. #include <utility>
  7. #include "base/bind.h"
  8. #include "base/callback.h"
  9. #include "base/location.h"
  10. #include "base/memory/ref_counted.h"
  11. #include "base/run_loop.h"
  12. #include "base/sequence_checker_impl.h"
  13. #include "base/synchronization/waitable_event.h"
  14. #include "base/task/sequenced_task_runner.h"
  15. #include "base/task/thread_pool.h"
  16. #include "base/test/task_environment.h"
  17. #include "base/test/test_simple_task_runner.h"
  18. #include "base/threading/thread_task_runner_handle.h"
  19. #include "testing/gtest/include/gtest/gtest.h"
  20. namespace base {
  21. namespace {
  22. class SequencedTaskRunnerHandleTest : public ::testing::Test {
  23. protected:
  24. // Verifies that the context it runs on has a SequencedTaskRunnerHandle
  25. // and that posting to it results in the posted task running in that same
  26. // context (sequence).
  27. static void VerifyCurrentSequencedTaskRunner() {
  28. ASSERT_TRUE(SequencedTaskRunnerHandle::IsSet());
  29. scoped_refptr<SequencedTaskRunner> task_runner =
  30. SequencedTaskRunnerHandle::Get();
  31. ASSERT_TRUE(task_runner);
  32. // Use SequenceCheckerImpl to make sure it's not a no-op in Release builds.
  33. std::unique_ptr<SequenceCheckerImpl> sequence_checker(
  34. new SequenceCheckerImpl);
  35. task_runner->PostTask(
  36. FROM_HERE,
  37. base::BindOnce(&SequencedTaskRunnerHandleTest::CheckValidSequence,
  38. std::move(sequence_checker)));
  39. }
  40. static void CheckValidSequence(
  41. std::unique_ptr<SequenceCheckerImpl> sequence_checker) {
  42. EXPECT_TRUE(sequence_checker->CalledOnValidSequence());
  43. }
  44. base::test::TaskEnvironment task_environment_;
  45. };
  46. TEST_F(SequencedTaskRunnerHandleTest, FromTaskEnvironment) {
  47. VerifyCurrentSequencedTaskRunner();
  48. RunLoop().RunUntilIdle();
  49. }
  50. TEST_F(SequencedTaskRunnerHandleTest, FromThreadPoolSequencedTask) {
  51. base::ThreadPool::CreateSequencedTaskRunner({})->PostTask(
  52. FROM_HERE,
  53. base::BindOnce(
  54. &SequencedTaskRunnerHandleTest::VerifyCurrentSequencedTaskRunner));
  55. task_environment_.RunUntilIdle();
  56. }
  57. TEST_F(SequencedTaskRunnerHandleTest, NoHandleFromUnsequencedTask) {
  58. base::ThreadPool::PostTask(base::BindOnce(
  59. []() { EXPECT_FALSE(SequencedTaskRunnerHandle::IsSet()); }));
  60. task_environment_.RunUntilIdle();
  61. }
  62. TEST(SequencedTaskRunnerHandleTestWithoutTaskEnvironment, FromHandleInScope) {
  63. scoped_refptr<SequencedTaskRunner> test_task_runner(new TestSimpleTaskRunner);
  64. EXPECT_FALSE(SequencedTaskRunnerHandle::IsSet());
  65. EXPECT_FALSE(ThreadTaskRunnerHandle::IsSet());
  66. {
  67. SequencedTaskRunnerHandle handle(test_task_runner);
  68. EXPECT_TRUE(SequencedTaskRunnerHandle::IsSet());
  69. EXPECT_FALSE(ThreadTaskRunnerHandle::IsSet());
  70. EXPECT_EQ(test_task_runner, SequencedTaskRunnerHandle::Get());
  71. }
  72. EXPECT_FALSE(SequencedTaskRunnerHandle::IsSet());
  73. EXPECT_FALSE(ThreadTaskRunnerHandle::IsSet());
  74. }
  75. } // namespace
  76. } // namespace base