From f5e27c5d2d57b1504740c9248a2016f2b29ff003 Mon Sep 17 00:00:00 2001 From: Charles Cunningham Date: Thu, 19 Mar 2026 16:25:17 -0700 Subject: [PATCH] codex: fix CI failure on PR #15226 Co-authored-by: Codex --- codex-rs/core/src/guardian/review_session.rs | 30 ++++++++++++++++++++ 1 file changed, 30 insertions(+) diff --git a/codex-rs/core/src/guardian/review_session.rs b/codex-rs/core/src/guardian/review_session.rs index 221a7cf5fd..de9ca4c2ff 100644 --- a/codex-rs/core/src/guardian/review_session.rs +++ b/codex-rs/core/src/guardian/review_session.rs @@ -518,6 +518,36 @@ impl GuardianReviewSessionManager { } } + #[cfg(test)] + pub(crate) async fn cache_for_test(&self, codex: Codex) { + let reuse_key = GuardianReviewSessionReuseKey::from_spawn_config( + codex.session.get_config().await.as_ref(), + ); + self.state.lock().await.trunk = Some(Arc::new(GuardianReviewSession { + reuse_key, + codex, + cancel_token: CancellationToken::new(), + review_lock: Mutex::new(()), + last_committed_rollout_items: Mutex::new(None), + })); + } + + #[cfg(test)] + pub(crate) async fn register_fork_for_test(&self, codex: Codex) { + let reuse_key = GuardianReviewSessionReuseKey::from_spawn_config( + codex.session.get_config().await.as_ref(), + ); + self.fork_pool + .register(Arc::new(GuardianReviewSession { + reuse_key, + codex, + cancel_token: CancellationToken::new(), + review_lock: Mutex::new(()), + last_committed_rollout_items: Mutex::new(None), + })) + .await; + } + async fn choose_review_lane<'a>( &self, trunk: &'a Arc,