Selenium WebDriver 处理cookie

老白的释然

在使用webdriver测试中，很多地方都使用登陆，cookie能够实现不必再次输入用户名密码进行登陆。

首先了解一下Java Cookie类的一些方法。

1. 
   
2. 在jsp中处理cookie数据的常用方法：
   
3. getDomain（）；返回cookie的域名.
   
4. getMaxAge（）；返回cookie的存活时间
   
5. getName（）；返回cookie的名字
   
6. getPath（）；返回cookie适用的路径
   
7. getSecure（）；如果浏览器通过安全协议发送Cookie将返回true值，如果浏览器使用标准协议刚返回false值
   
8. getValue（）；返回cookie的值
   
9. getVersion（）；返回cookie所遵从的协议版本setComment(String purpose）；设置cookie的注释
   
10. setPath(String url）；设置Cookie的适用路径
    
11. setSecure(Boolean flag）；设置浏览器是否仅仅使用安全协议来发送cookie，例如使用Https或ssl
    
12. setValue(String newvalue);cookie创建后设置一个新的值
    
13. setVersion(int v）；设置cookie所遵从的协议版本
    
14. selenium WebDriver 通过driver.manage().getCookies() 和driver.manage().addCookie(ck); 获取cookie 加载cookie
    
15. 
    
16. 
    
17. 首先，获取cookie 保存的browser.data内
    
18. 复制代码
    
19. package com.packt.webdriver.chapter3;
    
20. 
    
21. import java.io.BufferedWriter;
    
22. import java.io.File;
    
23. import java.io.FileWriter;
    
24. 
    
25. import java.util.concurrent.TimeUnit;
    
26. 
    
27. import org.openqa.selenium.By;
    
28. import org.openqa.selenium.WebDriver;
    
29. import org.openqa.selenium.WebElement;
    
30. import org.openqa.selenium.Cookie;
    
31. 
    
32. public class Cookies {
    
33.     /**
    
34.      * [url=home.php?mod=space&uid=267564]@Author[/url] Young
    
35.      *
    
36.      */
    
37.     public static void addCookies() {
    
38. 
    
39.         WebDriver driver = DriverFactory.getChromeDriver();
    
40.         driver.get("http://www.zhihu.com/#signin");
    
41.         driver.manage().timeouts().implicitlyWait(10, TimeUnit.SECONDS);
    
42. 
    
43.         WebElement user = driver
    
44.                 .findElement(By.xpath("//input[@name='email']"));
    
45.         user.clear();
    
46.         user.sendKeys("seleniumcookies@126.com");
    
47.         WebElement password = driver.findElement(By
    
48.                 .xpath("//input[@name='password']"));
    
49.         password.clear();
    
50.         password.sendKeys("cookies123");
    
51. 
    
52.         WebElement submit = driver.findElement(By
    
53.                 .xpath("//button[@class='sign-button']"));
    
54.         submit.submit();
    
55. 
    
56.         try {
    
57.             Thread.sleep(3000);
    
58.         } catch (InterruptedException e) {
    
59.             // TODO Auto-generated catch block
    
60.             e.printStackTrace();
    
61.         }
    
62. 
    
63.         File file = new File("broswer.data");
    
64.         try {
    
65.             // delete file if exists
    
66.             file.delete();
    
67.             file.createNewFile();
    
68.             FileWriter fw = new FileWriter(file);
    
69.             BufferedWriter bw = new BufferedWriter(fw);
    
70.             for (Cookie ck : driver.manage().getCookies()) {
    
71.                 bw.write(ck.getName() + ";" + ck.getValue() + ";"
    
72.                         + ck.getDomain() + ";" + ck.getPath() + ";"
    
73.                         + ck.getExpiry() + ";" + ck.isSecure());
    
74.                 bw.newLine();
    
75.             }
    
76.             bw.flush();
    
77.             bw.close();
    
78.             fw.close();
    
79. 
    
80.         } catch (Exception e) {
    
81.             e.printStackTrace();
    
82.         } finally {
    
83.             System.out.println("cookie write to file");
    
84.         }
    
85.     }
    
86. }
    
87. 复制代码
    
88. 
    
89. 
    
90. 这里，我简单封装了chrome webdriver
    
91. 
    
92. 复制代码
    
93. import java.util.Arrays;
    
94. 
    
95. import org.openqa.selenium.WebDriver;
    
96. import org.openqa.selenium.chrome.ChromeDriver;
    
97. import org.openqa.selenium.chrome.ChromeOptions;
    
98. import org.openqa.selenium.remote.DesiredCapabilities;
    
99. 
    
100. 
     
101. public class DriverFactory {
     
102. 
     
103.     public static WebDriver create() {
     
104.         
     
105.         // TODO Auto-generated method stub
     
106.         String chromdriver="E:\\chromedriver.exe";
     
107.         System.setProperty("webdriver.chrome.driver", chromdriver);
     
108.         ChromeOptions options = new ChromeOptions();
     
109.      
     
110.         DesiredCapabilities capabilities = DesiredCapabilities.chrome();
     
111.         capabilities.setCapability("chrome.switches",
     
112.                 Arrays.asList("--start-maximized"));
     
113.         options.addArguments("--test-type", "--start-maximized");
     
114.         WebDriver driver=new ChromeDriver(options);
     
115.         return driver;
     
116.     }
     
117. 
     
118. }
     
119. 复制代码
     
120. 接下来 ，读取browser.data,生成cookie  把cookie加载到浏览器
     
121. 
     
122. 复制代码
     
123. package com.packt.webdriver.chapter3;
     
124. 
     
125. import java.io.BufferedReader;
     
126. 
     
127. import java.io.File;
     
128. import java.io.FileReader;
     
129. 
     
130. 
     
131. import java.util.Date;
     
132. import java.util.StringTokenizer;
     
133. 
     
134. import org.openqa.selenium.Cookie;
     
135. import org.openqa.selenium.WebDriver;
     
136. import org.openqa.selenium.os.WindowsUtils;
     
137. 
     
138. public class UseCookieLogin {
     
139. 
     
140.     /**
     
141.      * @author Young
     
142.      * @param args
     
143.      */
     
144.    
     
145.     public static void main(String[] args) {
     
146.         // TODO Auto-generated method stub
     
147.         Cookies.addCookies();
     
148.         WindowsUtils.tryToKillByName("chrome.exe");
     
149.         WindowsUtils.getProgramFilesPath();
     
150.         WebDriver driver=DriverFactory.getChromeDriver();
     
151.         driver.get("http://www.zhihu.com/");
     
152.         try
     
153.         {
     
154.             File file=new File("broswer.data");
     
155.             FileReader fr=new FileReader(file);
     
156.             BufferedReader br=new BufferedReader(fr);
     
157.             String line;
     
158.             while((line=br.readLine())!= null)
     
159.             {
     
160.                 StringTokenizer str=new StringTokenizer(line,";");
     
161.                 while(str.hasMoreTokens())
     
162.                 {
     
163.                     String name=str.nextToken();
     
164.                     String value=str.nextToken();
     
165.                     String domain=str.nextToken();
     
166.                     String path=str.nextToken();
     
167.                     Date expiry=null;
     
168.                     String dt;
     
169.                     if(!(dt=str.nextToken()).equals(null))
     
170.                     {
     
171.                         //expiry=new Date(dt);
     
172.                         System.out.println();
     
173.                     }
     
174.                     boolean isSecure=new Boolean(str.nextToken()).booleanValue();
     
175.                     Cookie ck=new Cookie(name,value,domain,path,expiry,isSecure);
     
176.                     driver.manage().addCookie(ck);
     
177.                 }
     
178.             }
     
179.             
     
180.             
     
181.         }
     
182.         catch(Exception e)
     
183.         {
     
184.             e.printStackTrace();
     
185.         }
     
186. 
     
187.         driver.get("http://www.zhihu.com/");
     
188.     }
     
189. 
     
190. }
     
191. 复制代码
     
192. 
     
193. 
     
194. 再次打开之后，就是登陆后的页面

复制代码

梦想家

谢谢分享