/**
* 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 driver调用远程的Windows机器.
* drivername如remote: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
}
}