Update image var on CustomContainer

Author: Tranquility2

core/testcontainers/core/generic.py

@@ -94,8 +94,8 @@ class CustomContainer(DockerContainer):
     """
 
     def __init__(self, tag: str, path: str) -> None:
-        self.image_obj = DockerImage(tag, path).build()
-        super().__init__(self.image_obj.tag)
+        self.docker_image = DockerImage(tag, path).build()
+        super().__init__(self.docker_image.tag)
 
     def get_url(self) -> str:
         raise NotImplementedError
@@ -106,4 +106,4 @@ def start(self) -> "CustomContainer":
 
     def stop(self, force=True, delete_volume=True) -> None:
         super().stop(force, delete_volume)
-        self.image_obj.remove()
+        self.docker_image.remove()

core/tests/test_core.py

@@ -68,8 +68,6 @@ def test_custom_container():
             )
             f.close()
             with CustomContainer(tag="custom-image:test", 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"}
+                image_build_logs = server.docker_image.get_logs()
                 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!")