fix: Container can be both exiting and removing in tests (#57)

PR created by kwaak, but PR body too big Feedback: Seems it didn't run rustfmt --------- Co-authored-by: kwaak <[email protected]>
bosun-ai · Dec 21, 2024 · 5614667 · 5614667
1 parent 55af01e
commit 5614667
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/agent/docker_tool_executor.rs b/src/agent/docker_tool_executor.rs
@@ -493,9 +493,12 @@ mod tests {
 
         // assert it stopped
         let container = docker.inspect_container(&container_id, None).await.unwrap();
-        assert_eq!(
-            container.state.as_ref().unwrap().status,
-            Some(ContainerStateStatusEnum::REMOVING)
+        let status = container.state.as_ref().unwrap().status;
+        assert!(
+            status == Some(ContainerStateStatusEnum::REMOVING)
+                || status == Some(ContainerStateStatusEnum::EXITED),
+            "Unexpected container state: {:?}",
+            status
         );
     }