docker-compose added

Author: SergeyPirogov

Pipfile.lock

@@ -0,0 +1,41 @@
+Docker compose support
+======================
+
+Allows to spin up services configured via docker-compose.yml
+
+Example docker-compose.yml for grid
+-----------------------------------
+
+::
+
+    hub:
+      image: selenium/hub
+    ports:
+      - "4444:4444"
+    firefox:
+      image: selenium/node-firefox
+      links:
+        - hub
+      expose:
+        - "5555"
+    chrome:
+      image: selenium/node-chrome
+      links:
+        - hub
+      expose:
+        - "5555"
+
+Code
+----
+
+::
+
+    compose = DockerCompose("/home/project")
+    with compose:
+        host = compose.get_service_host("hub", 4444)
+        port = compose.get_service_port("hub", 4444)
+        driver = webdriver.Remote(
+            command_executor=("http://{}:{}/wd/hub".format(host,port)),
+            desired_capabilities=CHROME)
+        driver.get("http://automation-remarks.com")
+

docs/compose.rst

@@ -44,4 +44,4 @@ Usage modes
    Database containers <database>
    Selenium containers <selenium>
    Generic containers <generic> 
-
+   Docker Compose <compose>

docs/index.rst

@@ -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,29 @@
+import os
+
+import pytest
+
+from testcontainers.compose import DockerCompose
+from testcontainers.core.exceptions import NoSuchPortExposed
+
+
+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()
+
+
+def test_can_throw_exception_if_no_port_exposed():
+    compose = DockerCompose(os.path.dirname(__file__))
+
+    compose.start()
+    with pytest.raises(NoSuchPortExposed):
+        compose.get_service_host("hub", 5555)
+
+    compose.stop()