Skip to content

Commit 6deaf3b

Browse files
authored
fix(neo4j): Swap out use of depricated wait utils for wait_strategies (#1012)
Use `wait_strategies` to wait for container ready logs
1 parent 79d920e commit 6deaf3b

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

modules/neo4j/testcontainers/neo4j/__init__.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,7 @@
1818
from testcontainers.core.config import testcontainers_config as c
1919
from testcontainers.core.generic import DbContainer
2020
from testcontainers.core.utils import raise_for_deprecated_parameter
21-
from testcontainers.core.waiting_utils import wait_container_is_ready, wait_for_logs
22-
21+
from testcontainers.core.wait_strategies import LogMessageWaitStrategy
2322

2423
class Neo4jContainer(DbContainer):
2524
"""
@@ -60,9 +59,10 @@ def _configure(self) -> None:
6059
def get_connection_url(self) -> str:
6160
return f"bolt://{self.get_container_host_ip()}:{self.get_exposed_port(self.port)}"
6261

63-
@wait_container_is_ready()
6462
def _connect(self) -> None:
65-
wait_for_logs(self, "Remote interface available at", c.timeout)
63+
wait_strategy = LogMessageWaitStrategy("Remote interface available at")
64+
wait_strategy.with_startup_timeout(c.timeout)
65+
wait_strategy.wait_until_ready(self)
6666

6767
# Then we actually check that the container really is listening
6868
with self.get_driver() as driver:

0 commit comments

Comments
 (0)