Fix test hangs

zauberzeug · Apr 16, 2024 · c23643b · c23643b · me21 · Apr 16, 2024
1 parent 755187c
commit c23643b
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/nicegui/outbox.py b/nicegui/outbox.py
@@ -56,10 +56,11 @@ async def loop(self) -> None:
 
         while not self._should_stop:
             try:
-                try:
-                    await asyncio.wait_for(self._enqueue_event.wait(), timeout=1.0)
-                except (TimeoutError, asyncio.TimeoutError):
-                    continue
+                if not self._enqueue_event.is_set():
+                    try:
+                        await asyncio.wait_for(self._enqueue_event.wait(), timeout=1.0)
+                    except (TimeoutError, asyncio.TimeoutError):
+                        continue
 
                 if not self.client.has_socket_connection:
                     await asyncio.sleep(0.1)