diff --git a/tests/test_orchestration_logic.py b/tests/test_orchestration_logic.py index c4948d79..ed186ddd 100644 --- a/tests/test_orchestration_logic.py +++ b/tests/test_orchestration_logic.py @@ -105,11 +105,11 @@ def test_conductor_engine_parse_json_tickets() -> None: def test_run_worker_lifecycle_blocked() -> None: ticket = Ticket(id="T1", description="desc", status="todo", assigned_to="worker1") context = WorkerContext(ticket_id="T1", model_name="model", messages=[]) - with patch("src.ai_client.send") as mock_ai_client, \ + with patch("src.ai_client.send_result") as mock_ai_client, \ patch("src.ai_client.reset_session"), \ patch("src.ai_client.set_provider"), \ patch("src.multi_agent_conductor.confirm_spawn", return_value=(True, "p", "c")): - mock_ai_client.return_value = "BLOCKED because of missing info" + mock_ai_client.return_value = Result(data="BLOCKED because of missing info") multi_agent_conductor.run_worker_lifecycle(ticket, context) assert ticket.status == "blocked" assert ticket.blocked_reason == "BLOCKED because of missing info"