fr.jetoile.demo.breizhcamp.webapp.driver.WebDriverFactory.java Source code

Introduction

Here is the source code for fr.jetoile.demo.breizhcamp.webapp.driver.WebDriverFactory.java

Source

/**
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License
 */
package fr.jetoile.demo.breizhcamp.webapp.driver;

import org.fluentlenium.cucumber.adapter.FluentCucumberTest;
import org.fluentlenium.cucumber.adapter.exception.UnsupportedDriverException;
import org.fluentlenium.cucumber.adapter.util.SharedDriverHelper;
import org.openqa.selenium.Platform;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.htmlunit.HtmlUnitDriver;
import org.openqa.selenium.phantomjs.PhantomJSDriver;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.RemoteWebDriver;

import java.net.MalformedURLException;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;

public class WebDriverFactory {
    public static Map<SupportedWebDriver, WebDriver> webDriverInstances = new HashMap<SupportedWebDriver, WebDriver>();

    public static synchronized WebDriver newWebdriverInstance(FluentCucumberTest adapter,
            SupportedWebDriver driverType, DesiredCapabilities capabilities) throws UnsupportedDriverException {
        if (SharedDriverHelper.isSharedDriverPerFeature(adapter.getClass())) {
            if (webDriverInstances.get(driverType) == null) {
                webDriverInstances.put(driverType, getWebDriver(driverType, capabilities));
            }
            if (FluentCucumberTest.sharedDriver != null) {
                FluentCucumberTest.sharedDriver = webDriverInstances.get(driverType);
            }
            return webDriverInstances.get(driverType);
        } else {
            if (FluentCucumberTest.sharedDriver != null) {
                FluentCucumberTest.sharedDriver = webDriverInstances.get(driverType);
            }
            return getWebDriver(driverType, capabilities);
        }
    }

    private static WebDriver getWebDriver(SupportedWebDriver driverType, DesiredCapabilities capabilities)
            throws UnsupportedDriverException {
        try {
            switch (driverType) {
            case FIREFOX:
                return firefoxDriver(capabilities);
            case CHROME:
                return chromeDriver(capabilities);
            case HTMLUNIT:
                return htmlUnitDriver();
            case PHANTOMJS:
                return phantomjsDriver(capabilities);
            case REMOTE:
                return remoteDriver(capabilities);
            }
        } catch (Exception cause) {
            throw new UnsupportedDriverException("Could not instantiate " + driverType, cause);
        }
        return null;
    }

    private static WebDriver remoteDriver(DesiredCapabilities capabilities) throws UnsupportedDriverException {
        String osName = (String) capabilities.getCapability("os.name");

        capabilities.setPlatform(Platform.extractFromSysProperty(osName, ""));
        capabilities.setVersion((String) capabilities.getCapability("os.version"));

        String url = (String) capabilities.getCapability("webdriver.remote.url");
        try {
            return new RemoteWebDriver(new URL(url), capabilities);
        } catch (MalformedURLException e) {
            throw new UnsupportedDriverException("wrong parameters for remote webDriver: " + capabilities, e);
        }
    }

    private static WebDriver phantomjsDriver(DesiredCapabilities capabilities) {
        return new PhantomJSDriver(capabilities);
    }

    private static WebDriver htmlUnitDriver() {
        return new HtmlUnitDriver();
    }

    private static WebDriver chromeDriver(DesiredCapabilities capabilities) {
        System.setProperty("webdriver.chrome.driver",
                (String) capabilities.getCapability("webdriver.chrome.driver"));
        return new ChromeDriver(capabilities);
    }

    private static WebDriver firefoxDriver(DesiredCapabilities capabilities) {
        return new FirefoxDriver(capabilities);
    }

}