base/timer/mock_timer.cc


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
58
59
60
61
62
63

// Copyright 2014 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 "base/timer/mock_timer.h"

namespace base {

MockTimer::MockTimer(bool retain_user_task, bool is_repeating)
    : Timer(retain_user_task, is_repeating),
      is_running_(false) {
}

MockTimer::MockTimer(const tracked_objects::Location& posted_from,
                     TimeDelta delay,
                     const base::Closure& user_task,
                     bool is_repeating)
    : Timer(true, is_repeating),
      delay_(delay),
      is_running_(false) {
}

MockTimer::~MockTimer() {
}

bool MockTimer::IsRunning() const {
  return is_running_;
}

base::TimeDelta MockTimer::GetCurrentDelay() const {
  return delay_;
}

void MockTimer::Start(const tracked_objects::Location& posted_from,
                      TimeDelta delay,
                      const base::Closure& user_task) {
  delay_ = delay;
  user_task_ = user_task;
  Reset();
}

void MockTimer::Stop() {
  is_running_ = false;
  if (!retain_user_task())
    user_task_.Reset();
}

void MockTimer::Reset() {
  DCHECK(!user_task_.is_null());
  is_running_ = true;
}

void MockTimer::Fire() {
  DCHECK(is_running_);
  base::Closure old_task = user_task_;
  if (is_repeating())
    Reset();
  else
    Stop();
  old_task.Run();
}

}  // namespace base