Add test for CustomContainer

Author: Tranquility2

core/tests/test_core.py

@@ -4,6 +4,7 @@
 from testcontainers.core.container import DockerContainer
 from testcontainers.core.image import DockerImage
 from testcontainers.core.waiting_utils import wait_for_logs
+from testcontainers.core.generic import CustomContainer
 
 
 def test_timeout_is_raised_when_waiting_for_logs():
@@ -35,7 +36,7 @@ def test_can_get_logs():
         assert stdout, "There should be something on stdout"
 
 
-def test_from_image():
+def test_docker_image():
     test_image_tag = "test-image:latest"
     with tempfile.TemporaryDirectory() as temp_directory:
         with open(f"{temp_directory}/Dockerfile", "w") as f:
@@ -54,3 +55,21 @@ def test_from_image():
                 assert logs[3] == {"stream": 'Step 2/2 : CMD echo "Hello from Docker Image!"'}
                 with DockerContainer(image.tag) as container:
                     wait_for_logs(container, "Hello from Docker Image!")
+
+
+def test_custom_container():
+    with tempfile.TemporaryDirectory() as temp_directory:
+        with open(f"{temp_directory}/Dockerfile", "w") as f:
+            f.write(
+                """
+                FROM alpine:latest
+                CMD echo "This is a custom Docker Image!"
+                """
+            )
+            f.close()
+            with CustomContainer(tag="test-image:latest", path=temp_directory) as server:
+                image_build_logs = server.image_obj.get_logs()
+                assert isinstance(image_build_logs, list)
+                assert image_build_logs[0] == {"stream": "Step 1/2 : FROM alpine:latest"}
+                assert image_build_logs[3] == {"stream": 'Step 2/2 : CMD echo "This is a custom Docker Image!"'}
+                wait_for_logs(server, "This is a custom Docker Image!")