scoped_mock_time_message_loop_task_runner.h 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  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. #ifndef BASE_TEST_SCOPED_MOCK_TIME_MESSAGE_LOOP_TASK_RUNNER_H_
  5. #define BASE_TEST_SCOPED_MOCK_TIME_MESSAGE_LOOP_TASK_RUNNER_H_
  6. #include "base/memory/ref_counted.h"
  7. #include "base/test/test_mock_time_task_runner.h"
  8. namespace base {
  9. class SingleThreadTaskRunner;
  10. // A scoped wrapper around TestMockTimeTaskRunner that replaces
  11. // CurrentThread::Get()'s task runner (and consequently
  12. // ThreadTaskRunnerHandle) with a TestMockTimeTaskRunner and resets it back at
  13. // the end of its scope.
  14. //
  15. // Note: RunLoop() will not work in the scope of a
  16. // ScopedMockTimeMessageLoopTaskRunner, the underlying TestMockTimeTaskRunner's
  17. // methods must be used instead to pump tasks.
  18. //
  19. // Note: Use TaskEnvironment + TimeSource::MOCK_TIME instead of this in unit
  20. // tests. In browser tests you unfortunately still need this at the moment to
  21. // mock delayed tasks on the main thread...
  22. class ScopedMockTimeMessageLoopTaskRunner {
  23. public:
  24. ScopedMockTimeMessageLoopTaskRunner();
  25. ScopedMockTimeMessageLoopTaskRunner(
  26. const ScopedMockTimeMessageLoopTaskRunner&) = delete;
  27. ScopedMockTimeMessageLoopTaskRunner& operator=(
  28. const ScopedMockTimeMessageLoopTaskRunner&) = delete;
  29. ~ScopedMockTimeMessageLoopTaskRunner();
  30. TestMockTimeTaskRunner* task_runner() { return task_runner_.get(); }
  31. TestMockTimeTaskRunner* operator->() { return task_runner_.get(); }
  32. private:
  33. const scoped_refptr<TestMockTimeTaskRunner> task_runner_;
  34. scoped_refptr<SingleThreadTaskRunner> previous_task_runner_;
  35. };
  36. } // namespace base
  37. #endif // BASE_TEST_SCOPED_MOCK_TIME_MESSAGE_LOOP_TASK_RUNNER_H_