1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef BASE_TEST_SCOPED_MOCK_TIME_MESSAGE_LOOP_TASK_RUNNER_H_
#define BASE_TEST_SCOPED_MOCK_TIME_MESSAGE_LOOP_TASK_RUNNER_H_
#include "base/memory/ref_counted.h"
#include "base/test/test_mock_time_task_runner.h"
namespace base {
class SingleThreadTaskRunner;
// A scoped wrapper around TestMockTimeTaskRunner that replaces
// CurrentThread::Get()'s task runner (and consequently
// SingleThreadTaskRunner::GetCurrentDefault) with a TestMockTimeTaskRunner and
// resets it back at the end of its scope.
//
// Note: RunLoop() will not work in the scope of a
// ScopedMockTimeMessageLoopTaskRunner, the underlying TestMockTimeTaskRunner's
// methods must be used instead to pump tasks.
//
// Note: Use TaskEnvironment + TimeSource::MOCK_TIME instead of this in unit
// tests. In browser tests you unfortunately still need this at the moment to
// mock delayed tasks on the main thread...
class ScopedMockTimeMessageLoopTaskRunner {
public:
ScopedMockTimeMessageLoopTaskRunner();
ScopedMockTimeMessageLoopTaskRunner(
const ScopedMockTimeMessageLoopTaskRunner&) = delete;
ScopedMockTimeMessageLoopTaskRunner& operator=(
const ScopedMockTimeMessageLoopTaskRunner&) = delete;
~ScopedMockTimeMessageLoopTaskRunner();
TestMockTimeTaskRunner* task_runner() { return task_runner_.get(); }
TestMockTimeTaskRunner* operator->() { return task_runner_.get(); }
private:
const scoped_refptr<TestMockTimeTaskRunner> task_runner_;
scoped_refptr<SingleThreadTaskRunner> previous_task_runner_;
};
} // namespace base
#endif // BASE_TEST_SCOPED_MOCK_TIME_MESSAGE_LOOP_TASK_RUNNER_H_
|
