com.github.nscuro.wdm.factory.RemoteWebDriverFactory Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of webdriver-manager Show documentation

Simplifying WebDriver instantiation

There is a newer version: 0.3.0

package com.github.nscuro.wdm.factory;

import org.openqa.selenium.Capabilities;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.remote.RemoteWebDriver;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Optional;

public final class RemoteWebDriverFactory implements WebDriverFactory {

    private final URL gridHubUrl;

    /**
     * @param gridHubUrl URL of the Selenium grid hub
     * @throws IllegalArgumentException When the given URL is invalid or no
     *                                  URL was provided at all
     */
    public RemoteWebDriverFactory(final String gridHubUrl) {
        this.gridHubUrl = requireValidUrl(gridHubUrl);
    }

    /**
     * {@inheritDoc}
     */
    @Nonnull
    @Override
    public WebDriver getWebDriver(final Capabilities capabilities) {
        return new RemoteWebDriver(gridHubUrl, capabilities);
    }

    private static URL requireValidUrl(@Nullable final String url) {
        return Optional.ofNullable(url)
                .map(urlStr -> {
                    try {
                        return new URL(urlStr);
                    } catch (MalformedURLException e) {
                        throw new IllegalArgumentException(e);
                    }
                })
                .orElseThrow(() -> new IllegalArgumentException("No Grid Hub URL provided"));
    }

}