com.github.dactiv.common.unit.selenium.WebDriverFactory.java Source code

Introduction

Here is the source code for com.github.dactiv.common.unit.selenium.WebDriverFactory.java

Source

/**
 * Copyright (c) 2005-2012 springside.org.cn
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 */
package com.github.dactiv.common.unit.selenium;

import java.net.MalformedURLException;
import java.net.URL;

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.ie.InternetExplorerDriver;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.RemoteWebDriver;
import org.springframework.util.Assert;

/**
 * WebDriver.
 * 
 * @author calvin
 */
public class WebDriverFactory {
    /**
     * ?driverName??WebDriver.
     * 
     * ??firefox,ie,chrome??.
     * 
     * ????Windows, IE?XWindows, ???remote driverWindows.
     * drivernameremote:192.168.0.2:4444:firefox, ??http://192.168.0.2:4444/wd/hub?selenium remote?.
     */
    public static WebDriver createDriver(String driverName) {
        WebDriver driver = null;

        if (BrowserType.firefox.name().equals(driverName)) {
            driver = new FirefoxDriver();
        } else if (BrowserType.ie.name().equals(driverName)) {
            driver = new InternetExplorerDriver();
        } else if (BrowserType.chrome.name().equals(driverName)) {
            driver = new ChromeDriver();
        } else if (driverName.startsWith(BrowserType.remote.name())) {
            String[] params = driverName.split(":");
            Assert.isTrue(params.length == 4,
                    "Remote driver is not right, accept format is \"remote:localhost:4444:firefox\", but the input is\""
                            + driverName + "\"");

            String remoteHost = params[1];
            String remotePort = params[2];
            String driverType = params[3];

            String remoteUrl = "http://" + remoteHost + ":" + remotePort + "/wd/hub";

            DesiredCapabilities cap = null;
            if (BrowserType.firefox.name().equals(driverType)) {
                cap = DesiredCapabilities.firefox();
            } else if (BrowserType.ie.name().equals(driverType)) {
                cap = DesiredCapabilities.internetExplorer();
            } else if (BrowserType.chrome.name().equals(driverType)) {
                cap = DesiredCapabilities.chrome();
            }

            try {
                driver = new RemoteWebDriver(new URL(remoteUrl), cap);
            } catch (MalformedURLException e) {
                throw new RuntimeException(e);
            }
        }

        Assert.notNull(driver, "Driver could be found by name:" + driverName);

        return driver;
    }

    /**
     * ?
     * 
     */
    public enum BrowserType {
        /**
         * ???
         */
        firefox,
        /**
         * ie?
         */
        ie,
        /**
         * ?
         */
        chrome,
        /**
         * ??
         */
        remote
    }
}