Revision control
Copy as Markdown
pub trait Clock: Send + Sync + 'static {
fn now_epoch_seconds(&self) -> i64;
#[cfg(test)]
fn advance(&self, secs: i64);
}
pub struct CacheClock;
impl Clock for CacheClock {
fn now_epoch_seconds(&self) -> i64 {
chrono::Utc::now().timestamp()
}
#[cfg(test)]
fn advance(&self, _secs: i64) {
panic!(
"
You cannot advance a non-test clock.
Be sure to build the cache or store with the test clock for time-dependent tests.
"
)
}
}
#[cfg(test)]
pub struct TestClock {
now: std::sync::atomic::AtomicI64,
}
#[cfg(test)]
impl TestClock {
pub fn new(start: i64) -> Self {
Self {
now: std::sync::atomic::AtomicI64::new(start),
}
}
}
#[cfg(test)]
impl Clock for TestClock {
fn now_epoch_seconds(&self) -> i64 {
self.now.load(std::sync::atomic::Ordering::Relaxed)
}
fn advance(&self, secs: i64) {
self.now
.fetch_add(secs, std::sync::atomic::Ordering::Relaxed);
}
}