diff --git a/tests/test_gemini_cli_parity_regression.py b/tests/test_gemini_cli_parity_regression.py index dd37fcf2..29c76d12 100644 --- a/tests/test_gemini_cli_parity_regression.py +++ b/tests/test_gemini_cli_parity_regression.py @@ -1,4 +1,5 @@ from unittest.mock import patch, MagicMock +from src.result_types import Result def test_send_invokes_adapter_send() -> None: import src.ai_client as ai_client @@ -9,5 +10,6 @@ def test_send_invokes_adapter_send() -> None: mock_process.returncode = 0 mock_popen.return_value = mock_process ai_client.set_provider("gemini_cli", "gemini-2.0-flash") - res = ai_client.send("context", "msg") - assert res == "Hello from mock adapter" + res = ai_client.send_result("context", "msg") + assert res.ok + assert res.data == "Hello from mock adapter"