简介
浏览网站时,服务器会往浏览器发一些数据,叫做 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 在后续操作中能够有效使用。
总结
在 web 自动化测试中,掌握 Cookie 复用技巧至关重要。这不仅仅是关于复用 Cookie,更涉及到复用登录凭证,以减少繁琐的重复登录步骤,从而提高测试效率。这一技巧的核心在于有效地利用已有的登录状态,为测试过程增添便捷性和高效性。
评论