Selenium WebDriver 處理cookie

標籤：des   style   blog   http   color   io   os   使用   java   

在使用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或sslsetValue(String newvalue);cookie建立後設定一個新的值setVersion(int v）；設定cookie所遵從的協議版本selenium WebDriver 通過driver.manage().getCookies() 和driver.manage().addCookie(ck); 擷取cookie 載入cookie  首先，擷取cookie 儲存的browser.data內

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 {    public static void main(String[] args) {                 WebDriver driver=DriverFactory.create();        driver.get("http://selenium.jd-app.com/wp-login.php");        driver.manage().timeouts().implicitlyWait(10, TimeUnit.SECONDS);        WebElement user = driver.findElement(By.xpath("//*[@id=‘user_login‘]"));        user.clear();        user.sendKeys("username");        WebElement password = driver.findElement(By                .xpath("//*[@id=‘user_pass‘]"));        password.clear();        password.sendKeys("password");                 WebElement submit=driver.findElement(By.xpath("//*[@id=‘wp-submit‘]"));            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載入到瀏覽器

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;public class UseCookieLogin {    public static void main(String[] args) {        // TODO Auto-generated method stub        WebDriver driver=DriverFactory.create();        driver.get("http://selenium.jd-app.com/wp-admin/");        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://selenium.jd-app.com/wp-admin/");    }}

管理頁面只有登陸後才能開啟，如果未登入就會跳轉登陸頁面，當我們載入cookie後，看到的是登陸後的頁面

 

Selenium WebDriver 處理cookie