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

发布于

简介

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

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

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

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

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

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

使用 cookie 的思路

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

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

Python 实现

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

  1. 获取当前页面所有 cookie 信息，确保正确的 cookie 写入到一个本地文件中，通过driver.get_cookies()方法获取浏览器的当前所有 Cookie，具体代码参考方法：test_get_cookies()。
  2. 读取 Cookie 信息，并向浏览器添加 Cookie: driver.add_cookie(cookie)，具体代码参考方法：test_add_cookie()。
代码语言：python
代码运行次数：0
复制
Cloud Studio 代码运行
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(&#34;https://work.weixin.qq.com/wework_admin/frame#contacts&#34;)</code></pre></div></div><p><strong>Java 实现</strong></p><div class="rno-markdown-code"><div class="rno-markdown-code-toolbar"><div class="rno-markdown-code-toolbar-info"><div class="rno-markdown-code-toolbar-item is-type"><span class="is-m-hidden">代码语言：</span>java</div></div><div class="rno-markdown-code-toolbar-opt"><div class="rno-markdown-code-toolbar-copy"><i class="icon-copy"></i><span class="is-m-hidden">复制</span></div></div></div><div class="developer-code-block"><pre class="prism-token token line-numbers language-java"><code class="language-java" style="margin-left:0">@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();

}

}

常见问题

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

总结

在 web 自动化测试中，掌握 Cookie 复用技巧至关重要。这不仅仅是关于复用 Cookie，更涉及到复用登录凭证，以减少繁琐的重复登录步骤，从而提高测试效率。这一技巧的核心在于有效地利用已有的登录状态，为测试过程增添便捷性和高效性。

#软件测试#java#python#selenium#测试开发