fix: Handle None event_queue in _queue_put, fix test assertion
This commit is contained in:
@@ -271,7 +271,8 @@ class ConductorEngine:
|
|||||||
|
|
||||||
def _queue_put(event_queue: events.SyncEventQueue, event_name: str, payload) -> None:
|
def _queue_put(event_queue: events.SyncEventQueue, event_name: str, payload) -> None:
|
||||||
"""Thread-safe helper to push an event to the SyncEventQueue from a worker thread."""
|
"""Thread-safe helper to push an event to the SyncEventQueue from a worker thread."""
|
||||||
event_queue.put(event_name, payload)
|
if event_queue is not None:
|
||||||
|
event_queue.put(event_name, payload)
|
||||||
|
|
||||||
def confirm_execution(payload: str, event_queue: events.SyncEventQueue, ticket_id: str) -> bool:
|
def confirm_execution(payload: str, event_queue: events.SyncEventQueue, ticket_id: str) -> bool:
|
||||||
"""
|
"""
|
||||||
|
|||||||
@@ -282,8 +282,8 @@ def test_run_worker_lifecycle_pushes_response_via_queue(monkeypatch: pytest.Monk
|
|||||||
patch("src.multi_agent_conductor._queue_put") as mock_queue_put:
|
patch("src.multi_agent_conductor._queue_put") as mock_queue_put:
|
||||||
mock_spawn.return_value = (True, "prompt", "context")
|
mock_spawn.return_value = (True, "prompt", "context")
|
||||||
run_worker_lifecycle(ticket, context, event_queue=mock_event_queue)
|
run_worker_lifecycle(ticket, context, event_queue=mock_event_queue)
|
||||||
mock_queue_put.assert_called_once()
|
mock_queue_put.assert_called()
|
||||||
call_args = mock_queue_put.call_args[0]
|
call_args = mock_queue_put.call_args_list[0][0]
|
||||||
assert call_args[1] == "response"
|
assert call_args[1] == "response"
|
||||||
assert call_args[2]["stream_id"] == "Tier 3 (Worker): T1"
|
assert call_args[2]["stream_id"] == "Tier 3 (Worker): T1"
|
||||||
assert call_args[2]["text"] == "Task complete."
|
assert call_args[2]["text"] == "Task complete."
|
||||||
|
|||||||
Reference in New Issue
Block a user