From 40e12214cf8da4acd5a9aae6f9854bb148f3302d Mon Sep 17 00:00:00 2001 From: Jason Woltje Date: Tue, 3 Mar 2026 02:36:36 +0000 Subject: [PATCH] fix(test): make queue completion test more robust (#663) Co-authored-by: Jason Woltje Co-committed-by: Jason Woltje --- 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