fixes #122 (#123)

* fixes #122

* fixes #122

Author: shikhar08

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)

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"