diff --git a/tests/test_phase6_engine.py b/tests/test_phase6_engine.py index 81b48d47..6eb687b9 100644 --- a/tests/test_phase6_engine.py +++ b/tests/test_phase6_engine.py @@ -2,6 +2,7 @@ from unittest.mock import MagicMock, patch from src.multi_agent_conductor import ConductorEngine, run_worker_lifecycle from src.models import Ticket, Track, WorkerContext from src import ai_client +from src.result_types import Result def test_worker_streaming_intermediate(): ticket = Ticket(id="T-001", description="Test", status="todo", assigned_to="worker") @@ -9,12 +10,12 @@ def test_worker_streaming_intermediate(): event_queue = MagicMock() with ( - patch("src.ai_client.send") as mock_send, - patch("src.multi_agent_conductor._queue_put") as mock_q_put, - patch("src.multi_agent_conductor.confirm_spawn", return_value=(True, "p", "c")), - patch("src.ai_client.reset_session"), - patch("src.ai_client.set_provider"), - patch("src.ai_client.get_provider"), + patch("src.ai_client.send_result") as mock_send_result, + patch("src.multi_agent_conductor._queue_put") as mock_q_put, + patch("src.multi_agent_conductor.confirm_spawn", return_value=(True, "p", "c")), + patch("src.ai_client.reset_session"), + patch("src.ai_client.set_provider"), + patch("src.ai_client.get_provider"), patch("src.ai_client.get_comms_log", return_value=[]) ): @@ -23,9 +24,9 @@ def test_worker_streaming_intermediate(): if cb: cb({"kind": "tool_call", "payload": {"name": "test_tool", "script": "echo hello"}}) cb({"kind": "tool_result", "payload": {"name": "test_tool", "output": "hello"}}) - return "DONE" - - mock_send.side_effect = side_effect + return Result(data="DONE") + + mock_send_result.side_effect = side_effect run_worker_lifecycle(ticket, context, event_queue=event_queue) # _queue_put(event_queue, event_name, payload)