From b4c9ebd9630b7dd66ab719eff042e39d018f7b3e Mon Sep 17 00:00:00 2001 From: Ed_ Date: Mon, 15 Jun 2026 16:31:26 -0400 Subject: [PATCH] test(gemini_cli_edge_cases): migrate to send_result() (Phase 2.3) The test_gemini_cli_loop_termination test calls ai_client.send() and asserts the return value. Migrating to send_result() with assert result.ok and result.data == "Final answer". Changes: - Rename ai_client.send(...) to ai_client.send_result(...) - Add assert result.ok before accessing result.data - Add "from src.result_types import Result" import 3 tests pass. --- tests/test_gemini_cli_edge_cases.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/test_gemini_cli_edge_cases.py b/tests/test_gemini_cli_edge_cases.py index 253ccde9..7cba8455 100644 --- a/tests/test_gemini_cli_edge_cases.py +++ b/tests/test_gemini_cli_edge_cases.py @@ -1,6 +1,7 @@ from unittest.mock import patch, MagicMock from src.gemini_cli_adapter import GeminiCliAdapter from src import mcp_client +from src.result_types import Result def test_gemini_cli_context_bleed_prevention() -> None: import src.ai_client as ai_client @@ -35,5 +36,6 @@ def test_gemini_cli_loop_termination() -> None: mock_process.returncode = 0 mock_popen.return_value = mock_process ai_client.set_provider("gemini_cli", "gemini-2.0-flash") - result = ai_client.send("context", "prompt") - assert result == "Final answer" + result = ai_client.send_result("context", "prompt") + assert result.ok + assert result.data == "Final answer"