From b708d10db0afe3c3772fa0ad070ae8032b190535 Mon Sep 17 00:00:00 2001 From: kshitijk4poor <82637225+kshitijk4poor@users.noreply.github.com> Date: Tue, 14 Jul 2026 01:42:23 +0530 Subject: [PATCH] test(session): type finalizer clean-history assertions --- ...t_turn_finalizer_final_response_persistence.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/tests/agent/test_turn_finalizer_final_response_persistence.py b/tests/agent/test_turn_finalizer_final_response_persistence.py index 9c6089aacbd8..a007e4732644 100644 --- a/tests/agent/test_turn_finalizer_final_response_persistence.py +++ b/tests/agent/test_turn_finalizer_final_response_persistence.py @@ -1,4 +1,5 @@ from types import SimpleNamespace +from typing import Any from agent.turn_finalizer import finalize_turn @@ -30,7 +31,10 @@ class FakeAgent: self._skill_nudge_interval = 0 self._iters_since_skill = 0 self.valid_tool_names = [] - self.persisted_messages = None + self.persisted_messages: list[dict[str, Any]] | None = None + self._persist_user_message_idx: int | None = None + self._persist_user_message_override: Any = None + self._persist_user_message_timestamp: float | None = None def _handle_max_iterations(self, messages, api_call_count): raise AssertionError("not expected") @@ -56,9 +60,10 @@ class FakeAgent: self.persisted_messages = [dict(message) for message in messages] def _apply_persist_user_message_override(self, messages): - from run_agent import AIAgent - - return AIAgent._apply_persist_user_message_override(self, messages) + idx = self._persist_user_message_idx + override = self._persist_user_message_override + if idx is not None and override is not None: + messages[idx]["content"] = override def _file_mutation_verifier_enabled(self): return False @@ -104,6 +109,7 @@ def test_finalizer_restores_clean_api_local_text_before_return(monkeypatch): _turn_exit_reason="text_response(finish_reason=stop)", ) + assert agent.persisted_messages is not None assert agent.persisted_messages[0]["content"] == "clean prompt" assert result["messages"][0]["content"] == "clean prompt" @@ -144,6 +150,7 @@ def test_finalizer_restores_clean_api_local_multimodal_before_return(monkeypatch _turn_exit_reason="text_response(finish_reason=stop)", ) + assert agent.persisted_messages is not None assert agent.persisted_messages[0]["content"] == clean_content assert result["messages"][0]["content"] == clean_content