Docker-compose --build support

testcontainers/compose.py

@@ -3,19 +3,21 @@
 import blindspin
 import requests
 
-from testcontainers.core.waiting_utils import wait_container_is_ready
 from testcontainers.core.exceptions import NoSuchPortExposed
+from testcontainers.core.waiting_utils import wait_container_is_ready
 
 
 class DockerCompose(object):
     def __init__(
             self,
             filepath,
             compose_file_name="docker-compose.yml",
-            pull=False):
+            pull=False,
+            build=False):
         self.filepath = filepath
         self.compose_file_name = compose_file_name
         self.pull = pull
+        self.build = build
 
     def __enter__(self):
         self.start()
@@ -26,11 +28,15 @@ def __exit__(self, exc_type, exc_val, exc_tb):
 
     def start(self):
         with blindspin.spinner():
+            cmd = ["docker-compose", "-f", self.compose_file_name]
             if self.pull:
-                subprocess.call(["docker-compose", "-f", self.compose_file_name, "pull"],
-                                cwd=self.filepath)
-            subprocess.call(["docker-compose", "-f", self.compose_file_name, "up", "-d"],
-                            cwd=self.filepath)
+                subprocess.call(cmd + ["pull"], cwd=self.filepath)
+
+            cmd += ["up", "-d"]
+            if self.build:
+                cmd += ["--build"]
+
+            subprocess.call(cmd, cwd=self.filepath)
 
     def stop(self):
         with blindspin.spinner():

tests/__init__.py

@@ -0,0 +1,3 @@
+from os.path import realpath, dirname
+
+TESTS_DIR = realpath(dirname(__file__))

tests/test_docker_compose.py

@@ -3,31 +3,40 @@
 from testcontainers.compose import DockerCompose
 from testcontainers.core.exceptions import NoSuchPortExposed
 from testcontainers.core.utils import inside_container
+from tests import TESTS_DIR
 
 
 def test_can_spawn_service_via_compose():
-    with DockerCompose("tests") as compose:
+    with DockerCompose(TESTS_DIR) as compose:
         host = compose.get_service_host("hub", 4444)
         port = compose.get_service_port("hub", 4444)
         assert host == "0.0.0.0"
         assert port == "4444"
 
 
 def test_can_pull_images_before_spawning_service_via_compose():
-    with DockerCompose("tests", pull=True) as compose:
+    with DockerCompose(TESTS_DIR, pull=True) as compose:
+        host = compose.get_service_host("hub", 4444)
+        port = compose.get_service_port("hub", 4444)
+        assert host == "0.0.0.0"
+        assert port == "4444"
+
+
+def test_can_build_images_before_spawning_service_via_compose():
+    with DockerCompose(TESTS_DIR, build=True) as compose:
         host = compose.get_service_host("hub", 4444)
         port = compose.get_service_port("hub", 4444)
         assert host == "0.0.0.0"
         assert port == "4444"
 
 
 def test_can_throw_exception_if_no_port_exposed():
-    with DockerCompose("tests") as compose:
+    with DockerCompose(TESTS_DIR) as compose:
         with pytest.raises(NoSuchPortExposed):
             compose.get_service_host("hub", 5555)
 
 
 def test_compose_wait_for_container_ready():
-    with DockerCompose("tests") as compose:
+    with DockerCompose(TESTS_DIR) as compose:
         host = "host.docker.internal" if inside_container() else "localhost"
         compose.wait_for("http://%s:4444/wd/hub" % host)