Make it easier to use different Selenium images

Author: AB-xdev

testcontainers-selenium/src/main/java/software/xdev/testcontainers/selenium/containers/browser/CapabilitiesBrowserWebDriverContainer.java

@@ -37,15 +37,38 @@ public class CapabilitiesBrowserWebDriverContainer<SELF extends CapabilitiesBrow
 
 	public CapabilitiesBrowserWebDriverContainer(final Capabilities capabilities)
 	{
-		super(getStandardImageForCapabilities(capabilities, SeleniumUtils.getClasspathSeleniumVersion()));
+		this(capabilities, BROWSER_DOCKER_IMAGES);
+	}
+	
+	public CapabilitiesBrowserWebDriverContainer(
+		final Capabilities capabilities,
+		final Map<String, DockerImageName> browserDockerImages)
+	{
+		this(getStandardImageForCapabilities(
+			capabilities,
+			SeleniumUtils.getClasspathSeleniumVersion(),
+			browserDockerImages));
+	}
+	
+	public CapabilitiesBrowserWebDriverContainer(final DockerImageName dockerImageName)
+	{
+		super(dockerImageName);
 		this.waitStrategy = this.getDefaultWaitStrategy();
 	}
 
 	protected static DockerImageName getStandardImageForCapabilities(
 		final Capabilities capabilities,
 		final String seleniumVersion)
 	{
-		return Optional.ofNullable(BROWSER_DOCKER_IMAGES.get(Optional.ofNullable(capabilities)
+		return getStandardImageForCapabilities(capabilities, seleniumVersion, BROWSER_DOCKER_IMAGES);
+	}
+	
+	protected static DockerImageName getStandardImageForCapabilities(
+		final Capabilities capabilities,
+		final String seleniumVersion,
+		final Map<String, DockerImageName> browserDockerImages)
+	{
+		return Optional.ofNullable(browserDockerImages.get(Optional.ofNullable(capabilities)
 				.map(Capabilities::getBrowserName)
 				.orElse(BrowserType.CHROME)))
 			.map(image -> image.withTag(seleniumVersion))