| Index: base/test/scoped_mock_time_message_loop_task_runner.h
|
| diff --git a/base/test/scoped_mock_time_message_loop_task_runner.h b/base/test/scoped_mock_time_message_loop_task_runner.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..fe0d2c488ec9272ca3fd9ae3aaa6ee7facfa942e
|
| --- /dev/null
|
| +++ b/base/test/scoped_mock_time_message_loop_task_runner.h
|
| @@ -0,0 +1,41 @@
|
| +// Copyright 2015 The Chromium Authors. All rights reserved.
|
| +// 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/macros.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
|
| +// MessageLoop::current()'s task runner (and consequently
|
| +// ThreadTaskRunnerHandle) 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.
|
| +class ScopedMockTimeMessageLoopTaskRunner {
|
| + public:
|
| + ScopedMockTimeMessageLoopTaskRunner();
|
| + ~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_;
|
| +
|
| + DISALLOW_COPY_AND_ASSIGN(ScopedMockTimeMessageLoopTaskRunner);
|
| +};
|
| +
|
| +} // namespace base
|
| +
|
| +#endif // BASE_TEST_SCOPED_MOCK_TIME_MESSAGE_LOOP_TASK_RUNNER_H_
|
|
|