Parameterize actions sleep interval for faster test execution.

bittrance · Oct 2, 2023 · b8348b5 · b8348b5
1 parent 93a1533
commit b8348b5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/actions.rs b/src/actions.rs
@@ -5,7 +5,7 @@ use std::{
     process::{Command, Stdio},
     sync::{Arc, Mutex},
     thread::{sleep, spawn, JoinHandle},
-    time::{Duration, Instant},
+    time::Instant,
 };
 
 use serde::Deserialize;
@@ -14,6 +14,7 @@ use crate::{
     errors::GitOpsError,
     opts::CliOptions,
     receiver::{SourceType, WorkloadEvent},
+    utils::POLL_INTERVAL,
 };
 
 #[derive(Debug, PartialEq)]
@@ -140,7 +141,7 @@ where
             sink.lock().unwrap()(WorkloadEvent::Timeout(name.to_string()))?;
             break Ok(ActionResult::Failure);
         }
-        sleep(Duration::from_secs(1));
+        sleep(POLL_INTERVAL);
     }
 }
 
@@ -149,6 +150,7 @@ mod tests {
     use std::{
         process::ExitStatus,
         sync::{Arc, Mutex},
+        time::Duration,
     };
 
     use super::*;

diff --git a/src/utils.rs b/src/utils.rs
@@ -4,9 +4,9 @@ use std::thread::sleep;
 use std::time::{Duration, Instant};
 
 #[cfg(test)]
-const POLL_INTERVAL: Duration = Duration::from_millis(10);
+pub const POLL_INTERVAL: Duration = Duration::from_millis(10);
 #[cfg(not(test))]
-const POLL_INTERVAL: Duration = Duration::from_millis(500);
+pub const POLL_INTERVAL: Duration = Duration::from_millis(500);
 
 pub struct Watchdog {
     deadline: Instant,