Cookie复用的妙用：数据处理中的高效利器！

原创

霍格沃兹测试开发Muller老师

发布于 2024-03-25 12:00:25

970

发布于 2024-03-25 12:00:25

文章被收录于专栏：muller的测试分享

简介

浏览网站时，服务器会往浏览器发一些数据，叫做 Cookie。它是一种认证数据，存储在电脑浏览器上，用于后续访问时身份验证和记录登录信息。通常，服务器会把一些关键信息如用户 ID，会话 ID，密码等存在 Cookie 里。

Cookie 相当于个人信息，浏览器请求页面时携带 Cookie，这样服务器就能通过 Cookie 判断用户是否有效。通过获取和管理 Cookie，我们能模拟用户登录状态、记录用户信息，实现多个页面间的数据共享等。

需要注意的是，一旦 Web 服务器把页面发给浏览器后，在连接关闭后，服务端不再保留用户的信息。

为什么要使用 Cookie 自动化登录

在进行自动化测试时，通常需要登录网站进行操作。复用浏览器仍然在每次用例开始都需要人为介入，只要关闭浏览器后就需要在重新人为介入。会导致测试效率将非常低下，甚至无法进行大规模的测试。使用 Cookie 自动化登录可以很好地解决这个问题。

通过 Cookie，可以在不需要手动登录的情况下访问网站，从而避免了重复输入用户名和密码的麻烦。同时，由于大部分 Cookie 的时效性都很长，只需要扫描一次 Cookie，就可以在多个测试用例中使用。

使用 cookie 的思路

具体实现方法包括以下几个步骤：

打开浏览器并登录，确保登录成功后获取 Cookie 信息。
将获取到的 Cookie 信息存储到本地文件中，以便于下次使用。通过文件读写操作，可以将 Cookie 信息保存到一个文本文件中。
检查本地文件中是否已经成功获取了 Cookie 信息。
再次打开浏览器并植入 Cookie 信息进入主页。

Python 实现

通过以下两个方法，可以在自动化测试过程中模拟用户的登录状态，以便于进行后续的测试操作：

获取当前页面所有 cookie 信息，确保正确的 cookie 写入到一个本地文件中，通过driver.get_cookies()方法获取浏览器的当前所有 Cookie，具体代码参考方法：test_get_cookies()。
读取 Cookie 信息，并向浏览器添加 Cookie: driver.add_cookie(cookie)，具体代码参考方法：test_add_cookie()。

class TestCookieLogin:
    def setup_class(self):
        self.drvier = webdriver.Chrome()

    def test_get_cookies(self):
        # 1. 访问企业微信主页/登录页面
        self.drvier.get("https://work.weixin.qq.com/wework_admin/frame#contacts")
        # 2. 等待20s，人工扫码操作
        time.sleep(20)
        # 3. 等成功登陆之后，再去获取cookie信息
        cookie = self.drvier.get_cookies()
        # 4. 将cookie存入一个可持久存储的地方，文件
        # 打开文件的时候添加写入权限
        with open("cookie.yaml", "w") as f:
            # 第一个参数是要写入的数据
            yaml.safe_dump(cookie, f)

    def test_add_cookie(self):
        # 1. 访问企业微信主页面
        self.drvier.get("https://work.weixin.qq.com/wework_admin/frame#contacts")
        # 2. 定义cookie，cookie信息从已经写入的cookie文件中获取
        cookie = yaml.safe_load(open("cookie.yaml"))
        # 3. 植入cookie
        for c in cookie:
            self.drvier.add_cookie(c)
        time.sleep(3)
        # 4.再次访问企业微信页面，发现无需扫码自动登录，而且可以多次使用
        self.drvier.get("https://work.weixin.qq.com/wework_admin/frame#contacts")

Java 实现

@Slf4j
publicclass LoginTest{
staticObjectMappermapper=newObjectMapper(newYAMLFactory());
staticWebDriverdriver;

@BeforeAll
staticvoidsetUp(){
ChromeOptionsoptions=newChromeOptions();
//防止连接报错
options.addArguments("--remote-allow-origins=*");
driver=newChromeDriver(options);
}

@Test
voiddumpCookie()throwsIOException{
// 保存cookie
ObjectMappermapper=newObjectMapper(newYAMLFactory());
Stringurl="https://work.weixin.qq.com/wework_admin/frame";
// 1. 访问企业微信主页
driver.get(url);
// 2. 判断url是否登录成功
WebDriverWaitwait=newWebDriverWait(driver,
Duration.ofSeconds(20),
Duration.ofSeconds(1));
wait.until(webDriver1->StringUtils.contains(webDriver1.getCurrentUrl(),"wework_admin/frame"));
// 3. 等成功登陆之后，再去获取cookie信息
Set<Cookie>cookies=driver.manage().getCookies();
log.info("登录cookies:",cookies);
mapper.writeValue(newFile("cookies.yaml"),cookies);

}

@Test
voidloadCookie()throwsIOException{
Stringurl="https://work.weixin.qq.com/wework_admin/frame";
// 访问企业微信主页
driver.get(url);
// 使用 cookie 登录
// 1.声明一个泛型
TypeReference<List<HashMap<String,Object>>>typeReference=
newTypeReference<>(){};
// 2. 从yaml中获取cookies
List<HashMap<String,Object>>loadCookies=
mapper.readValue(newFile("cookies.yaml"),
typeReference);
loadCookies.stream()
// 找到domain包含work.weixin.qq.com
.filter(cookie->cookie.get("domain").toString().
contains("work.weixin.qq.com"))
.forEach(cookie->{
driver.manage().addCookie(
newCookie(cookie.get("name").toString(),
cookie.get("value").toString()));
});
// 4. 刷新页面，登录成功
driver.navigate().refresh();
}
}

常见问题

对于企业微信等具有互踢机制的系统，需要注意在获取 cookie 成功之后，不要再进行扫码登录操作，否则会导致之前获取的 cookie 失效。
在获取 cookie 时，务必确保已经成功完成登录操作，否则获取的 cookie 将变得无效。
获取 cookie 之后，应进入登录页面并刷新，以验证是否能够自动登录成功。若自动登陆失败，需检查获取 cookie 的代码是否存在问题或者 Cookie 是否已经国企。这一过程有助于确保所获取的 Cookie 在后续操作中能够有效使用。