From b668dae0354b523efdf433914400ca8f2e235ec8 Mon Sep 17 00:00:00 2001 From: Charles Cunningham Date: Fri, 6 Mar 2026 14:11:28 -0800 Subject: [PATCH] codex: fix CI failure on PR #13096 Co-authored-by: Codex --- codex-rs/core/src/codex_tests.rs | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/codex-rs/core/src/codex_tests.rs b/codex-rs/core/src/codex_tests.rs index 4f3b82da6b..dbb9e0318a 100644 --- a/codex-rs/core/src/codex_tests.rs +++ b/codex-rs/core/src/codex_tests.rs @@ -659,7 +659,8 @@ fn extract_mcp_tool_selection_from_rollout_reads_search_tool_output() { ), ]; - let selected = Session::extract_mcp_tool_selection_from_rollout(&rollout_items); + let rollout_source = InMemoryRolloutSource::new(rollout_items); + let selected = Session::extract_mcp_tool_selection_from_rollout_source(&rollout_source); assert_eq!( selected, Some(vec![ @@ -690,7 +691,8 @@ fn extract_mcp_tool_selection_from_rollout_latest_valid_payload_wins() { ), ]; - let selected = Session::extract_mcp_tool_selection_from_rollout(&rollout_items); + let rollout_source = InMemoryRolloutSource::new(rollout_items); + let selected = Session::extract_mcp_tool_selection_from_rollout_source(&rollout_source); assert_eq!( selected, Some(vec!["mcp__codex_apps__calendar_delete_event".to_string(),]) @@ -726,7 +728,8 @@ fn extract_mcp_tool_selection_from_rollout_ignores_non_search_and_malformed_payl ), ]; - let selected = Session::extract_mcp_tool_selection_from_rollout(&rollout_items); + let rollout_source = InMemoryRolloutSource::new(rollout_items); + let selected = Session::extract_mcp_tool_selection_from_rollout_source(&rollout_source); assert_eq!( selected, Some(vec!["mcp__codex_apps__calendar_list_events".to_string(),]) @@ -739,7 +742,8 @@ fn extract_mcp_tool_selection_from_rollout_returns_none_without_valid_search_out SEARCH_TOOL_BM25_TOOL_NAME, "search-1", )]; - let selected = Session::extract_mcp_tool_selection_from_rollout(&rollout_items); + let rollout_source = InMemoryRolloutSource::new(rollout_items); + let selected = Session::extract_mcp_tool_selection_from_rollout_source(&rollout_source); assert_eq!(selected, None); }