docker-compose wait_for reade method added

Author: SergeyPirogov

testcontainers/compose.py

@@ -1,7 +1,9 @@
 import subprocess
 
 import blindspin
+import requests
 
+from testcontainers import wait_container_is_ready
 from testcontainers.core.exceptions import NoSuchPortExposed
 
 
@@ -36,3 +38,8 @@ def _get_service_info(self, service, port):
         if len(result) == 1:
             raise NoSuchPortExposed("Port {} was not exposed for service {}".format(port, service))
         return result
+
+    @wait_container_is_ready()
+    def wait_for(self, url):
+        requests.get(url)
+        return self

tests/test_docker_compose.py

@@ -1,6 +1,7 @@
 import os
 
 import pytest
+from selenium import webdriver
 
 from testcontainers.compose import DockerCompose
 from testcontainers.core.exceptions import NoSuchPortExposed
@@ -27,3 +28,9 @@ def test_can_throw_exception_if_no_port_exposed():
         compose.get_service_host("hub", 5555)
 
     compose.stop()
+
+
+def test_compose_wait_for_container_ready():
+    compose = DockerCompose(os.path.dirname(__file__))
+    with compose:
+        compose.wait_for("http://localhost:4444/wd/hub")