BrowserWebDriverContainerSeleniumAddressTest.java

package org.testcontainers.containers;

import org.junit.jupiter.api.Test;

import java.net.MalformedURLException;
import java.net.URL;

import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatThrownBy;

class BrowserWebDriverContainerSeleniumAddressTest {

    @Test
    void getSeleniumAddressReturnsUrlBuiltFromHostAndMappedPort() throws MalformedURLException {
        BrowserWebDriverContainer<?> container = new FixedAddressBrowserWebDriverContainer("localhost", 32768);

        assertThat(container.getSeleniumAddress()).isEqualTo(new URL("http", "localhost", 32768, "/wd/hub"));
    }

    @Test
    void getSeleniumAddressThrowsContainerLaunchExceptionWhenUrlCannotBeConstructed() {
        BrowserWebDriverContainer<?> container = new FixedAddressBrowserWebDriverContainer("http://bad", 32768);

        assertThatThrownBy(container::getSeleniumAddress)
            .isInstanceOf(ContainerLaunchException.class)
            .hasMessage("Could not construct Selenium address")
            .hasCauseInstanceOf(MalformedURLException.class);
    }

    private static class FixedAddressBrowserWebDriverContainer
        extends BrowserWebDriverContainer<FixedAddressBrowserWebDriverContainer> {

        private final String host;

        private final int mappedPort;

        FixedAddressBrowserWebDriverContainer(String host, int mappedPort) {
            super();
            this.host = host;
            this.mappedPort = mappedPort;
        }

        @Override
        public String getHost() {
            return host;
        }

        @Override
        public Integer getMappedPort(int originalPort) {
            return mappedPort;
        }
    }
}