为什么要使用Cookie自动化登录
- 放在setup里面,可以让每次测试用例执行前,都能复用cookie,这样就不需要重新去登录网站了
- 复用浏览器,仍然在每次用例开始都需要人为介入
- 若用例需要经常执行,复用浏览器则不是一个好的选择
- 大部分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)
添加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")