当前位置: 代码迷 >> Web前端 >> selenium webdriver 以署理proxy方式启动firefox,ie,chrome
  详细解决方案

selenium webdriver 以署理proxy方式启动firefox,ie,chrome

热度:880   发布时间:2012-09-04 14:19:30.0
selenium webdriver 以代理proxy方式启动firefox,ie,chrome
原创文章,转载请注明出处:http://passerbyy.iteye.com/blog/1286292      作者:passer_by
1.题前话
没有发现之前,自己傻不垃圾的自己写了各个浏览器修改代理的方法,结果发现webdriver有现成,悔恨不已,希望其他同仁能够少走弯路。

本文是在Webdriver 2.12.0下面测试得到的结论


2. webdriver的maven配置
<repositories>
		<repository>
			<id>selenium</id>
			<name>selenium</name>
			<url>http://repo1.maven.org/maven2/</url>
		</repository>
	</repositories>
	<dependencies>
		<dependency>
			<groupId>org.seleniumhq.selenium</groupId>
			<artifactId>selenium-java</artifactId>
			<version>2.12.0</version>
		</dependency>
	</dependencies>



3. webdriver中firefox以代理方式启动
普通情况下,firefox的代理修改是直接修改其配置文件,即prefs.js,把对应的配置修改掉
user_pref("network.proxy.ftp_port", 1000); 
user_pref("network.proxy.gopher", "10.0.0.0"); 
user_pref("network.proxy.gopher_port", 1000); 
user_pref("network.proxy.http", "10.0.0.0"); 
user_pref("network.proxy.http_port", 1000); 
user_pref("network.proxy.no_proxies_on", ""); 
user_pref("network.proxy.share_proxy_settings", true); 
user_pref("network.proxy.socks", "10.0.0.0"); 
user_pref("network.proxy.socks_port", 1000); 
user_pref("network.proxy.ssl", "10.0.0.0"); 
user_pref("network.proxy.ssl_port", 1000); 
user_pref("network.proxy.type", 1); 


firefoxdriver初始化时,我们可以通过配置FirefoxProfile,来修改上面的配置,特别要注意的是localhost的配置,请看下述例子:
String proxyIp = "localhost";
int proxyPort = 8080;
FirefoxProfile profile = new FirefoxProfile();
// 使用代理
profile.setPreference("network.proxy.type", 1);
// http协议代理配置
profile.setPreference("network.proxy.http", proxyIp);
profile.setPreference("network.proxy.http_port", proxyPort);
		
// 所有协议公用一种代理配置,如果单独配置,这项设置为false,再类似于http的配置
profile.setPreference("network.proxy.share_proxy_settings", true);
		
// 对于localhost的不用代理,这里必须要配置,否则无法和webdriver通讯
profile.setPreference("network.proxy.no_proxies_on", "localhost");
		
// 以代理方式启动firefox
FirefoxDriver ff  = new FirefoxDriver(profile);
ff.get("www.xxx.com");
ff.quit();


4. webdriver中IE以代理方式启动,chrome类似
方式不同于firefox,这里是利用webdriver提供的Proxy和WindowsProxyManager来处理,这里也要特别注意localhost的处理。


处理步骤为:
1. InternetExplorerDriver初始化时,调用WindowsProxyManager的backupRegistrySettings方法保存老的代理配置
2. 调用WindowsProxyManager的changeRegistrySettings方法类修改代理配置
3. 程序运行结束后,调用WindowsProxyManager的restoreRegistrySettings来恢复到老的配置。
特别需要提醒的是第3条,我认为webdriver这里处理的不好。恢复到默认配置是在程序结束后,如果程序启动了多个InternetExplorerDriver,每个InternetExplorerDriver保存的是该InternetExplorerDriver启动时IE的配置,而程序结束是调用的shutdownhooker,同时恢复,线程的运行快慢不确定,最后是否恢复到初始配置还很难说,所以,如果同个程序只启动一个InternetExplorerDriver,使用webdriver自带的初始InternetExplorerDriver时修改代理是没有问题,如果启动多个,就要采取其他方式,可以参看本文的第5部分。

程序只启动一个InternetExplorerDriver,以代理模式启动的代码如下:
String proxyIpAndPort= "localhost:8080";

// 代理配置
DesiredCapabilities cap = new DesiredCapabilities();
org.openqa.selenium.Proxy proxy = new org.openqa.selenium.Proxy();
// 配置http、ftp、ssl代理(注:当前版本只支持所有的协议公用http协议,下述代码等同于只配置http)
proxy.setHttpProxy(proxyIpAndPort)
     .setFtpProxy(proxyIpAndPort)
     .setSslProxy(proxyIpAndPort);

// 以下三行是为了避免localhost和selenium driver的也使用代理,务必要加,否则无法与iedriver通讯
cap.setCapability(CapabilityType.ForSeleniumServer.AVOIDING_PROXY, true);
cap.setCapability(CapabilityType.ForSeleniumServer.ONLY_PROXYING_SELENIUM_TRAFFIC, true);
System.setProperty("http.nonProxyHosts", "localhost");

cap.setCapability(CapabilityType.PROXY, proxy);
WebDriver driver = new InternetExplorerDriver(cap);
driver.get("www.baidu.com");
driver.close();


InternetExplorerDriver初始化后,IE对应的代理配置为:


打开上图中的代理配置文件,可以看到下述配置,具体的可以自己查阅资料:
function FindProxyForURL(url, host) {
  if (shExpMatch(host, 'localhost')) { return 'DIRECT'; }
  if (shExpMatch(url, '*/selenium-server/*')) { return 'PROXY localhost:0; DIRECT'; }
  return 'PROXY 10.16.16.38:3229';
}


程序结束后,恢复原来配置


5. ie和chrome,手动存储老的代理、修改代理、恢复代理
这里同样使用的是webdriver提供的Proxy和WindowsProxyManager,只不过是我们显示调用而已,不让webdriver帮我们处理。几个接口,我在上面也提起过,这里就直接上代码了。
final WindowsProxyManager proxyManager = new WindowsProxyManager(true,
		"webdriver-ie", 0, 0);
// 备份老代理配置
proxyManager.backupRegistrySettings();

// 增加hooker,jvm退出时,把代理修改为之前的。当然,这里可以自己决定什么时候恢复,比如,在每次InternetExplorerDriver关闭后掉用
new Thread() {
	@Override
	public void run() {
		proxyManager.restoreRegistrySettings(true);
	}
};

// 修改代理
DesiredCapabilities cap = changeProxy("localhost",8080);
proxyManager.changeRegistrySettings(cap);

// 启动ie
WebDriver driver1 = new InternetExplorerDriver(cap);
driver1.get("www.baidu.com");
driver1.close();

// 再次修改代理
DesiredCapabilities cap2 = changeProxy("localhost",80);
proxyManager.changeRegistrySettings(cap);

// 再次启动ie
WebDriver driver2 = new InternetExplorerDriver(cap);
driver2.get("www.google.com");
driver2.close();







  相关解决方案