diff --git a/codex-rs/app-server/tests/suite/auth.rs b/codex-rs/app-server/tests/suite/auth.rs index 5d3a3dcd4a..e6134e4802 100644 --- a/codex-rs/app-server/tests/suite/auth.rs +++ b/codex-rs/app-server/tests/suite/auth.rs @@ -356,6 +356,7 @@ async fn get_auth_status_omits_token_after_proactive_refresh_failure() -> Result "code": "refresh_token_reused" } }))) + .expect(2) .mount(&server) .await; @@ -395,6 +396,7 @@ async fn get_auth_status_omits_token_after_proactive_refresh_failure() -> Result } ); + server.verify().await; Ok(()) } @@ -421,6 +423,7 @@ async fn get_auth_status_returns_token_after_proactive_refresh_recovery() -> Res "code": "refresh_token_reused" } }))) + .expect(2) .mount(&server) .await; @@ -493,6 +496,7 @@ async fn get_auth_status_returns_token_after_proactive_refresh_recovery() -> Res } ); + server.verify().await; Ok(()) } diff --git a/codex-rs/core/src/codex_tests.rs b/codex-rs/core/src/codex_tests.rs index a1c05e5bb0..a6a60854b3 100644 --- a/codex-rs/core/src/codex_tests.rs +++ b/codex-rs/core/src/codex_tests.rs @@ -1164,45 +1164,6 @@ async fn record_initial_history_reconstructs_forked_transcript() { assert_eq!(expected, history.raw_items()); } -fn normalize_fork_startup_request_snapshot(snapshot: &str) -> String { - let lines = snapshot.lines().collect::>(); - let developer_index = lines - .iter() - .position(|line| { - *line == "00:message/developer[2]:" || line.ends_with(":message/developer[2]:") - }) - .expect("fork request should contain developer[2] instructions"); - let _env_line = lines - .iter() - .find(|line| line.contains("message/user:>")) - .copied() - .expect("fork request should contain environment context"); - let _after_fork_line = lines - .iter() - .find(|line| line.ends_with("message/user:after fork")) - .copied() - .expect("fork request should contain the post-fork user input"); - - let mut normalized = vec![ - lines[0].to_string(), - lines[1].to_string(), - lines[2].to_string(), - "00:message/developer[2]:".to_string(), - ]; - normalized.extend( - lines[developer_index + 1..] - .iter() - .copied() - .take_while(|line| line.starts_with(" ")) - .map(str::to_string), - ); - normalized.extend([ - "01:message/user:>".to_string(), - "02:message/user:after fork".to_string(), - ]); - normalized.join("\n") -} - #[tokio::test(flavor = "multi_thread", worker_threads = 2)] async fn fork_startup_context_then_first_turn_diff_snapshot() -> anyhow::Result<()> { let server = start_mock_server().await; @@ -1304,7 +1265,6 @@ async fn fork_startup_context_then_first_turn_diff_snapshot() -> anyhow::Result< .strip_capability_instructions() .strip_agents_md_user_context(), ); - let snapshot = normalize_fork_startup_request_snapshot(&snapshot); let mut settings = insta::Settings::clone_current(); settings.set_snapshot_path("snapshots"); diff --git a/codex-rs/core/src/snapshots/codex_core__codex_tests__fork_startup_context_then_first_turn_diff.snap b/codex-rs/core/src/snapshots/codex_core__codex_tests__fork_startup_context_then_first_turn_diff.snap index c4b5f5f963..d55f29a2a7 100644 --- a/codex-rs/core/src/snapshots/codex_core__codex_tests__fork_startup_context_then_first_turn_diff.snap +++ b/codex-rs/core/src/snapshots/codex_core__codex_tests__fork_startup_context_then_first_turn_diff.snap @@ -6,8 +6,10 @@ expression: snapshot Scenario: First request after fork when startup preserves the parent baseline, the fork changes approval policy, and the first forked turn enters plan mode. ## First Forked Turn Request -00:message/developer[2]: +00:message/developer: +01:message/user:> +02:message/user:fork seed +03:message/developer[2]: [01] [02] Fork turn collaboration instructions. -01:message/user:> -02:message/user:after fork +04:message/user:after fork