blob: 78977ad21dd750d7069f446d112daaaaa6ce04c1 (
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
|
// Copyright 2018 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.
#include "platform/test/fake_clock.h"
#include "platform/test/fake_task_runner.h"
#include "util/logging.h"
namespace openscreen {
namespace platform {
FakeClock::FakeClock(Clock::time_point start_time) {
now_ = start_time;
}
FakeClock::~FakeClock() {
OSP_CHECK(task_runners_.empty());
}
Clock::time_point FakeClock::now() noexcept {
return now_.load();
}
void FakeClock::Advance(Clock::duration delta) {
Clock::time_point x = now_.load();
while (!now_.compare_exchange_weak(x, x + delta)) {
x = now_.load();
}
for (FakeTaskRunner* task_runner : task_runners_) {
task_runner->RunTasksUntilIdle();
}
}
void FakeClock::SubscribeToTimeChanges(FakeTaskRunner* task_runner) {
OSP_CHECK(std::find(task_runners_.begin(), task_runners_.end(),
task_runner) == task_runners_.end());
task_runners_.push_back(task_runner);
}
void FakeClock::UnsubscribeFromTimeChanges(FakeTaskRunner* task_runner) {
auto it = std::find(task_runners_.begin(), task_runners_.end(), task_runner);
OSP_CHECK(it != task_runners_.end());
task_runners_.erase(it);
}
// static
std::atomic<Clock::time_point> FakeClock::now_{Clock::time_point{}};
} // namespace platform
} // namespace openscreen
|