test: stabilize done agent status api test

1 files changed, 7 insertions, 1 deletions

diff --git a/tests/api_ping.rs b/tests/api_ping.rs
index 1289abe6..7e00c315 100644
--- a/tests/api_ping.rs
+++ b/tests/api_ping.rs
@@ -1654,9 +1654,13 @@ fn pane_info_and_subscriptions_expose_done_agent_status() {
 
     fs::create_dir_all(&bin_dir).unwrap();
     let fake_pi = bin_dir.join("pi");
+    let stop_file = base.join("pi-stop");
     fs::write(
         &fake_pi,
-        "#!/bin/sh\nprintf 'Working...\\n'\nsleep 1\nprintf '\\033[2J\\033[Hdone\\n'\n",
+        format!(
+            "#!/bin/sh\nprintf 'Working...\\n'\nsleep 1\nprintf '\\033[2J\\033[Hdone\\n'\nwhile [ ! -f '{}' ]; do sleep 0.05; done\n",
+            stop_file.display()
+        ),
     )
     .unwrap();
     #[cfg(unix)]
@@ -1764,5 +1768,7 @@ fn pane_info_and_subscriptions_expose_done_agent_status() {
     );
     assert_eq!(pane_after_focus["result"]["pane"]["agent_status"], "idle");
 
+    fs::write(&stop_file, "stop").unwrap();
+
     cleanup_spawned_herdr(child, base);
 }