From 143c08ea124daab25550792154d6a8647ffa716d Mon Sep 17 00:00:00 2001 From: Jason Woltje Date: Mon, 2 Mar 2026 20:35:42 -0600 Subject: [PATCH] fix(test): make queue completion test more robust with polling loop --- apps/coordinator/tests/test_coordinator.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/apps/coordinator/tests/test_coordinator.py b/apps/coordinator/tests/test_coordinator.py index 8835218..3e84a18 100644 --- a/apps/coordinator/tests/test_coordinator.py +++ b/apps/coordinator/tests/test_coordinator.py @@ -601,9 +601,21 @@ class TestCoordinatorIntegration: coordinator = Coordinator(queue_manager=queue_manager, poll_interval=0.02) task = asyncio.create_task(coordinator.start()) - await asyncio.sleep(0.5) # Allow time for processing - await coordinator.stop() + # Poll for completion with timeout instead of fixed sleep + deadline = asyncio.get_event_loop().time() + 5.0 # 5 second timeout + while asyncio.get_event_loop().time() < deadline: + all_completed = True + for i in range(157, 162): + item = queue_manager.get_item(i) + if item is None or item.status != QueueItemStatus.COMPLETED: + all_completed = False + break + if all_completed: + break + await asyncio.sleep(0.05) + + await coordinator.stop() task.cancel() try: await task