diff --git a/codex-rs/core/tests/suite/workspace_roots.rs b/codex-rs/core/tests/suite/workspace_roots.rs index 0238b74077..ee95b58434 100644 --- a/codex-rs/core/tests/suite/workspace_roots.rs +++ b/codex-rs/core/tests/suite/workspace_roots.rs @@ -26,14 +26,16 @@ use core_test_support::skip_if_no_remote_env; use core_test_support::test_codex::TestCodex; use core_test_support::test_codex::test_codex; use core_test_support::test_target_os; -use core_test_support::wait_for_event; +use core_test_support::wait_for_event_with_timeout; use pretty_assertions::assert_eq; use serde_json::Value; use serde_json::json; +use tokio::time::Duration; use wiremock::MockServer; const IMAGE_CALL_ID: &str = "workspace-root-image"; const COMMAND_CALL_ID: &str = "workspace-root-command"; +const TURN_COMPLETE_TIMEOUT: Duration = Duration::from_secs(30); const PNG_BASE64: &str = "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNk+A8AAQUBAScY42YAAAAASUVORK5CYII="; @@ -128,9 +130,11 @@ async fn mount_file_and_command_calls( async fn submit_workspace_turn(test: &TestCodex, prompt: &str) -> Result<()> { test.submit_turn_with_permission_profile(prompt, workspace_roots_read_profile()) .await?; - wait_for_event(&test.codex, |event| { - matches!(event, EventMsg::TurnComplete(_)) - }) + wait_for_event_with_timeout( + &test.codex, + |event| matches!(event, EventMsg::TurnComplete(_)), + TURN_COMPLETE_TIMEOUT, + ) .await; Ok(()) }