Selenium WebDriver 處理cookie

在使用webdriver測試中，很多地方都使用登陸，cookie能夠實現不必再次輸入用戶名密碼進行登陸。

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

在jsp中處理cookie數據的常用方法：

getDomain（）；返回cookie的域名.

getMaxAge（）；返回cookie的存活時間

getName（）；返回cookie的名字

getPath（）；返回cookie適用的路徑

getSecure（）；如果瀏覽器通過安全協議發送Cookie將返回true值，如果瀏覽器使用標准協議剛返回false值

getValue（）；返回cookie的值

getVersion（）；返回cookie所遵從的協議版本setComment(String purpose）；設置cookie的注釋

setPath(String url）；設置Cookie的適用路徑

setSecure(Boolean flag）；設置瀏覽器是否僅僅使用安全協議來發送cookie，例如使用Https或ssl

setValue(String newvalue);cookie創建后設置一個新的值

setVersion(int v）；設置cookie所遵從的協議版本

selenium WebDriver 通過driver.manage().getCookies() 和driver.manage().addCookie(ck); 獲取cookie 加載cookie

 

 

首先，獲取cookie 保存的browser.data內

package com.packt.webdriver.chapter3;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;

import java.util.concurrent.TimeUnit;

import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.Cookie;

public class Cookies {
    /**
     * @author Young
     * 
     */
    public static void addCookies() {

        WebDriver driver = DriverFactory.getChromeDriver();
        driver.get("http://www.zhihu.com/#signin");
        driver.manage().timeouts().implicitlyWait(10, TimeUnit.SECONDS);

        WebElement user = driver
                .findElement(By.xpath("//input[@name='email']"));
        user.clear();
        user.sendKeys("[e-mail]");
        WebElement password = driver.findElement(By
                .xpath("//input[@name='password']"));
        password.clear();
        password.sendKeys("cookies123");

        WebElement submit = driver.findElement(By
                .xpath("//button[@class='sign-button']"));
        submit.submit();

        try {
            Thread.sleep(3000);
        } catch (InterruptedException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        File file = new File("broswer.data");
        try {
            // delete file if exists
            file.delete();
            file.createNewFile();
            FileWriter fw = new FileWriter(file);
            BufferedWriter bw = new BufferedWriter(fw);
            for (Cookie ck : driver.manage().getCookies()) {
                bw.write(ck.getName() + ";" + ck.getValue() + ";"
                        + ck.getDomain() + ";" + ck.getPath() + ";"
                        + ck.getExpiry() + ";" + ck.isSecure());
                bw.newLine();
            }
            bw.flush();
            bw.close();
            fw.close();

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            System.out.println("cookie write to file");
        }
    }
}

 

這里，我簡單封裝了chrome webdriver

import java.util.Arrays;

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;
import org.openqa.selenium.remote.DesiredCapabilities;


public class DriverFactory {

    public static WebDriver create() {
        
        // TODO Auto-generated method stub
        String chromdriver="E:\\chromedriver.exe";
        System.setProperty("webdriver.chrome.driver", chromdriver);
        ChromeOptions options = new ChromeOptions();
     
        DesiredCapabilities capabilities = DesiredCapabilities.chrome();
        capabilities.setCapability("chrome.switches",
                Arrays.asList("--start-maximized"));
        options.addArguments("--test-type", "--start-maximized");
        WebDriver driver=new ChromeDriver(options);
        return driver;
    }

}

接下來 ，讀取browser.data,生成cookie  把cookie加載到瀏覽器

package com.packt.webdriver.chapter3;

import java.io.BufferedReader;
 
import java.io.File;
import java.io.FileReader;
 

import java.util.Date;
import java.util.StringTokenizer;

import org.openqa.selenium.Cookie;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.os.WindowsUtils;

public class UseCookieLogin {

    /**
     * @author Young
     * @param args
     */
    
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Cookies.addCookies();
        WindowsUtils.tryToKillByName("chrome.exe");
        WindowsUtils.getProgramFilesPath();
        WebDriver driver=DriverFactory.getChromeDriver();
        driver.get("http://www.zhihu.com/");
        try 
        {
            File file=new File("broswer.data");
            FileReader fr=new FileReader(file);
            BufferedReader br=new BufferedReader(fr);
            String line;
            while((line=br.readLine())!= null)
            {
                StringTokenizer str=new StringTokenizer(line,";");
                while(str.hasMoreTokens())
                {
                    String name=str.nextToken();
                    String value=str.nextToken();
                    String domain=str.nextToken();
                    String path=str.nextToken();
                    Date expiry=null;
                    String dt;
                    if(!(dt=str.nextToken()).equals(null))
                    {
                        //expiry=new Date(dt);
                        System.out.println();
                    }
                    boolean isSecure=new Boolean(str.nextToken()).booleanValue();
                    Cookie ck=new Cookie(name,value,domain,path,expiry,isSecure);
                    driver.manage().addCookie(ck);
                }
            }
            
            
        }
        catch(Exception e)
        {
            e.printStackTrace();
        }

        driver.get("http://www.zhihu.com/");
    }

}

 

再次打開之后，就是登陸后的頁面