fixes testcontainers#122 (testcontainers#123)

shikhar08 · web-flow · commit f5113fb1bb77 · 2020-12-08T09:32:12.000Z
* fixes testcontainers#122 * fixes testcontainers#122
diff --git a/testcontainers/selenium.py b/testcontainers/selenium.py
@@ -49,8 +49,7 @@ class BrowserWebDriverContainer(DockerContainer):
     """
     def __init__(self, capabilities, image=None):
         self.capabilities = capabilities
-        if not image:
-            self.image = get_image_name(capabilities)
+        self.image = image or get_image_name(capabilities)
         self.port_to_expose = 4444
         self.vnc_port_to_expose = 5900
         super(BrowserWebDriverContainer, self).__init__(image=self.image)
diff --git a/tests/test_selenium.py b/tests/test_selenium.py
@@ -0,0 +1,9 @@
+
+def test_selenium_custom_image():
+    from testcontainers.selenium import BrowserWebDriverContainer
+    from selenium.webdriver import DesiredCapabilities
+
+    image = "selenium/standalone-chrome:latest"
+    chrome = BrowserWebDriverContainer(DesiredCapabilities.CHROME, image=image)
+    assert "image" in dir(chrome), "`image` attribute was not instantialized."
+    assert chrome.image == image, "`image` attribute was not set to the user provided value"
