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

Go to download
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.DesiredCapabilities;

public class WebDriverProvider {

    public WebDriver createInstance(Type type) {
        switch (type) {
            case FIREFOX:
                return new FirefoxDriver();
            case CHROME:
                return new ChromeDriver();
            case OPERA:
                return new OperaDriver(getOperaCapabilities());
            default:
                throw new RuntimeException("Unsupported web driver type.");
        }
    }
    
    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,
        CHROME,
        OPERA;

        public static Type fromString(String typeString) {
            if (typeString != null) {
                for (Type type : Type.values()) {
                    if (typeString.equalsIgnoreCase(type.name())) {
                        return type;
                    }
                }
            }

            return null;
        }
    }


}