Cookie操作-----Selenium快速入门（十一）

495 0 0

作者:网络

　　什么是cookie？顾名思义，就是饼干，小甜饼。而根据读音则是“曲奇”的意思，相信不少的人都吃过。而在网络上，cookie是指浏览器在本地的一种少量数据的存储方式。例如，我们常见的，登陆的时候有个checkbox选项，上面写着“记住我”，这个功能的实现靠的就是cookie。如果你之前未接触过cookie，更多的的信息，请自行百度。

　　Selenium操作cookie非常简单，操作的方法主要有：

返回类型	方法和说明
`void`	`addCookie(Cookie cookie)` 添加一个cookie
`void`	`deleteAllCookies()` 删除当前域名下的所有cookie
`void`	`deleteCookie(Cookie cookie)` 删除一个cookie
`void`	`deleteCookieNamed(java.lang.String name)` 删除一个cookie，参数是cookie名
`Cookie`	`getCookieNamed(java.lang.String name) 通过名字得到cookie`
`java.util.Set<Cookie>`	`getCookies() 获得当前域名下的所有cookie`

　　下面直接通过实例来说明对cookie的操作。

　　首先，我们需要安装Web服务器，我在win10上安装了iis，并在默认网站根目录增加两个文件，cookie1.html和cookie2.html，这两个html的作用就是模拟用户登陆。

　　cookie1.html的代码：

<!DOCTYPE html>
<html>
<head>
    <meta charset="utf-8" />
    <title></title>
    <script>
        //设置cookie
        function setCookie(c_name, value, expiredays) {
            var exdate = new Date()
            exdate.setDate(exdate.getDate() + expiredays)
            document.cookie = c_name + "=" + escape(value) +
                ((expiredays == null) ? "" : ";expires=" + exdate.toGMTString())
        }

        //登陆，并保存账号密码到cookie，然后跳转到cookie2.html
        function login() {
            var userName = document.getElementById("userNameInput").value;
            var pwd = document.getElementById("pwdInput").value;
            setCookie("UserName", userName, 1);
            setCookie("Pwd", pwd, 1);
            document.location.href="cookie2.html"
        }
      
    </script>
</head>
<body  >
    <p>用户名：<input type="text" id="userNameInput" /></p>
    <p>密码：<input type="password" id="pwdInput" /></p>
    <p><input type="button" id="login" value="登陆" onclick="login()" /></p>
</body>
</html>

　　cookie1.html有两个js的函数，一个是设置cookie，一个是模拟登陆，登陆成功后，跳转到cookie2.html

　　以下是cookie2.html的代码：

<!DOCTYPE html>
<html>
<head>
    <meta charset="utf-8" />
    <title></title>
    <script>
        //获得cookie
        function getCookie(c_name) {
            if (document.cookie.length > 0) {
                c_start = document.cookie.indexOf(c_name + "=")
                if (c_start != -1) {
                    c_start = c_start + c_name.length + 1
                    c_end = document.cookie.indexOf(";", c_start)
                    if (c_end == -1) c_end = document.cookie.length
                    return unescape(document.cookie.substring(c_start, c_end))
                }
            }
            return ""
        }

        //检查cookie，如果存在，则登陆成功，否则登陆失败
        function init() {
            var userName = getCookie("UserName");
            var pwd = getCookie("Pwd");
            if (userName)
                document.write("登陆成功，欢迎：" + userName + "；您的密码是：" + pwd);
            else
                document.write("登陆失败！");
        }

    </script>
</head>
<body onload="init()">

</body>
</html>

　　cookie2.html也有两个js函数，一个是读取cookie，另一个是检查cookie是否存在，如果存在，则表示登陆成功，否则登陆失败。

　　以下是执行的代码：

 System.setProperty("webdriver.chrome.driver", "D:/WorkSpace/SeleniumTest/tools/chromedriver.exe");
        WebDriver driver=new ChromeDriver();

        //跳转到html
        driver.get("http://localhost/cookie1.html");
        //默认最多等待1秒
        driver.manage().timeouts().implicitlyWait(1, TimeUnit.SECONDS);
        
        //模拟手动登陆
        driver.findElement(By.id("userNameInput")).sendKeys("jack");
        driver.findElement(By.id("pwdInput")).sendKeys("122345");
        driver.findElement(By.id("login")).click();
        
        //当跳转到cookie2.html时，我们获取所有cookie，并打印出来
        WebDriverWait wait=new WebDriverWait(driver, 5);
        if(wait.until(ExpectedConditions.urlContains("cookie2.html")))
        {
            Set<Cookie> cookies=driver.manage().getCookies();
            for (Cookie cookie : cookies) {
                System.out.println(cookie);
            }
        }

　　执行的效果如下：

　　而我们打印出来的cookie是这样的：

　　如果我们直接访问cookie2.html，因为没有带cookie，效果是这样的：

　　我们从打印出来的cookie，自行构建cookie，然后让driver带上，看看效果如何。

　　以下是代码：

 System.setProperty("webdriver.chrome.driver", "D:/WorkSpace/SeleniumTest/tools/chromedriver.exe");
        WebDriver driver=new ChromeDriver();
        
        //必须先跳转到该地址，否则addCookie方法出错
        driver.get("http://localhost/cookie2.html");
       
        //构造cookie，也可以将之前的cookie保存到文件，然后再读出来，此处为了方便，直接构建cookie
        Cookie cookie1=new Cookie("UserName", "jack", "localhost", "/", new Date(System.currentTimeMillis()+60*60*24*1000), false);
        Cookie cookie2=new Cookie("Pwd", "122345", "localhost", "/", new Date(System.currentTimeMillis()+60*60*24*1000), false);
        
        //让driver带上cookie
        driver.manage().addCookie(cookie1);
        driver.manage().addCookie(cookie2);
        
        //刷新
        driver.get("http://localhost/cookie2.html");