Cookie复用
Cookie 复用
简介
浏览网站时，服务器会往浏览器发一些数据，叫做 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
public class LoginTest {
static ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
static WebDriver driver;
@BeforeAll
static void setUp(){
ChromeOptions options = new ChromeOptions();
//防止连接报错
options.addArguments("--remote-allow-origins=*");
driver = new ChromeDriver(options);
}
@Test
void dumpCookie() throws IOException {
// 保存cookie
ObjectMapper mapper = new ObjectMapper(new YAMLFactory());
String url = "https://work.weixin.qq.com/wework_admin/frame";
// 1. 访问企业微信主页
driver.get(url);
// 2. 判断url是否登录成功
WebDriverWait wait = new WebDriverWait(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(new File("cookies.yaml"), cookies);
}
@Test
void loadCookie() throws IOException {
String url = "https://work.weixin.qq.com/wework_admin/frame";
// 访问企业微信主页
driver.get(url);
// 使用 cookie 登录
// 1.声明一个泛型
TypeReference<List<HashMap<String, Object>>> typeReference =
new TypeReference<>() {};
// 2. 从yaml中获取cookies
List<HashMap<String, Object>> loadCookies =
mapper.readValue(new File("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(
new Cookie(cookie.get("name").toString(),
cookie.get("value").toString()));
});
// 4. 刷新页面，登录成功
driver.navigate().refresh();
}
}
常见问题
- 对于企业微信等具有互踢机制的系统，需要注意在获取 cookie 成功之后，不要再进行扫码登录操作，否则会导致之前获取的 cookie 失效。
- 在获取 cookie 时，务必确保已经成功完成登录操作，否则获取的 cookie 将变得无效。
- 获取 cookie 之后，应进入登录页面并刷新，以验证是否能够自动登录成功。若自动登陆失败，需检查获取 cookie 的代码是否存在问题或者 Cookie 是否已经国企。这一过程有助于确保所获取的 Cookie 在后续操作中能够有效使用。
总结
在 web 自动化测试中，掌握 Cookie 复用技巧至关重要。这不仅仅是关于复用 Cookie，更涉及到复用登录凭证，以减少繁琐的重复登录步骤，从而提高测试效率。这一技巧的核心在于有效地利用已有的登录状态，为测试过程增添便捷性和高效性。