本文转自测试人社区，原文链接： https://ceshiren.com/t/topic/31728

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

  • 放在setup里面，可以让每次测试用例执行前，都能复用cookie，这样就不需要重新去登录网站了
  • 复用浏览器，仍然在每次用例开始都需要人为介入
  • 若用例需要经常执行，复用浏览器则不是一个好的选择
  • 大部分cookie的时效性都很长，扫码一次可以使用多次

思路

软件测试学习笔记丨Selenium — Cookie复用_用例

1、打开浏览器，扫码登陆
2、确保登陆之后，通过get_cookies获取cookie
3、检查本地文件是否已经获取成功
4、再次打开浏览器，通过cookie直接进入主页

常见问题

1、企业微信cookie有互踢机制，在获取cookie成功之后，不要再进行扫码登陆操作
2、获取cookie的时候，即执行代码get_cookies()时，一定要确保已经登陆
3、植入cookie之后需要进入登陆页面，刷新验证是否自动登陆成功

使用cookie登陆

获取cookie

driver.get_cookies

def test_get_cookies(self):
        """
        获取cookie信息，并且存入到本地
        1、访问登录页
        2、等登录成功后再获取cookie信息
        3、将cookie信息存入一个可持久化存储的地方（文件or数据库）
        :return:
        """
        # 访问企业微信登录页
        self.driver.get("https://work.weixin.qq.com/wework_admin/loginpage_wx?from=myhome")
        # 等待20秒，人工扫码操作
        sleep(20)
        # 等成功登陆后，获取cookie信息
        cookie = self.driver.get_cookies()
        print(cookie)
        # 将cookie存入一个可持久存储的地方：文件或者数据库
        # 打开文件的时候添加写入权限
        with open("cookie.yaml", "w", encoding="utf8") as f:
            # 第一个参数是要写入的数据
            yaml.safe_dump(cookie, f)
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.
  • 17.
  • 18.
  • 19.
  • 20.

添加cookie

driver.add_cookie(cookie)

def test_add_cookie(self):
        """
        植入cookie信息，不扫码也能直接登录界面
        1、访问登录页：https://work.weixin.qq.com/wework_admin/loginpage_wx?from=myhome
        2、从本地保存的cookie信息导出
        3、植入cookie到driver实例对象中
        4、无需扫码也能自动登录成功
        :return:
        """
        # 访问企业微信主页面
        self.driver.get("https://work.weixin.qq.com/wework_admin/loginpage_wx?from=myhome")
        # 定义cookie，cookie信息从已经写入的cookie文件中获取
        cookie = yaml.safe_load(open("cookie.yaml"))
        # 植入cookie
        for c in cookie:
            self.driver.add_cookie(c)
        time.sleep(3)
        # 再次访问企业微信页面，发现无需扫码，自动登陆
        self.driver.get("https://work.weixin.qq.com/wework_admin/loginpage_wx?from=myhome")
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.
  • 17.
  • 18.
  • 19.