需求:
需要测试API,而实际上整个流程需要跳转到第三方网站让用户登录和授权。因此需要模拟这个过程
解决:
使用selenium的HtmlUnitDriver,由于项目中使用了selenium,也可以直接使用HtmlUnit。
问题:
从第三方网站授权完后,重定向到自己网站后,会设置把第三方网站返回的信息放到session中。结果丢失。
分析:
因为在跳转到第三方网站之前,使用HttpClient调用了网站的API去做登录等操作,而HtmlUnit里面也会有一个自己的HttpClient,因此判断在重定向之后Cookie丢失。
为了验证分析正确,需要监控一下HttpClient发出的请求和得到的返回信息。
解决方案:
使用Fiddler,之前只是知道Fiddler可以监控Http的请求和返回消息。其实Fiddler 是以代理web服务器的形式工作的,它使用代理地址:127.0.0.1, 端口:8888. 因此可以给HttpClient的代理设为Fiddler就可以监控了。
在Fiddler的左下角可以选择Non-Browser来过滤只显示非浏览器的请求。
HttpHost proxy = new HttpHost("127.0.0.1", 8888);
httpClient.getParams().setParameter(ConnRoutePNames.DEFAULT_PROXY, proxy);
同时给HtmlUnitDriver也设为相同的代理就可以监控了。发现分析的是正确的。
因此在使用HtmlUnitDriver和结束时需要相互设置Cookie。
public static class ODHtmlUnitDriver extends HtmlUnitDriver {
public ODHtmlUnitDriver(Capabilities capabilities) {
super(capabilities);
}
public void setCookieToHttpClient(DefaultHttpClient httpClient) {
Set cookies = getWebClient().getCookieManager().getCookies();
org.apache.http.cookie.Cookie [] httpClientCookies = Cookie.toHttpClient(cookies);
for (org.apache.http.cookie.Cookie cookie : httpClientCookies) {
httpClient.getCookieStore().addCookie(cookie);
}
}
public void setCookieFromHttpClient(DefaultHttpClient httpClient) {
List httpClientCookies = httpClient.getCookieStore().getCookies();
List cookies = Cookie.fromHttpClient(httpClientCookies);
for (Cookie cookie : cookies) {
getWebClient().getCookieManager().addCookie(cookie);
}
}
}
HtmlUnit也提供了函数在自己的Cookie和HttpClient的Cookie直接进行转换。
详细的解释了Fiddler的使用,强悍啊。。。
http://www.cnblogs.com/TankXiao/archive/2012/02/06/2337728.html