diff --git a/tests/test_spawn_interception_v2.py b/tests/test_spawn_interception_v2.py index 324e6f4d..aa8c70dd 100644 --- a/tests/test_spawn_interception_v2.py +++ b/tests/test_spawn_interception_v2.py @@ -3,6 +3,7 @@ from unittest.mock import MagicMock, patch from src import multi_agent_conductor from src.models import Ticket, WorkerContext from src import events +from src.result_types import Result import threading from typing import Generator @@ -19,9 +20,9 @@ class MockDialog: @pytest.fixture def mock_ai_client() -> Generator[MagicMock, None, None]: - with patch("src.ai_client.send") as mock_send: - mock_send.return_value = "Task completed" - yield mock_send + with patch("src.ai_client.send_result") as mock_send_result: + mock_send_result.return_value = Result(data="Task completed") + yield mock_send_result def test_confirm_spawn_pushed_to_queue() -> None: event_queue = events.SyncEventQueue()