diff --git a/simlib/mixnet-sims/src/node/mix/consensus_streams.rs b/simlib/mixnet-sims/src/node/mix/consensus_streams.rs index 09d1f0b..ea1edf1 100644 --- a/simlib/mixnet-sims/src/node/mix/consensus_streams.rs +++ b/simlib/mixnet-sims/src/node/mix/consensus_streams.rs @@ -67,6 +67,7 @@ mod tests { let (update_sender, update_receiver) = channel::unbounded(); let mut interval = CounterInterval::new(Duration::from_secs(1), update_receiver); + update_sender.send(Duration::from_secs(0)).unwrap(); assert_eq!(interval.poll_next_unpin(&mut cx), Poll::Ready(Some(0))); update_sender.send(Duration::from_secs(0)).unwrap(); assert_eq!(interval.poll_next_unpin(&mut cx), Poll::Pending); @@ -88,6 +89,7 @@ mod tests { let (update_sender, update_receiver) = channel::unbounded(); let mut slot = Slot::new(3, Duration::from_secs(1), update_receiver); + update_sender.send(Duration::from_secs(0)).unwrap(); assert_eq!(slot.poll_next_unpin(&mut cx), Poll::Ready(Some(0))); update_sender.send(Duration::from_secs(0)).unwrap(); assert_eq!(slot.poll_next_unpin(&mut cx), Poll::Pending); diff --git a/simlib/mixnet-sims/src/node/mix/scheduler.rs b/simlib/mixnet-sims/src/node/mix/scheduler.rs index 6865e4b..ac403c8 100644 --- a/simlib/mixnet-sims/src/node/mix/scheduler.rs +++ b/simlib/mixnet-sims/src/node/mix/scheduler.rs @@ -118,6 +118,7 @@ mod tests { let (tx, rx) = channel::unbounded(); let mut interval = Interval::new(Duration::from_secs(2), rx); + tx.send(Duration::from_secs(0)).unwrap(); assert_eq!(interval.poll_next_unpin(&mut cx), Poll::Ready(Some(()))); tx.send(Duration::from_secs(0)).unwrap(); assert_eq!(interval.poll_next_unpin(&mut cx), Poll::Pending); @@ -133,7 +134,7 @@ mod tests { fn temporal_release_update() { let (_tx, rx) = channel::unbounded(); let mut temporal_release = - TemporalRelease::new(rand_chacha::ChaCha8Rng::from_entropy(), rx, (1, 2)); + TemporalRelease::new(rand_chacha::ChaCha8Rng::from_entropy(), rx, (1, 1)); assert!(!temporal_release.update(Duration::from_secs(0))); assert!(!temporal_release.update(Duration::from_millis(999))); @@ -148,12 +149,8 @@ mod tests { let (tx, rx) = channel::unbounded(); let mut temporal_release = - TemporalRelease::new(rand_chacha::ChaCha8Rng::from_entropy(), rx, (1, 2)); + TemporalRelease::new(rand_chacha::ChaCha8Rng::from_entropy(), rx, (1, 1)); - assert_eq!( - temporal_release.poll_next_unpin(&mut cx), - Poll::Ready(Some(())) - ); tx.send(Duration::from_secs(0)).unwrap(); assert_eq!(temporal_release.poll_next_unpin(&mut cx), Poll::Pending); tx.send(Duration::from_millis(999)).unwrap();