test(deployment): use test_states in deployment_to_be_queued

Instead of relying on a one second sleep.
shuttle-hq · Jun 7, 2023 · 2901f0f · 2901f0f
1 parent b2914b5
commit 2901f0f
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/deployer/src/deployment/deploy_layer.rs b/deployer/src/deployment/deploy_layer.rs
@@ -633,12 +633,8 @@ mod tests {
         // Send kill signal
         deployment_manager.kill(id).await;
 
-        sleep(Duration::from_secs(1)).await;
-
-        let states = RECORDER.lock().unwrap().get_deployment_states(&id);
-
-        assert_eq!(
-            *states,
+        let test = test_states(
+            &id,
             vec![
                 StateLog {
                     id,
@@ -664,8 +660,16 @@ mod tests {
                     id,
                     state: State::Stopped,
                 },
-            ]
+            ],
         );
+
+        select! {
+            _ = sleep(Duration::from_secs(60)) => {
+                let states = RECORDER.lock().unwrap().get_deployment_states(&id);
+                panic!("states should go into 'Stopped' for a valid service: {:#?}", states);
+            },
+            _ = test => {}
+        };
     }
 
     #[tokio::test(flavor = "multi_thread")]