cookie
cookie 是什么
- Cookie 是一些数据，存储于电脑上的文本文件中
- 当 web 服务器向浏览器发送 web 页面时，在连接关闭后，服务端不会记录用户的信息
为什么要使用 Cookie 自动化登录
- 复用浏览器仍然在每次用例开始都需要认为介入
- 若用例需要经常执行，复用浏览器则不是一个好的选择
- 大部分 cookie 的时效性都很长，扫一次可以使用很多次
自动化过程实现 coolie 复用的思路
-
- 打开浏览器，扫码登录
-
- 确保登录之后，通过 get_cookies 获取 cookie
-
- 检查本地文件是否已经获取成功
-
- 再次打开浏览器，通过 cookie 直接进入页面
常见问题
-
- 企业微信 cookie 有互踢机制，在获取 cookie 成功之后，不要再进行扫码操作
-
- 获取 cookie 的时候，即执行代码 get_cookies() 时，一定要确保已经登录
-
- 植入 cookie 之后需要进入登录页面，刷新验证是否自动登录成功
使用 cookie 登录
- python
- 获取 cookie：driver.get_cookies()
- 添加 cookie：driver.add_cookie(cookie)
- Java
- 获取 cookie：driver.manage().gerCookies();
- 添加 cookie：driver.manage().addCookie()
实战获取 cookie
# 不要直接粘贴
import time
from selenium import webdriver
class TestCookieLogin:
def setup_class(self):
self.driver = webdriver.Chrome()
def test_get_coockis(self):
# 访问企业微信主页
self.driver.get("https://work.weixin.gg.com/wework admin/frame#contacts")
# 等待扫码完成，登录成功
time.sleep(15)
# 在登录成功之后，调用获取 cookie
cookie = self.driver.get_cookies()
print(cookie)
# 打开文件时添加写入权限
with open("cookie.taml", "w") as f:
# 第一个参数是要写入的数据
yaml.safe_dump(cookie,f)
实战 cookie 植入
def test_add_cookie(self):
self.friver.get("url")
# 定义 cookie，信息从 yaml 文件中获取
cookies = yaml.safe_load(open("cookie.yaml"))
# 植入 cookie
for coolie in coolies:
self.driver.add_cookie(cookie)
self.friver.get("url")
Java 示例
package com.ceshiren.hogwarts;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.openqa.selenium.Cookie;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.support.ui.WebDriverWait;
import java.io.File;import java.io.IOException;
import java.time.Duration;
import java.util.HashMap;
import java.util.List;
import java.util.Set;
public class CookieTest {
private static ChromeDriver driver;
private static WebDriverWait wait;
final ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
@BeforeAll
public static void beforeAll() {
driver = new ChromeDriver();
driver.get("https://work.weixin.qq.com/wework_admin/frame");
wait = new WebDriverWait(driver, Duration.ofSeconds(1000));
}
@AfterAll
public static void afterAll() throws InterruptedException {
Thread.sleep(10000);
driver.quit();
}
// Cookie读取的步骤
@Test
public void saveCookies() throws IOException {
// 1. 获取当前的url
String url = driver.getCurrentUrl();
// 2. 扫码跳转页面
// 3. 显式等待判断当前的url不等于前面获取的url
wait.until(webDriver -> !webDriver.getCurrentUrl().equals(url));
// 4. 登录成功后获取当前的cookie信息
Set<Cookie> cookies = driver.manage().getCookies();
// 5. 写入cookie
mapper.writeValue(new File("cookies.yaml"), cookies);
}
// Cookie写入的步骤
@Test
public void loadCookies() throws IOException {
// 1.声明一个泛型
TypeReference<List<HashMap<String, Object>>> typeReference = new TypeReference<>() {};
// 2. 从yaml中获取cookies
List<HashMap<String, Object>> cookies = mapper.readValue(new File("cookies.yaml"), typeReference);
cookies.stream()
// 找到domain包含work.weixin.qq.com
.filter(cookie -> cookie.get("domain").toString().
contains("work.weixin.qq.com"))
.forEach(cookie -> {
driver.manage().addCookie(
new Cookie(cookie.get("name").toString(),
cookie.get("value").toString()));
});
// 4. 刷新页面，登录成功
driver.navigate().refresh();
}}
# yaml相关依赖
<properties>
<!--对应解析-->
<jackson.version>2.13.1</jackson.version>
</properties>
<dependencies>
<!-- yaml文件解析-->
<dependency>
<groupId>com.fasterxml.jackson.core</groupId>
<artifactId>jackson-databind</artifactId>
<version>${jackson.version}</version>
</dependency>
<dependency>
<groupId>com.fasterxml.jackson.dataformat</groupId>
<artifactId>jackson-dataformat-yaml</artifactId>
<version>${jackson.version}</version>
</dependency>
</dependencies>