docker-compose support added

Author: gospodinkot

Pipfile.lock

@@ -0,0 +1,38 @@
+import subprocess
+
+import blindspin
+
+from testcontainers.core.exceptions import NoSuchPortExposed
+
+
+class DockerCompose(object):
+    def __init__(self, filepath):
+        self.filepath = filepath
+
+    def __enter__(self):
+        return self.start()
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.stop()
+
+    def start(self):
+        with blindspin.spinner():
+            subprocess.call(["docker-compose", "up", "-d"], cwd=self.filepath)
+
+    def stop(self):
+        with blindspin.spinner():
+            subprocess.call(["docker-compose", "down"], cwd=self.filepath)
+
+    def get_service_port(self, service_name, port):
+        return self._get_service_info(service_name, port)[1]
+
+    def get_service_host(self, service_name, port):
+        return self._get_service_info(service_name, port)[0]
+
+    def _get_service_info(self, service, port):
+        output = subprocess.check_output(["docker-compose", "port", service, str(port)],
+                                         cwd=self.filepath).decode("utf-8")
+        result = str(output).rstrip().split(":")
+        if len(result) == 1:
+            raise NoSuchPortExposed("Port {} was not exposed for service {}".format(port, service))
+        return result

testcontainers/compose.py

@@ -22,3 +22,7 @@ class TimeoutException(Exception):
 
 class NoSuchBrowserException(Exception):
     pass
+
+
+class NoSuchPortExposed(Exception):
+    pass

testcontainers/core/exceptions.py

@@ -0,0 +1,16 @@
+import os
+
+from testcontainers.compose import DockerCompose
+
+
+def test_can_spawn_service_via_compose():
+    compose = DockerCompose(os.path.dirname(__file__))
+
+    try:
+        compose.start()
+        host = compose.get_service_host("hub", 4444)
+        port = compose.get_service_port("hub", 4444)
+        assert host == "0.0.0.0"
+        assert port == "4444"
+    finally:
+        compose.stop()