Private
Public Access
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:
@@ -1,6 +1,7 @@
|
|||||||
from unittest.mock import patch, MagicMock
|
from unittest.mock import patch, MagicMock
|
||||||
from src.gemini_cli_adapter import GeminiCliAdapter
|
from src.gemini_cli_adapter import GeminiCliAdapter
|
||||||
from src import mcp_client
|
from src import mcp_client
|
||||||
|
from src.result_types import Result
|
||||||
|
|
||||||
def test_gemini_cli_context_bleed_prevention() -> None:
|
def test_gemini_cli_context_bleed_prevention() -> None:
|
||||||
import src.ai_client as ai_client
|
import src.ai_client as ai_client
|
||||||
@@ -35,5 +36,6 @@ def test_gemini_cli_loop_termination() -> None:
|
|||||||
mock_process.returncode = 0
|
mock_process.returncode = 0
|
||||||
mock_popen.return_value = mock_process
|
mock_popen.return_value = mock_process
|
||||||
ai_client.set_provider("gemini_cli", "gemini-2.0-flash")
|
ai_client.set_provider("gemini_cli", "gemini-2.0-flash")
|
||||||
result = ai_client.send("context", "prompt")
|
result = ai_client.send_result("context", "prompt")
|
||||||
assert result == "Final answer"
|
assert result.ok
|
||||||
|
assert result.data == "Final answer"
|
||||||
|
|||||||
Reference in New Issue
Block a user