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

Go to download
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 com.github.nscuro.wdm.Browser;
import com.github.nscuro.wdm.binary.BinaryManager;
import org.openqa.selenium.Capabilities;
import org.openqa.selenium.WebDriver;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.io.File;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.util.Optional;

import static java.lang.String.format;

/**
 * A {@link WebDriverFactory} that creates local {@link WebDriver} instances.
 */
public final class LocalWebDriverFactory implements WebDriverFactory {

    private final BinaryManager binaryManager;

    private final WebDriverFactoryConfig config;

    /**
     * Create a {@link LocalWebDriverFactory} that is able to download binaries if necessary.
     *
     * @param binaryManager The {@link BinaryManager} to use for downloading binaries
     * @param config        The {@link WebDriverFactoryConfig} to use
     */
    public LocalWebDriverFactory(@Nullable final BinaryManager binaryManager,
                                 final WebDriverFactoryConfig config) {
        this.binaryManager = binaryManager;
        this.config = config;
    }

    /**
     * Create a {@link LocalWebDriverFactory} that is able to download binaries if necessary.
     *
     * @param binaryManager The {@link BinaryManager} to use for downloading binaries
     */
    public LocalWebDriverFactory(@Nullable final BinaryManager binaryManager) {
        this(binaryManager, new WebDriverFactoryConfig());
    }

    /**
     * Create a {@link LocalWebDriverFactory} without {@link BinaryManager}.
     * 
     * Note that the resulting {@link LocalWebDriverFactory} instance will not be
     * able to download any binaries, so you might only be able to create instances
     * of {@link WebDriver} for browsers that do not require a separate binary.
     *
     * @since 0.1.3
     */
    public LocalWebDriverFactory() {
        this(null);
    }

    /**
     * Get a {@link WebDriver} instance based on the given {@link Capabilities}.
     * 

     * Per default, {@link LocalWebDriverFactory} will always attempt to download
     * the latest version of a binary when necessary. This behavior can be altered
     * by providing a {@link WebDriverFactoryConfig} at construction which specifies
     * a specific version on a per-browser basis.
     * 
     * Note that {@link Capabilities#getPlatform()} will be ignored.
     * Instead, the current operating system and architecture will be detected
     * and a matching binary will be downloaded (if possible and / or necessary).
     *
     * @param capabilities The desired capabilities
     * @return A {@link WebDriver} instance
     * @throws WebDriverFactoryException When creating the {@link WebDriver} instance failed
     * @throws IllegalArgumentException  When the given {@link Capabilities} do not specify a browser name
     * @see LocalWebDriverFactory#LocalWebDriverFactory(BinaryManager)
     * @see LocalWebDriverFactory#LocalWebDriverFactory(BinaryManager, WebDriverFactoryConfig)
     */
    @Nonnull
    @Override
    public WebDriver createWebDriver(final Capabilities capabilities) {
        final Browser browser = Optional.ofNullable(capabilities.getBrowserName())
                .filter(browserName -> !browserName.isEmpty())
                .map(Browser::byName)
                .orElseThrow(() -> new IllegalArgumentException("No browser name specified"));

        if (browser.doesRequireBinary()) {
            if (binaryManager == null) {
                throw new WebDriverFactoryException(capabilities,
                        format("WebDriver for Browser \"%s\" requires a binary, but no %s was provided when constructing %s",
                                browser, BinaryManager.class.getSimpleName(), LocalWebDriverFactory.class.getSimpleName()));
            }

            final File webDriverBinary;

            final Optional desiredVersion = config.getBinaryVersionForBrowser(browser);

            try {
                if (desiredVersion.isPresent()) {
                    webDriverBinary = binaryManager.getWebDriverBinary(browser, desiredVersion.get());
                } else {
                    webDriverBinary = binaryManager.getLatestWebDriverBinary(browser);
                }
            } catch (IOException e) {
                throw new WebDriverFactoryException(capabilities, e);
            }

            binaryManager.registerWebDriverBinary(browser, webDriverBinary);
        }

        return getWebDriverInstance(browser, capabilities);
    }

    private WebDriver getWebDriverInstance(final Browser browser, final Capabilities capabilities) {
        try {
            return Class
                    .forName(browser.getWebDriverClassName())
                    .asSubclass(WebDriver.class)
                    .getConstructor(Capabilities.class)
                    .newInstance(capabilities);
        } catch (NoSuchMethodException | IllegalAccessException | InstantiationException | InvocationTargetException | ClassNotFoundException e) {
            throw new WebDriverFactoryException(capabilities, e);
        }
    }

}