From ca65f2951307e88f9f94dfe282b68d8a9fdd7f24 Mon Sep 17 00:00:00 2001 From: Ed_ Date: Sat, 7 Mar 2026 16:53:45 -0500 Subject: [PATCH] fix: Handle None event_queue in _queue_put, fix test assertion --- src/multi_agent_conductor.py | 3 ++- tests/test_conductor_engine_v2.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/multi_agent_conductor.py b/src/multi_agent_conductor.py index ce9fa65..8b7a9e8 100644 --- a/src/multi_agent_conductor.py +++ b/src/multi_agent_conductor.py @@ -271,7 +271,8 @@ class ConductorEngine: 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.""" - 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: """ diff --git a/tests/test_conductor_engine_v2.py b/tests/test_conductor_engine_v2.py index 5c53e56..bbec12c 100644 --- a/tests/test_conductor_engine_v2.py +++ b/tests/test_conductor_engine_v2.py @@ -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: mock_spawn.return_value = (True, "prompt", "context") run_worker_lifecycle(ticket, context, event_queue=mock_event_queue) - mock_queue_put.assert_called_once() - call_args = mock_queue_put.call_args[0] + mock_queue_put.assert_called() + call_args = mock_queue_put.call_args_list[0][0] assert call_args[1] == "response" assert call_args[2]["stream_id"] == "Tier 3 (Worker): T1" assert call_args[2]["text"] == "Task complete."