website.automate.jwebrobot.executor.WebDriverProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jwebrobot Show documentation
Reference implementation of the web automation markup language (WAML).
The newest version!
package website.automate.jwebrobot.executor;

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.opera.OperaDriver;
import org.openqa.selenium.remote.CommandExecutor;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.HttpCommandExecutor;
import org.openqa.selenium.remote.RemoteWebDriver;
import org.springframework.stereotype.Service;

import java.net.URL;

@Service
public class WebDriverProvider {

    public WebDriver createInstance(Type type) {
        switch (type) {
            case CHROME:
                return new ChromeDriver();
            case CHROME_HEADLESS:
                ChromeOptions options = new ChromeOptions();
                options.addArguments("--headless", "--disable-gpu");
                ChromeDriver chromeDriver = new ChromeDriver(options);

                return chromeDriver;
            case OPERA:
                return new OperaDriver(getOperaCapabilities());
            case FIREFOX:
            default:
                return new FirefoxDriver();
        }
    }

    public WebDriver createInstance(Type type, URL webDriverUrl){
        if(webDriverUrl == null){
            return createInstance(type);
        }

        return createRemoteInstance(type, webDriverUrl);
    }

    private WebDriver createRemoteInstance(Type type, URL webDriverUrl) {
        CommandExecutor commandExecutor = new HttpCommandExecutor(webDriverUrl);
        switch (type) {
            case CHROME:
            case CHROME_HEADLESS:
                return new RemoteWebDriver(commandExecutor, DesiredCapabilities.chrome());
            case OPERA:
                return new RemoteWebDriver(commandExecutor, DesiredCapabilities.operaBlink());
            case FIREFOX:
            default:
                return new RemoteWebDriver(commandExecutor, DesiredCapabilities.firefox());
        }
    }

    private DesiredCapabilities getOperaCapabilities(){
        DesiredCapabilities capabilities = DesiredCapabilities.operaBlink();
        String operaBinaryPath = System.getProperty("webdriver.opera.binary");
        if(operaBinaryPath != null){
            ChromeOptions options = new ChromeOptions();
            options.setBinary(operaBinaryPath);
            capabilities.setCapability(ChromeOptions.CAPABILITY, options);
        }
        return capabilities;
    }

    public enum Type {
        FIREFOX("firefox"),
        CHROME("chrome"),
        CHROME_HEADLESS("chrome-headless"),
        OPERA("opera");

        private final String canonicalName;

        Type(String canonicalName) {
            this.canonicalName = canonicalName;
        }

        public static Type fromString(String typeString) {
            for (Type type : Type.values()) {
                if (type.canonicalName.equalsIgnoreCase(typeString)) {
                    return type;
                }
            }

            return null;
        }
    }


}