Private
Public Access
test(spawn_interception_v2): mock send_result not send (Phase 2.18, pre-empts Phase 1.3 regression)
Phase 1.3 migrated run_worker_lifecycle to send_result(). The mock_ai_client fixture in test_spawn_interception_v2.py mocked src.ai_client.send and returned a string. The test_run_worker_lifecycle_approved test asserts on the call_args (user_message + md_content), which still works with the new mock name. Changes: - Replace patch(src.ai_client.send) with patch(src.ai_client.send_result) - Rename mock_send to mock_send_result - Wrap mock return_value with Result(data="Task completed") - Add "from src.result_types import Result" import All 3 tests in test_spawn_interception_v2.py pass.
This commit is contained in:
@@ -3,6 +3,7 @@ from unittest.mock import MagicMock, patch
|
|||||||
from src import multi_agent_conductor
|
from src import multi_agent_conductor
|
||||||
from src.models import Ticket, WorkerContext
|
from src.models import Ticket, WorkerContext
|
||||||
from src import events
|
from src import events
|
||||||
|
from src.result_types import Result
|
||||||
import threading
|
import threading
|
||||||
from typing import Generator
|
from typing import Generator
|
||||||
|
|
||||||
@@ -19,9 +20,9 @@ class MockDialog:
|
|||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def mock_ai_client() -> Generator[MagicMock, None, None]:
|
def mock_ai_client() -> Generator[MagicMock, None, None]:
|
||||||
with patch("src.ai_client.send") as mock_send:
|
with patch("src.ai_client.send_result") as mock_send_result:
|
||||||
mock_send.return_value = "Task completed"
|
mock_send_result.return_value = Result(data="Task completed")
|
||||||
yield mock_send
|
yield mock_send_result
|
||||||
|
|
||||||
def test_confirm_spawn_pushed_to_queue() -> None:
|
def test_confirm_spawn_pushed_to_queue() -> None:
|
||||||
event_queue = events.SyncEventQueue()
|
event_queue = events.SyncEventQueue()
|
||||||
|
|||||||
Reference in New Issue
Block a user