Add unit test for ConstantInterval

mullvad · Sep 19, 2023 · b509134 · b509134
1 parent ed44842
commit b509134
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/talpid-core/src/future_retry.rs b/talpid-core/src/future_retry.rs
@@ -152,6 +152,22 @@ fn apply_jitter(duration: Duration, jitter: f64) -> Duration {
 mod test {
     use super::*;
 
+    #[test]
+    fn test_constant_interval() {
+        let mut ivl = ConstantInterval::new(Duration::from_secs(2), Some(3));
+
+        assert_eq!(ivl.next(), Some(Duration::from_secs(2)));
+        assert_eq!(ivl.next(), Some(Duration::from_secs(2)));
+        assert_eq!(ivl.next(), Some(Duration::from_secs(2)));
+        assert_eq!(ivl.next(), None);
+    }
+
+    #[test]
+    fn test_constant_interval_no_max() {
+        let mut ivl = ConstantInterval::new(Duration::from_secs(2), None);
+        assert_eq!(ivl.next(), Some(Duration::from_secs(2)));
+    }
+
     #[test]
     fn test_exponential_backoff() {
         let mut backoff = ExponentialBackoff::new(Duration::from_secs(2), 3);