Precompute TIMEOUT.

Author: tillahoffmann

arangodb/testcontainers/arangodb/__init__.py

@@ -2,7 +2,7 @@
 ArangoDB container support.
 """
 from os import environ
-from testcontainers.core.config import MAX_TRIES
+from testcontainers.core.config import TIMEOUT
 from testcontainers.core.generic import DbContainer
 from testcontainers.core.waiting_utils import wait_for_logs
 import typing
@@ -81,4 +81,4 @@ def get_connection_url(self) -> str:
         return f"http://{self.get_container_host_ip()}:{port}"
 
     def _connect(self) -> None:
-        wait_for_logs(self, predicate="is ready for business", timeout=MAX_TRIES)
+        wait_for_logs(self, predicate="is ready for business", timeout=TIMEOUT)

core/testcontainers/core/config.py

@@ -2,3 +2,4 @@
 
 MAX_TRIES = int(environ.get("TC_MAX_TRIES", 120))
 SLEEP_TIME = int(environ.get("TC_POOLING_INTERVAL", 1))
+TIMEOUT = MAX_TRIES * SLEEP_TIME

core/testcontainers/core/waiting_utils.py

@@ -64,8 +64,8 @@ def wrapper(wrapped: Callable, instance: Any, args: Iterable, kwargs: Mapping) -
                 time.sleep(config.SLEEP_TIME)
                 exception = e
         raise TimeoutError(
-            f'Wait time ({config.MAX_TRIES * config.SLEEP_TIME}s) exceeded for {wrapped.__name__}'
-            f'(args: {args}, kwargs {kwargs}). Exception: {exception}'
+            f'Wait time ({config.TIMEOUT}s) exceeded for {wrapped.__name__}(args: {args}, kwargs '
+            f'{kwargs}). Exception: {exception}'
         )
 
     return wrapper