Improved ready logs

celery · Oct 12, 2023 · 1b0ec5f · 1b0ec5f
1 parent fb786a2
commit 1b0ec5f
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/src/pytest_celery/api/container.py b/src/pytest_celery/api/container.py
@@ -1,3 +1,4 @@
+import sys
 from typing import Any
 from typing import Optional
 
@@ -38,14 +39,14 @@ def _wait_port(self, port: str) -> int:
 
     @retry(pytest_docker_tools.exceptions.TimeoutError, delay=10, tries=3)
     def _wait_ready(self, timeout: int = 30) -> bool:
-        if self.ready_prompt is None:
-            return True
-
-        wait_for_callable(
-            f"Waiting for {self.__class__.__name__}::{self.name} to get ready",
-            lambda: self.ready_prompt in self.logs(),
-            timeout=timeout,
-        )
+        if self.ready_prompt is not None:
+            wait_for_callable(
+                f"Waiting for {self.__class__.__name__}::{self.name} to get ready",
+                lambda: self.ready_prompt in self.logs(),
+                timeout=timeout,
+            )
+
+        sys.stdout.write(f"{self.__class__.__name__}::{self.name} is ready\n")
 
         return True