diff --git a/tests/test_ai_client_cli.py b/tests/test_ai_client_cli.py index ddc21597..8bbd4f68 100644 --- a/tests/test_ai_client_cli.py +++ b/tests/test_ai_client_cli.py @@ -1,5 +1,6 @@ from unittest.mock import patch, MagicMock from src import ai_client +from src.result_types import Result def test_ai_client_send_gemini_cli() -> None: @@ -19,7 +20,7 @@ def test_ai_client_send_gemini_cli() -> None: MockAdapterClass.return_value = mock_adapter_instance ai_client._gemini_cli_adapter = mock_adapter_instance with patch.object(ai_client.events, "emit") as mock_emit: - response = ai_client.send( + result = ai_client.send_result( md_content="", user_message=test_message, base_dir=".", @@ -28,4 +29,5 @@ def test_ai_client_send_gemini_cli() -> None: emitted_event_names = [call.args[0] for call in mock_emit.call_args_list] assert "request_start" in emitted_event_names assert "response_received" in emitted_event_names - assert response == test_response + assert result.ok + assert result.data == test_response