diff --git a/modules/sftp/testcontainers/sftp/__init__.py b/modules/sftp/testcontainers/sftp/__init__.py index 0e073ea1a..1f39b9b83 100644 --- a/modules/sftp/testcontainers/sftp/__init__.py +++ b/modules/sftp/testcontainers/sftp/__init__.py @@ -20,7 +20,7 @@ from cryptography.hazmat.primitives.asymmetric import rsa from testcontainers.core.container import DockerContainer -from testcontainers.core.waiting_utils import wait_for_logs +from testcontainers.core.wait_strategies import LogMessageWaitStrategy if TYPE_CHECKING: from typing_extensions import Self @@ -265,8 +265,9 @@ def _configure(self) -> None: self.with_exposed_ports(self.port) def start(self) -> Self: + strategy = LogMessageWaitStrategy(f".*Server listening on 0.0.0.0 port {self.port}.*") + self.waiting_for(strategy) super().start() - wait_for_logs(self, f".*Server listening on 0.0.0.0 port {self.port}.*") return self def get_exposed_sftp_port(self) -> int: