Add unit tests to ExponentialBackoff

mullvad · Oct 3, 2023 · 62e9694 · 62e9694
1 parent 0e38a3c
commit 62e9694
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/src/exponential_backoff.rs b/src/exponential_backoff.rs
@@ -35,3 +35,29 @@ impl ExponentialBackoff {
         delay
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn correct_delays() {
+        let mut backoff =
+            ExponentialBackoff::new(Duration::from_millis(60), Duration::from_millis(300));
+        assert_eq!(backoff.next_delay(), Duration::from_millis(60));
+        assert_eq!(backoff.next_delay(), Duration::from_millis(120));
+        assert_eq!(backoff.next_delay(), Duration::from_millis(240));
+        assert_eq!(backoff.next_delay(), Duration::from_millis(300));
+        assert_eq!(backoff.next_delay(), Duration::from_millis(300));
+    }
+
+    #[test]
+    fn reset() {
+        let mut backoff =
+            ExponentialBackoff::new(Duration::from_millis(60), Duration::from_millis(300));
+        assert_eq!(backoff.next_delay(), Duration::from_millis(60));
+        backoff.reset();
+        assert_eq!(backoff.next_delay(), Duration::from_millis(60));
+        assert_eq!(backoff.next_delay(), Duration::from_millis(120));
+    }
+}