com.github.metalloid.webdriver.Instance Maven / Gradle / Ivy

Go to download
package com.github.metalloid.webdriver;

import org.openqa.selenium.Capabilities;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeOptions;
import org.openqa.selenium.edge.EdgeOptions;
import org.openqa.selenium.firefox.FirefoxOptions;
import org.openqa.selenium.ie.InternetExplorerOptions;
import org.openqa.selenium.remote.BrowserType;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.RemoteWebDriver;

import java.lang.reflect.InvocationTargetException;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Optional;

abstract class Instance implements LocalInstance, RemoteInstance {
    static final Boolean HEADLESS = Boolean.parseBoolean(System.getProperty("headless"));
    private static final String BROWSER_NAME = System.getProperty("browser.name");
    private static final String HUB_URL = System.getProperty("webdriver.hub.url");

    WebDriver createRemoteInstance(WebDriverOptions options) {
        if (options == null) {
            DesiredCapabilities desiredCapabilities = new DesiredCapabilities();
            desiredCapabilities.setBrowserName(getSeleniumBrowserName());

            return createRemote(desiredCapabilities);
        } else {
            return createRemote(new DesiredCapabilities(options.get()));
        }
    }

    WebDriver createLocal(Class webDriverClass, WebDriverOptions options) {
        try {
            if (options == null) {
                return webDriverClass.getConstructor().newInstance();
            } else {
                Object capabilities = options.get();
                return webDriverClass.getConstructor(capabilities.getClass()).newInstance(capabilities);
            }
        } catch (InstantiationException | IllegalAccessException | NoSuchMethodException e) {
            throw new RuntimeException(e);
        } catch (InvocationTargetException e) {
            throw new IllegalStateException(e.getCause().getMessage());
        }
    }

    Optional> getCustomOptionClass() {
        String optionsClass;

        switch(getSeleniumBrowserName()) {
            case BrowserType.CHROME:
                optionsClass = System.getProperty("chrome.options");

                if (optionsClass != null) {
                    return Optional.of(new WebDriverOptions().put(optionsClass));
                }

                break;
            case BrowserType.FIREFOX:
                optionsClass = System.getProperty("firefox.options");

                if (optionsClass != null) {
                    return Optional.of(new WebDriverOptions().put(optionsClass));
                }

                break;
            case BrowserType.IE:
                optionsClass = System.getProperty("internet.explorer.options");

                if (optionsClass != null) {
                    return Optional.of(new WebDriverOptions().put(optionsClass));
                }

                break;
            case BrowserType.EDGE:
                optionsClass = System.getProperty("edge.options");

                if (optionsClass != null) {
                    return Optional.of(new WebDriverOptions().put(optionsClass));
                }

                break;
        }

        return Optional.empty();
    }

    private static WebDriver createRemote(Capabilities capabilities) {
        try {
            return new RemoteWebDriver(new URL(HUB_URL), capabilities);
        } catch (MalformedURLException e) {
            throw new RuntimeException(e);
        }
    }

    private static String getSeleniumBrowserName() {
        switch (BROWSER_NAME.toLowerCase()) {
            case "chrome":
                return BrowserType.CHROME;
            case "ff":
            case "firefox":
                return BrowserType.FIREFOX;
            case "ie":
            case "internet explorer":
                return BrowserType.IE;
            case "edge":
                return BrowserType.EDGE;
        }

        throw new RuntimeException("browser not implemented");
    }
}