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
use std::time::{Instant, Duration};

pub trait Interval : Default {
	fn now(&self) -> Instant {
		Instant::now()
	}

	fn elapsed(&self, instant: Instant) -> Duration {
		instant.elapsed()
	}
}

#[derive(Default)]
pub struct RealInterval;

impl Interval for RealInterval { }

#[derive(Default)]
#[cfg(test)]
pub struct FixedIntervalSpawner {
	step_millis: u64,
}

#[cfg(test)]
impl FixedIntervalSpawner {
	pub fn new(step_millis: u64) -> Self {
		FixedIntervalSpawner { step_millis : step_millis }
	}
}

#[cfg(test)]
impl Interval for FixedIntervalSpawner {
	fn now(&self) -> Instant {
		Instant::now()
	}

	fn elapsed(&self, instant: Instant) -> Duration {
		(instant - Duration::from_millis(self.step_millis)).elapsed()
	}
}