Private
Public Access
0
0

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.
This commit is contained in:
2026-06-15 16:31:26 -04:00
parent fab9196bea
commit b4c9ebd963
+4 -2
View File
@@ -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"