Selenium2学习-017-WebUI自动化实战实例-015-获取浏览器中的 cookie 信息

日常我们在使用浏览器时，尤其是登录 WEB 应用时，我们的一些信息其实是保存在了浏览器的 cookie 信息中。我们可以通过浏览器自带的开发工具，进行查看相应的 cookie 信息，例如在火狐、chrome 中均可通过 F12 打开开发者工具打开。以下截图为在 chrome 中打开易迅网时的 cookie 信息：

若是小主您登录了相应的网页应用，cookie 中还会含有您的一些用户信息内容。那么这些 cookie 信息在我们日常 WebUI 自动化脚本的编写过程中有什么用途呢 ？ 可以作为网页打开是否正确的校验依据，可以作为用户是否正确登录的判定依据等等。那么我们如何获取相应的 cookie 信息呢，以下自动化测试脚本以易迅网易迅账号登录获取其 cookie 给出了明确的实例演示，其他网站的登录 cookie 信息获取相同。

闲话少述，小二上码。。。敬请各位小主参阅，希望能对您在日常的 WebUI 自动化脚本编写有一定的启发和帮助。若有不足或错误之处，敬请大神指正，不胜感激！

/**
 * Aaron.ffp Inc.
 * Copyright (c) 2004-2015 All Rights Reserved.
 */
package main.aaron.demo.cookie;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.util.concurrent.TimeUnit;

import javax.swing.JDialog;
import javax.swing.JOptionPane;
import javax.swing.JPasswordField;

import org.openqa.selenium.By;
import org.openqa.selenium.Cookie;
import org.openqa.selenium.WebElement;
import org.testng.annotations.AfterClass;
import org.testng.annotations.Test;

import main.aaron.sele.core.TestCase;

/**
 * get cookie info by manual login
 * 
 * @author Aaron.ffp
 * @version V1.0.0: autoSeleniumDemo main.aaron.demo.cookie Cookies.java, 2015-6-19 13:53:05 Exp $
 */
public class Cookies extends TestCase{
    private final String f_cookie = this.PROJECTHOME + this.FILESEPARATOR + "cookie" + this.FILESEPARATOR + "browser.data";
    private String baseUrl = "";
    File file;
    
    @Test
    public void addCookie(){
        this.baseUrl = "https://base.yixun.com/login.html?url=http%3A%2F%2Fbase.yixun.com%2F";
        
        // wait ten seconds
        this.webdriver.manage().timeouts().implicitlyWait(10, TimeUnit.SECONDS);
        
        // navigate to login page
        this.webdriver.get(this.baseUrl);
        
        // change to login frame
        this.webdriver.switchTo().frame("id_iframe_login");
        
        // click yixun_login, change to yixun login tab
        WebElement yx_login = this.webdriver.findElement(By.cssSelector(".fast_yx"));
        yx_login.click();
        
//        this.webdriver.switchTo().frame("id_iframe_login");
        
        WebElement user = this.webdriver.findElement(By.id("yxname"));
        WebElement pass = this.webdriver.findElement(By.id("yxpw"));
        WebElement code = this.webdriver.findElement(By.id("yxcode"));
        WebElement subm = this.webdriver.findElement(By.id("submit-button"));
        
        // clean the input dialog
        user.clear();
        pass.clear();
        code.clear();
        
        // input login info
        user.sendKeys("请小主们自行注册易迅网易迅账号");
        pass.sendKeys("请小主们自行注册易迅网易迅账号");
        code.sendKeys(this.inputDialog());
        
        // click login button
        subm.click();
        
        this.file = new File(this.f_cookie);
        
        // store login cookie
        try {
            Thread.sleep(15000);
            
            this.file.delete();
            
            this.file.createNewFile();
            
            FileWriter fw = new FileWriter(this.f_cookie);
            
            BufferedWriter bw = new BufferedWriter(fw);
            
            for (Cookie cookie : this.webdriver.manage().getCookies()) {
                bw.write(cookie.getName() + ";" + 
                         ("".equals(cookie.getValue()) || (cookie.getValue() == null) ? "": cookie.getValue()) + ";" + 
                         cookie.getDomain() + ";" + 
                         cookie.getPath() + ";" + 
                         cookie.getExpiry() + ";" + 
                         cookie.isSecure());
                bw.newLine();
            }
            
            bw.flush();
            bw.close();
            fw.close();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            System.out.println("cookie write to file : " + this.f_cookie);
        }
    }
    
    /**
     * input dialog
     * 
     * @author Aaron.ffp
     * @version V1.0.0: autoSeleniumDemo main.aaron.demo.cookie Cookies.java inputDialog, 2015-6-19 13:55:05 Exp $
     * 
     * @return
     */
    public String inputDialog() {        
        JPasswordField pass = new JPasswordField(30);
        JOptionPane id = new JOptionPane("Please input verify code...", JOptionPane.PLAIN_MESSAGE);
    
        id.add(pass);
    
        pass.setEchoChar('*');
        JDialog inputPassDialog = id.createDialog(id, "inputdialog");
    
        inputPassDialog.setVisible(true);
    
        String inStr = String.valueOf(pass.getPassword());
        return inStr;
    }
    
    @AfterClass
    public void afterClass(){
        this.webdriver.close();
        this.webdriver.quit();
    }
}

注意：在实际的使用中， 小主们需要将 cookie 文件保存的位置修改为本地的合法路径才可，对应代码如下：

private final String f_cookie = this.PROJECTHOME + this.FILESEPARATOR + "cookie" + this.FILESEPARATOR + "browser.data";

 

至此，WebUI 自动化功能测试脚本第 015-获取浏览器中的 cookie 信息 顺利完结，希望此文能够给初学 Selenium 的您一份参考。

最后，非常感谢亲的驻足，希望此文能对亲有所帮助。热烈欢迎亲一起探讨，共同进步。非常感谢！ ^_^