11from dataclasses import asdict , dataclass , field , fields , is_dataclass
22from functools import cached_property
33from json import loads
4- from logging import warning
4+ from logging import getLogger , warning
55from os import PathLike
66from platform import system
77from re import split
8- from subprocess import CompletedProcess
8+ from subprocess import CalledProcessError , CompletedProcess
99from subprocess import run as subprocess_run
1010from types import TracebackType
1111from typing import Any , Callable , Literal , Optional , TypeVar , Union , cast
12- from urllib .error import HTTPError , URLError
13- from urllib .request import urlopen
1412
1513from testcontainers .core .exceptions import ContainerIsNotRunning , NoSuchPortExposed
16- from testcontainers .core .waiting_utils import wait_container_is_ready
14+ from testcontainers .core .waiting_utils import WaitStrategy
1715
1816_IPT = TypeVar ("_IPT" )
1917_WARNINGS = {"DOCKER_COMPOSE_GET_CONFIG" : "get_config is experimental, see testcontainers/testcontainers-python#669" }
2018
19+ logger = getLogger (__name__ )
20+
2121
2222def _ignore_properties (cls : type [_IPT ], dict_ : Any ) -> _IPT :
2323 """omits extra fields like @JsonIgnoreProperties(ignoreUnknown = true)
@@ -80,6 +80,7 @@ class ComposeContainer:
8080 Health : Optional [str ] = None
8181 ExitCode : Optional [int ] = None
8282 Publishers : list [PublishedPortModel ] = field (default_factory = list )
83+ _docker_compose : Optional ["DockerCompose" ] = field (default = None , init = False , repr = False )
8384
8485 def __post_init__ (self ) -> None :
8586 if self .Publishers :
@@ -116,6 +117,41 @@ def _matches_protocol(prefer_ip_version: str, r: PublishedPortModel) -> bool:
116117 r_url = r .URL
117118 return (r_url is not None and ":" in r_url ) is (prefer_ip_version == "IPv6" )
118119
120+ # WaitStrategy compatibility methods
121+ def get_container_host_ip (self ) -> str :
122+ """Get the host IP for the container."""
123+ # Simplified implementation - wait strategies don't use this yet
124+ return "127.0.0.1"
125+
126+ def get_exposed_port (self , port : int ) -> int :
127+ """Get the exposed port mapping for the given internal port."""
128+ # Simplified implementation - wait strategies don't use this yet
129+ return port
130+
131+ def get_logs (self ) -> tuple [bytes , bytes ]:
132+ """Get container logs."""
133+ if not self ._docker_compose :
134+ raise RuntimeError ("DockerCompose reference not set on ComposeContainer" )
135+ if not self .Service :
136+ raise RuntimeError ("Service name not set on ComposeContainer" )
137+ stdout , stderr = self ._docker_compose .get_logs (self .Service )
138+ return stdout .encode (), stderr .encode ()
139+
140+ def get_wrapped_container (self ) -> "ComposeContainer" :
141+ """Get the underlying container object for compatibility."""
142+ return self
143+
144+ def reload (self ) -> None :
145+ """Reload container information for compatibility with wait strategies."""
146+ # ComposeContainer doesn't need explicit reloading as it's fetched fresh
147+ # each time through get_container(), but we need this method for compatibility
148+ pass
149+
150+ @property
151+ def status (self ) -> str :
152+ """Get container status for compatibility with wait strategies."""
153+ return self .State or "unknown"
154+
119155
120156@dataclass
121157class DockerCompose :
@@ -178,6 +214,7 @@ class DockerCompose:
178214 services : Optional [list [str ]] = None
179215 docker_command_path : Optional [str ] = None
180216 profiles : Optional [list [str ]] = None
217+ _wait_strategies : Optional [dict [str , Any ]] = field (default = None , init = False , repr = False )
181218
182219 def __post_init__ (self ) -> None :
183220 if isinstance (self .compose_file_name , str ):
@@ -216,6 +253,15 @@ def compose_command_property(self) -> list[str]:
216253 docker_compose_cmd += ["--env-file" , env_file ]
217254 return docker_compose_cmd
218255
256+ def waiting_for (self , strategies : dict [str , WaitStrategy ]) -> "DockerCompose" :
257+ """
258+ Set wait strategies for specific services.
259+ Args:
260+ strategies: Dictionary mapping service names to wait strategies
261+ """
262+ self ._wait_strategies = strategies
263+ return self
264+
219265 def start (self ) -> None :
220266 """
221267 Starts the docker compose environment.
@@ -244,6 +290,11 @@ def start(self) -> None:
244290
245291 self ._run_command (cmd = up_cmd )
246292
293+ if self ._wait_strategies :
294+ for service , strategy in self ._wait_strategies .items ():
295+ container = self .get_container (service_name = service )
296+ strategy .wait_until_ready (container )
297+
247298 def stop (self , down : bool = True ) -> None :
248299 """
249300 Stops the docker compose environment.
@@ -320,7 +371,7 @@ def get_containers(self, include_all: bool = False) -> list[ComposeContainer]:
320371 result = self ._run_command (cmd = cmd )
321372 stdout = split (r"\r?\n" , result .stdout .decode ("utf-8" ))
322373
323- containers = []
374+ containers : list [ ComposeContainer ] = []
324375 # one line per service in docker 25, single array for docker 24.0.2
325376 for line in stdout :
326377 if not line :
@@ -331,6 +382,10 @@ def get_containers(self, include_all: bool = False) -> list[ComposeContainer]:
331382 else :
332383 containers .append (_ignore_properties (ComposeContainer , data ))
333384
385+ # Set the docker_compose reference on each container
386+ for container in containers :
387+ container ._docker_compose = self
388+
334389 return containers
335390
336391 def get_container (
@@ -355,6 +410,7 @@ def get_container(
355410 if not matching_containers :
356411 raise ContainerIsNotRunning (f"{ service_name } is not running in the compose context" )
357412
413+ matching_containers [0 ]._docker_compose = self
358414 return matching_containers [0 ]
359415
360416 def exec_in_container (
@@ -391,12 +447,18 @@ def _run_command(
391447 context : Optional [str ] = None ,
392448 ) -> CompletedProcess [bytes ]:
393449 context = context or str (self .context )
394- return subprocess_run (
395- cmd ,
396- capture_output = True ,
397- check = True ,
398- cwd = context ,
399- )
450+ try :
451+ return subprocess_run (
452+ cmd ,
453+ capture_output = True ,
454+ check = True ,
455+ cwd = context ,
456+ )
457+ except CalledProcessError as e :
458+ logger .error (f"Command '{ e .cmd } ' failed with exit code { e .returncode } " )
459+ logger .error (f"STDOUT:\n { e .stdout .decode (errors = 'ignore' )} " )
460+ logger .error (f"STDERR:\n { e .stderr .decode (errors = 'ignore' )} " )
461+ raise e from e
400462
401463 def get_service_port (
402464 self ,
@@ -455,16 +517,54 @@ def get_service_host_and_port(
455517 publisher = self .get_container (service_name ).get_publisher (by_port = port ).normalize ()
456518 return publisher .URL , publisher .PublishedPort
457519
458- @wait_container_is_ready (HTTPError , URLError )
459520 def wait_for (self , url : str ) -> "DockerCompose" :
460521 """
461522 Waits for a response from a given URL. This is typically used to block until a service in
462523 the environment has started and is responding. Note that it does not assert any sort of
463524 return code, only check that the connection was successful.
464525
526+ This is a convenience method that internally uses HttpWaitStrategy. For more complex
527+ wait scenarios, consider using the structured wait strategies with `waiting_for()`.
528+
465529 Args:
466530 url: URL from one of the services in the environment to use to wait on.
531+
532+ Example:
533+ # Simple URL wait (legacy style)
534+ compose.wait_for("http://localhost:8080") \
535+ \
536+ # For more complex scenarios, use structured wait strategies:
537+ from testcontainers.core.waiting_utils import HttpWaitStrategy, LogMessageWaitStrategy \
538+ \
539+ compose.waiting_for({ \
540+ "web": HttpWaitStrategy(8080).for_status_code(200), \
541+ "db": LogMessageWaitStrategy("database system is ready to accept connections") \
542+ })
467543 """
544+ import time
545+ from urllib .error import HTTPError , URLError
546+ from urllib .request import Request , urlopen
547+
548+ # For simple URL waiting when we have multiple containers,
549+ # we'll do a direct HTTP check instead of using the container-based strategy
550+ start_time = time .time ()
551+ timeout = 120 # Default timeout
552+
553+ while True :
554+ if time .time () - start_time > timeout :
555+ raise TimeoutError (f"URL { url } not ready within { timeout } seconds" )
556+
557+ try :
558+ request = Request (url , method = "GET" )
559+ with urlopen (request , timeout = 1 ) as response :
560+ if 200 <= response .status < 400 :
561+ return self
562+ except (URLError , HTTPError , ConnectionResetError , ConnectionRefusedError , BrokenPipeError , OSError ):
563+ # Any connection error means we should keep waiting
564+ pass
565+
566+ time .sleep (1 )
567+
468568 with urlopen (url ) as response :
469569 response .read ()
470570 return self
0 commit comments