// Copyright 2019 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use fuchsia_zircon as zx;

#[cfg(not(test))]
pub(crate) fn now() -> zx::Time {
    zx::Time::get_monotonic()
}

#[cfg(test)]
pub(crate) use mock::now;

#[cfg(test)]
pub(crate) mod mock {
    use {super::*, std::cell::Cell};

    thread_local!(
        static MOCK_TIME: Cell<zx::Time> = Cell::new(zx::Time::get_monotonic())
    );

    pub fn now() -> zx::Time {
        MOCK_TIME.with(|time| time.get())
    }

    pub fn set(new_time: zx::Time) {
        MOCK_TIME.with(|time| time.set(new_time));
    }
}