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

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

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

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

  1 /**
  2  * Aaron.ffp Inc.
  3  * Copyright (c) 2004-2015 All Rights Reserved.
  4  */
  5 package main.aaron.demo.cookie;
  6
  7 import java.io.BufferedWriter;
  8 import java.io.File;
  9 import java.io.FileWriter;
 10 import java.util.concurrent.TimeUnit;
 11
 12 import javax.swing.JDialog;
 13 import javax.swing.JOptionPane;
 14 import javax.swing.JPasswordField;
 15
 16 import org.openqa.selenium.By;
 17 import org.openqa.selenium.Cookie;
 18 import org.openqa.selenium.WebElement;
 19 import org.testng.annotations.AfterClass;
 20 import org.testng.annotations.Test;
 21
 22 import main.aaron.sele.core.TestCase;
 23
 24 /**
 25  * get cookie info by manual login
 26  *
 27  * @author Aaron.ffp
 28  * @version V1.0.0: autoSeleniumDemo main.aaron.demo.cookie Cookies.java, 2015-6-19 13:53:05 Exp $
 29  */
 30 public class Cookies extends TestCase{
 31     private final String f_cookie = this.PROJECTHOME + this.FILESEPARATOR + "cookie" + this.FILESEPARATOR + "browser.data";
 32     private String baseUrl = "";
 33     File file;
 34
 35     @Test
 36     public void addCookie(){
 37         this.baseUrl = "https://base.yixun.com/login.html?url=http%3A%2F%2Fbase.yixun.com%2F";
 38
 39         // wait ten seconds
 40         this.webdriver.manage().timeouts().implicitlyWait(10, TimeUnit.SECONDS);
 41
 42         // navigate to login page
 43         this.webdriver.get(this.baseUrl);
 44
 45         // change to login frame
 46         this.webdriver.switchTo().frame("id_iframe_login");
 47
 48         // click yixun_login, change to yixun login tab
 49         WebElement yx_login = this.webdriver.findElement(By.cssSelector(".fast_yx"));
 50         yx_login.click();
 51
 52 //        this.webdriver.switchTo().frame("id_iframe_login");
 53
 54         WebElement user = this.webdriver.findElement(By.id("yxname"));
 55         WebElement pass = this.webdriver.findElement(By.id("yxpw"));
 56         WebElement code = this.webdriver.findElement(By.id("yxcode"));
 57         WebElement subm = this.webdriver.findElement(By.id("submit-button"));
 58
 59         // clean the input dialog
 60         user.clear();
 61         pass.clear();
 62         code.clear();
 63
 64         // input login info
 65         user.sendKeys("请小主们自行注册易迅网易迅账号");
 66         pass.sendKeys("请小主们自行注册易迅网易迅账号");
 67         code.sendKeys(this.inputDialog());
 68
 69         // click login button
 70         subm.click();
 71
 72         this.file = new File(this.f_cookie);
 73
 74         // store login cookie
 75         try {
 76             Thread.sleep(15000);
 77
 78             this.file.delete();
 79
 80             this.file.createNewFile();
 81
 82             FileWriter fw = new FileWriter(this.f_cookie);
 83
 84             BufferedWriter bw = new BufferedWriter(fw);
 85
 86             for (Cookie cookie : this.webdriver.manage().getCookies()) {
 87                 bw.write(cookie.getName() + ";" +
 88                          ("".equals(cookie.getValue()) || (cookie.getValue() == null) ? "": cookie.getValue()) + ";" +
 89                          cookie.getDomain() + ";" +
 90                          cookie.getPath() + ";" +
 91                          cookie.getExpiry() + ";" +
 92                          cookie.isSecure());
 93                 bw.newLine();
 94             }
 95
 96             bw.flush();
 97             bw.close();
 98             fw.close();
 99         } catch (Exception e) {
100             e.printStackTrace();
101         } finally {
102             System.out.println("cookie write to file : " + this.f_cookie);
103         }
104     }
105
106     /**
107      * input dialog
108      *
109      * @author Aaron.ffp
110      * @version V1.0.0: autoSeleniumDemo main.aaron.demo.cookie Cookies.java inputDialog, 2015-6-19 13:55:05 Exp $
111      *
112      * @return
113      */
114     public String inputDialog() {
115         JPasswordField pass = new JPasswordField(30);
116         JOptionPane id = new JOptionPane("Please input verify code...", JOptionPane.PLAIN_MESSAGE);
117
118         id.add(pass);
119
120         pass.setEchoChar(‘*‘);
121         JDialog inputPassDialog = id.createDialog(id, "inputdialog");
122
123         inputPassDialog.setVisible(true);
124
125         String inStr = String.valueOf(pass.getPassword());
126         return inStr;
127     }
128
129     @AfterClass
130     public void afterClass(){
131         this.webdriver.close();
132         this.webdriver.quit();
133     }
134 }

易迅网易迅账号登录获取 cookie 实例源代码

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

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

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

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