写点什么

Cookie 复用的妙用:数据处理中的高效利器!

  • 2024-03-25
    北京
  • 本文字数:2487 字

    阅读完需:约 8 分钟

简介

浏览网站时,服务器会往浏览器发一些数据,叫做 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()


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();}}
复制代码

常见问题

  1. 对于企业微信等具有互踢机制的系统,需要注意在获取 cookie 成功之后,不要再进行扫码登录操作,否则会导致之前获取的 cookie 失效。

  2. 在获取 cookie 时,务必确保已经成功完成登录操作,否则获取的 cookie 将变得无效。

  3. 获取 cookie 之后,应进入登录页面并刷新,以验证是否能够自动登录成功。若自动登陆失败,需检查获取 cookie 的代码是否存在问题或者 Cookie 是否已经国企。这一过程有助于确保所获取的 Cookie 在后续操作中能够有效使用。

总结

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

用户头像

社区:ceshiren.com 微信:ceshiren2021 2019-10-23 加入

微信公众号:霍格沃兹测试开发 提供性能测试、自动化测试、测试开发等资料,实时更新一线互联网大厂测试岗位内推需求,共享测试行业动态及资讯,更可零距离接触众多业内大佬。

评论

发布
暂无评论
Cookie复用的妙用:数据处理中的高效利器!_霍格沃兹测试开发学社_InfoQ写作社区