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
|
// 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.
#include "base/test/scoped_mock_time_message_loop_task_runner.h"
#include "base/check_op.h"
#include "base/functional/bind.h"
#include "base/run_loop.h"
#include "base/task/current_thread.h"
#include "base/task/single_thread_task_runner.h"
#include "base/test/test_pending_task.h"
#include "base/time/time.h"
namespace base {
ScopedMockTimeMessageLoopTaskRunner::ScopedMockTimeMessageLoopTaskRunner()
: task_runner_(new TestMockTimeTaskRunner),
previous_task_runner_(SingleThreadTaskRunner::GetCurrentDefault()) {
DCHECK(CurrentThread::Get());
// To ensure that we process any initialization tasks posted to the
// MessageLoop by a test fixture before replacing its TaskRunner.
RunLoop().RunUntilIdle();
CurrentThread::Get()->SetTaskRunner(task_runner_);
}
ScopedMockTimeMessageLoopTaskRunner::~ScopedMockTimeMessageLoopTaskRunner() {
DCHECK(previous_task_runner_->RunsTasksInCurrentSequence());
DCHECK_EQ(task_runner_, SingleThreadTaskRunner::GetCurrentDefault());
for (auto& pending_task : task_runner_->TakePendingTasks()) {
previous_task_runner_->PostDelayedTask(
pending_task.location, std::move(pending_task.task),
pending_task.GetTimeToRun() - task_runner_->NowTicks());
}
CurrentThread::Get()->SetTaskRunner(std::move(previous_task_runner_));
}
} // namespace base
|
