Fixed IndexError when accessing the container port info (#417)

celery · Sep 20, 2024 · cadcc9e · cadcc9e
1 parent 00b9b5a
commit cadcc9e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/pytest_celery/api/container.py b/src/pytest_celery/api/container.py
@@ -98,6 +98,12 @@ def ready_prompt(self) -> str | None:
         """
         return None
 
+    @retry(
+        stop=stop_after_attempt(3),
+        wait=wait_fixed(3),
+        retry=retry_if_exception_type(IndexError),
+        reraise=True,
+    )
     def _wait_port(self, port: str) -> int:
         """Blocks until the specified port is ready.