Webdriver+Java实现使用cookie跳过登录
Webdriver模拟登录过程中很有可能遇到验证码，最近认真学习了下如何使用cookie直接跳过登录过程。
一、cookie的定义
来源百度百科：
Cookie，有时也用其复数形式 Cookies，指某些网站为了辨别用户身份、进行 session 跟踪而储存在用户本地终端上的数据（通常经过加密）。定义于 RFC2109 和 2965 中的都已废弃，最新取代的规范是 RFC6265（可以叫做浏览器缓存）。
Cookie 是在 HTTP 协议下，服务器或脚本可以维护客户工作站上信息的一种方式。Cookie 是由 Web 服务器保存在用户浏览器（客户端）上的小文本文件，它可以包含有关用户的信息。无论何时用户链接到服务器，Web 站点都可以访问 Cookie 信息 。
目前有些 Cookie 是临时的，有些则是持续的。临时的 Cookie 只在浏览器上保存一段规定的时间，一旦超过规定的时间，该 Cookie 就会被系统清除。持续的 Cookie 则保存在用户的 Cookie 文件中，下一次用户返回时，仍然可以对它进行调用。
临时的Cookie 是个存储在浏览器目录的文本文件，当浏览器运行时，存储在 RAM 中。当访客结束其浏览器对话时，即终止的所有 Cookie。持续的Cookie 可存储在计算机的硬驱上。
二、cookie的组成
一个cookie一般包括name，value，domain，path，expiry，isSecure。
简单实例如下：
Set-Cookie: name = VALUE;
expires = DATE;
path = PATH;
domain = DOMAIN_NAME;
三、使用Webdriver模拟登录后，将cookie写入cookie.data文件中，再次从文件中读取后跳过登录，具体代码如下：
package XXX;
import XXX;
public class CookieTest implements Serializable{
/**
* 序列化
*/
private static final long serialVersionUID = 1L;
private WebDriver driver;
private String baseUrl;
private String homeUrl;
private String cookiePath;
private boolean acceptNextAlert = true;
private StringBuffer verificationErrors = new StringBuffer();
@Before
/*
* 方法名 setUp()
* 加载chrome浏览器的驱动文件
* 初始化chrome浏览器配置信息
* 设置默认访问baseUrl = "XXXXXX";
*/
public void setUp() throws Exception {
System.setProperty("webdriver.chrome.driver","lib\\chromedriver_new.exe");
ChromeOptions options = new ChromeOptions();
options.addArguments("--start-maximized");// 让Chrome浏览器窗口最大化
options.addArguments("--disable-popup-blocking");
options.addArguments("no-sandbox");
options.addArguments("disable-extensions");
options.addArguments("no-default-browser-check");
Map<String, Object> prefs = new HashMap<String, Object>();
prefs.put("credentials_enable_service", false);
prefs.put("profile.password_manager_enabled", false);
options.setExperimentalOption("prefs", prefs);
driver = new ChromeDriver(options);
baseUrl = "XXXXXX";
homeUrl = baseUrl+"XXXXXX";
cookiePath = "cookie.data";
driver.manage().timeouts().implicitlyWait(30, TimeUnit.SECONDS);
}
@Test
public void saveCookieTest() throws FileNotFoundException, InterruptedException, IOException{
//saveCookieNew("XXX","XXXX","cookie.data");
getCookieLogin(homeUrl,cookiePath);
}
@SuppressWarnings("deprecation")
public void getCookieLogin(String url,String cookiePath){
driver.get(url);
sleep(3000);
File file = new File(cookiePath);
try {
FileReader fr = new FileReader(file);
BufferedReader br = new BufferedReader(fr);
String line;
while((line = br.readLine())!= null){
StringTokenizer st = new StringTokenizer(line,";");
while(st.hasMoreTokens()){
String name = st.nextToken();
//System.out.println("name:"+name);//for test
String value = st.nextToken();
//System.out.println("value:"+value);//for test
String domain = st.nextToken();
//System.out.println("domain:"+domain);//for test
String path = st.nextToken();
//System.out.println("path:"+path);//for test
Date expiry = null;//cookie的失效时间，默认存在浏览器打开期间
String expiryString = st.nextToken();
/*if(!(expiryString.equals(null)&& expiryString.equals("false")))
{
//expiry = new Date(expiryString);
System.out.println("expiry:"+expiryString);//for test
}*/
boolean isSecure = new Boolean(st.nextToken()).booleanValue();
//System.out.println("isSecure:"+isSecure);//for test
Cookie ck = new Cookie(name,value,domain,path,expiry,isSecure);
//System.out.println(ck.toString());//for test
driver.manage().addCookie(ck);
}
}
} catch (IOException e) {
e.printStackTrace();
}
driver.get(url);
sleep(60000);
}
/**
* 分别打印cookie信息
* 使用BufferedWriter和FileWriter对象保存cookies到指定文件中
* 保存格式:name;value;domain;expires;isSecure
* @param username
* @param password
* @param cookiepath
* @throws IOException
*/
public void saveCookieNew(String username,String password,String cookiepath) {
driver.get(baseUrl+"XXXXXX");
sleep(1000);
driver.findElement(By.id("userName")).clear();
driver.findElement(By.id("userName")).sendKeys(username);
sleep(1000);
driver.findElement(By.id("password")).clear();
driver.findElement(By.id("password")).sendKeys(password);
sleep(2000);
driver.findElement(By.xpath("//button[@type='button']")).click();
sleep(5000);
//driver.get(baseUrl+"ExerciseBookManager/web/home");
//获取cookies
Set<Cookie> cookies = driver.manage().getCookies();
System.out.println("cookie_size:"+cookies.size());
Iterator<Cookie> it = cookies.iterator();
File file = new File(cookiepath);
file.delete();
try {
file.createNewFile();
FileWriter fw = new FileWriter(file);
BufferedWriter bw = new BufferedWriter(fw);
for(Cookie ck : cookies){
bw.write(ck.getName()+";"+ck.getValue()+";"+ck.getDomain()+";"+ck.getPath()+";"+ck.getExpiry()+";"+ck.isSecure());
bw.newLine();
}
bw.flush();
bw.close();
fw.close();
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* 分别打印cookie信息
* 使用ObjectOutputStream和FileOutputStream对象保存cookies到指定文件中
* @param username
* @param password
* @param savecookiepath
* @throws InterruptedException
* @throws FileNotFoundException
* @throws IOException
*/
public void saveCookie(String username,String password,String savecookiepath) throws InterruptedException, FileNotFoundException, IOException {
driver.get(baseUrl+"ExerciseBookManager/web/auth/login");
Thread.sleep(1000);
driver.findElement(By.id("userName")).clear();
driver.findElement(By.id("userName")).sendKeys(username);
Thread.sleep(1000);
driver.findElement(By.id("password")).clear();
driver.findElement(By.id("password")).sendKeys(password);
Thread.sleep(1000);
driver.findElement(By.xpath("//button[@type='button']")).click();
Thread.sleep(5000);
//driver.get(baseUrl+"ExerciseBookManager/web/home");
//获取cookies
Set<Cookie> cookies = driver.manage().getCookies();
System.out.println("cookie_size:"+cookies.size());
Iterator<Cookie> it = cookies.iterator();
CookieStore cookieStore = new BasicCookieStore();
while(it.hasNext()){
Cookie cookie = it.next();
BasicClientCookie bcs = new BasicClientCookie(cookie.getName(),cookie.getValue());
bcs.setDomain(cookie.getDomain());
bcs.setPath(cookie.getPath());
System.out.println(cookie.toString());
cookieStore.addCookie(bcs);
}
//将Cookies保存到文件中
ObjectOutputStream oos = new ObjectOutputStream(new FileOutputStream(new File(savecookiepath)));
oos.writeObject(cookieStore);
oos.close();
}
@After
public void tearDown() throws Exception {
driver.quit();//关闭浏览器
String verificationErrorString = verificationErrors.toString();
if (!"".equals(verificationErrorString)) {
fail(verificationErrorString);
}
}
//等待时间
public void sleep(long time){
try {
Thread.sleep(time);
} catch (InterruptedException e) {
e.printStackTrace();
}
}