blob: 88e078b136a2e428ccc2a7aa89b01dce0c6bccf6 (
plain)
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
49
50
51
52
53
54
55
56
57
|
// Copyright 2015 The Weave Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <weave/provider/test/fake_task_runner.h>
namespace weave {
namespace provider {
namespace test {
class FakeTaskRunner::TestClock : public base::Clock {
public:
base::Time Now() override { return now_; }
void SetNow(base::Time now) { now_ = now; }
private:
base::Time now_{base::Time::Now()};
};
FakeTaskRunner::FakeTaskRunner() : test_clock_{new TestClock} {}
FakeTaskRunner::~FakeTaskRunner() {}
bool FakeTaskRunner::RunOnce() {
if (queue_.empty())
return false;
auto top = queue_.top();
queue_.pop();
test_clock_->SetNow(std::max(test_clock_->Now(), top.first.first));
top.second.Run();
return true;
}
void FakeTaskRunner::Run(size_t number_of_iterations) {
break_ = false;
for (size_t i = 0; i < number_of_iterations && !break_ && RunOnce(); ++i) {
}
}
void FakeTaskRunner::Break() {
break_ = true;
}
base::Clock* FakeTaskRunner::GetClock() {
return test_clock_.get();
}
void FakeTaskRunner::PostDelayedTask(const tracked_objects::Location& from_here,
const base::Closure& task,
base::TimeDelta delay) {
queue_.emplace(std::make_pair(test_clock_->Now() + delay, ++counter_), task);
}
} // namespace test
} // namespace provider
} // namespace weave
|
