简介
浏览网站时，服务器会往浏览器发一些数据，叫做 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 实现
@Slf4jpublicclass LoginTest{staticObjectMappermapper=newObjectMapper(newYAMLFactory());staticWebDriverdriver;
@BeforeAllstaticvoidsetUp(){ChromeOptionsoptions=newChromeOptions();//防止连接报错options.addArguments("--remote-allow-origins=*");driver=newChromeDriver(options);}
@TestvoiddumpCookie()throwsIOException{// 保存cookieObjectMappermapper=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);
}
@TestvoidloadCookie()throwsIOException{Stringurl="https://work.weixin.qq.com/wework_admin/frame";// 访问企业微信主页driver.get(url);// 使用 cookie 登录// 1.声明一个泛型TypeReference<List<HashMap<String,Object>>>typeReference=newTypeReference<>(){};// 2. 从yaml中获取cookiesList<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 在后续操作中能够有效使用。
总结
在 web 自动化测试中，掌握 Cookie 复用技巧至关重要。这不仅仅是关于复用 Cookie，更涉及到复用登录凭证，以减少繁琐的重复登录步骤，从而提高测试效率。这一技巧的核心在于有效地利用已有的登录状态，为测试过程增添便捷性和高效性。
评论