From b0a3b6ca16d6149355ec7c4d273cdbb627d92cc1 Mon Sep 17 00:00:00 2001 From: Ruben Date: Tue, 5 May 2026 17:22:36 +0300 Subject: [PATCH] Swap out use of depricated wait utils for wait_strategies Use `wait_strategies` to wait for container ready logs --- modules/neo4j/testcontainers/neo4j/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/modules/neo4j/testcontainers/neo4j/__init__.py b/modules/neo4j/testcontainers/neo4j/__init__.py index 7939c013f..4ac3387ac 100644 --- a/modules/neo4j/testcontainers/neo4j/__init__.py +++ b/modules/neo4j/testcontainers/neo4j/__init__.py @@ -18,8 +18,7 @@ from testcontainers.core.config import testcontainers_config as c from testcontainers.core.generic import DbContainer from testcontainers.core.utils import raise_for_deprecated_parameter -from testcontainers.core.waiting_utils import wait_container_is_ready, wait_for_logs - +from testcontainers.core.wait_strategies import LogMessageWaitStrategy class Neo4jContainer(DbContainer): """ @@ -60,9 +59,10 @@ def _configure(self) -> None: def get_connection_url(self) -> str: return f"bolt://{self.get_container_host_ip()}:{self.get_exposed_port(self.port)}" - @wait_container_is_ready() def _connect(self) -> None: - wait_for_logs(self, "Remote interface available at", c.timeout) + wait_strategy = LogMessageWaitStrategy("Remote interface available at") + wait_strategy.with_startup_timeout(c.timeout) + wait_strategy.wait_until_ready(self) # Then we actually check that the container really is listening with self.get_driver() as driver: