openai/codex
1
0
Fork 0
mirror of https://github.com/openai/codex.git synced 2026-10-18 17:18:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Wait for reloaded worker completion in the resume test (#35175)

Browse Source 
## Why

The follow-up request assertion can run before the lazily reloaded worker has
finished handling its task.

## What changed

Poll the reloaded worker's status with a two-second deadline, then inspect the
recorded request after the worker reports completion.

GitOrigin-RevId: 4d53cdc5e5ed53009b4445385f8e835e19be7b8b
This commit is contained in:
jif
2026-07-24 13:04:23 +00:00
committed by copyberry
parent 5a1c54fc21
commit 1a817bb95d
1 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
codex-rs/core/tests/suite/multi_agent_resume.rs
+17 -4
View File
@@ -279,15 +279,28 @@ async fn cold_root_resume_restores_agent_identity_and_role_on_followup() -> Resu
resumed.submit_turn(FOLLOWUP_PROMPT).await?;
assert!(followup_child_request.requests().iter().any(|request| {
request.body_contains_text(FOLLOWUP_TASK)
&& request.body_contains_text(ROLE_DEVELOPER_INSTRUCTIONS)
}));
let reloaded_worker = resumed
.thread_manager
.get_thread(worker_thread_id)
.await
.expect("follow-up should lazily reload the original worker");
let deadline = Instant::now() + Duration::from_secs(2);
loop {
if matches!(
reloaded_worker.agent_status().await,
AgentStatus::Completed(_)
) {
break;
}
if Instant::now() >= deadline {
anyhow::bail!("timed out waiting for reloaded worker completion");
}
sleep(Duration::from_millis(10)).await;
}
assert!(followup_child_request.requests().iter().any(|request| {
request.body_contains_text(FOLLOWUP_TASK)
&& request.body_contains_text(ROLE_DEVELOPER_INSTRUCTIONS)
}));
let reloaded_worker_config = reloaded_worker.config_snapshot().await;
let reloaded_worker_role_config = (
reloaded_worker_config.model,