diff --git a/tests/test_headless_verification.py b/tests/test_headless_verification.py index 088213b1..d39bafca 100644 --- a/tests/test_headless_verification.py +++ b/tests/test_headless_verification.py @@ -5,6 +5,7 @@ from src.models import Ticket, Track from src import multi_agent_conductor from src.multi_agent_conductor import ConductorEngine from src import ai_client +from src.result_types import Result @pytest.mark.asyncio async def test_headless_verification_full_run(vlogger) -> None: @@ -31,7 +32,7 @@ async def test_headless_verification_full_run(vlogger) -> None: patch("src.multi_agent_conductor.ai_client.reset_session") as mock_reset, \ patch("src.multi_agent_conductor.confirm_spawn", return_value=(True, "mock_prompt", "mock_ctx")): # We need mock_send to return something that doesn't contain "BLOCKED" - mock_send.return_value = "Task completed successfully." + mock_send.return_value = Result(data="Task completed successfully.") engine.run() vlogger.log_state("T1 Status Final", "todo", t1.status)