update_engine/common/fake_clock.h

// Copyright 2013 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef UPDATE_ENGINE_COMMON_FAKE_CLOCK_H_
#define UPDATE_ENGINE_COMMON_FAKE_CLOCK_H_

#include "update_engine/common/clock_interface.h"

namespace chromeos_update_engine {

// Implements a clock that can be made to tell any time you want.
class FakeClock : public ClockInterface {
 public:
  FakeClock() {}
  FakeClock(const FakeClock&) = delete;
  FakeClock& operator=(const FakeClock&) = delete;

  base::Time GetWallclockTime() const override { return wallclock_time_; }

  base::Time GetMonotonicTime() const override { return monotonic_time_; }

  base::Time GetBootTime() const override { return boot_time_; }

  void SetWallclockTime(const base::Time& time) { wallclock_time_ = time; }

  void SetMonotonicTime(const base::Time& time) { monotonic_time_ = time; }

  void SetBootTime(const base::Time& time) { boot_time_ = time; }

 private:
  base::Time wallclock_time_;
  base::Time monotonic_time_;
  base::Time boot_time_;
};

}  // namespace chromeos_update_engine

#endif  // UPDATE_ENGINE_COMMON_FAKE_CLOCK_H_