1// Copyright 2015 The Weave Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include <weave/provider/test/fake_task_runner.h>
6
7namespace weave {
8namespace provider {
9namespace test {
10
11class FakeTaskRunner::TestClock : public base::Clock {
12 public:
13  base::Time Now() override { return now_; }
14
15  void SetNow(base::Time now) { now_ = now; }
16
17 private:
18  base::Time now_{base::Time::Now()};
19};
20
21FakeTaskRunner::FakeTaskRunner() : test_clock_{new TestClock} {}
22
23FakeTaskRunner::~FakeTaskRunner() {}
24
25bool FakeTaskRunner::RunOnce() {
26  if (queue_.empty())
27    return false;
28  auto top = queue_.top();
29  queue_.pop();
30  test_clock_->SetNow(std::max(test_clock_->Now(), top.first.first));
31  top.second.Run();
32  return true;
33}
34
35void FakeTaskRunner::Run(size_t number_of_iterations) {
36  break_ = false;
37  for (size_t i = 0; i < number_of_iterations && !break_ && RunOnce(); ++i) {
38  }
39}
40
41void FakeTaskRunner::Break() {
42  break_ = true;
43}
44
45base::Clock* FakeTaskRunner::GetClock() {
46  return test_clock_.get();
47}
48
49void FakeTaskRunner::PostDelayedTask(const tracked_objects::Location& from_here,
50                                     const base::Closure& task,
51                                     base::TimeDelta delay) {
52  queue_.emplace(std::make_pair(test_clock_->Now() + delay, ++counter_), task);
53}
54
55size_t FakeTaskRunner::GetTaskQueueSize() const {
56  return queue_.size();
57}
58
59}  // namespace test
60}  // namespace provider
61}  // namespace weave
62