clone/codex
1
0
Fork 0
mirror of https://github.com/openai/codex.git synced 2026-04-24 22:54:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests: harden compact_remote trim test timing on windows

Browse Source
This commit is contained in:
Sayan Sisodiya
2026-02-02 21:42:38 -08:00
parent af050804d3
commit b68867e379
1 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
codex-rs/core/tests/suite/compact_remote.rs
View File

@@ -23,7 +23,9 @@ use core_test_support::test_codex::TestCodexHarness;
use core_test_support::test_codex::test_codex;
use core_test_support::wait_for_event;
use core_test_support::wait_for_event_match;
use core_test_support::wait_for_event_with_timeout;
use pretty_assertions::assert_eq;
use tokio::time::Duration;
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
async fn remote_compact_replaces_history_for_followups() -> Result<()> {
@@ -232,7 +234,7 @@ async fn remote_compact_trims_function_call_history_to_fit_context_window() -> R
let trimmed_call_id = "trimmed-call";
let retained_command = "echo retained-shell-output";
let trimmed_command = if cfg!(windows) {
"1..3000 | ForEach-Object { 'x' }"
"'x' * 50000"
} else {
"yes x | head -n 3000"
};
@@ -260,7 +262,13 @@ async fn remote_compact_trims_function_call_history_to_fit_context_window() -> R
responses::ev_completed("retained-final-response"),
]),
sse(vec![
responses::ev_shell_command_call(trimmed_call_id, trimmed_command),
responses::ev_shell_command_call_with_args(
trimmed_call_id,
&serde_json::json!({
"command": trimmed_command,
"timeout_ms": 5_000,
}),
),
responses::ev_completed("trimmed-call-response"),
]),
sse(vec![responses::ev_completed("trimmed-final-response")]),
@@ -277,7 +285,12 @@ async fn remote_compact_trims_function_call_history_to_fit_context_window() -> R
final_output_json_schema: None,
})
.await?;
wait_for_event(&codex, |event| matches!(event, EventMsg::TurnComplete(_))).await;
wait_for_event_with_timeout(
&codex,
|event| matches!(event, EventMsg::TurnComplete(_)),
Duration::from_secs(20),
)
.await;
codex
.submit(Op::UserInput {