diff options
| -rw-r--r-- | talpid-core/src/future_retry.rs | 19 |
1 files changed, 3 insertions, 16 deletions
diff --git a/talpid-core/src/future_retry.rs b/talpid-core/src/future_retry.rs index a88f6c3b36..8b4745a7b2 100644 --- a/talpid-core/src/future_retry.rs +++ b/talpid-core/src/future_retry.rs @@ -139,7 +139,6 @@ fn apply_jitter(duration: Duration, jitter: f64) -> Duration { #[cfg(test)] mod test { use super::*; - use rand::SeedableRng; #[test] fn test_exponetnial_backoff() { @@ -187,22 +186,10 @@ mod test { assert_eq!(apply_jitter(second, 1.0), second); } - #[derive(Clone, Debug)] - struct ArbitraryJitter(f64); - impl quickcheck::Arbitrary for ArbitraryJitter { - fn arbitrary(_g: &mut quickcheck::Gen) -> Self { - // This should not be private: - // g.gen_range - - let mut rng = rand::rngs::SmallRng::from_entropy(); - let jitter: f64 = rng.sample(OpenClosed01); - ArbitraryJitter(jitter) - } - } - #[quickcheck_macros::quickcheck] - fn test_jitter(millis: u64, jitter: ArbitraryJitter) { - let jitter = jitter.0; + fn test_jitter(millis: u64, jitter: u64) { + let max_num = 2u64.checked_pow(f64::MANTISSA_DIGITS).unwrap(); + let jitter = (jitter % max_num) as f64 / (max_num as f64); let unjittered_duration = Duration::from_millis(millis); let jittered_duration = apply_jitter(unjittered_duration, jitter); assert!(jittered_duration <= unjittered_duration); |