【selenium】获取token及session数据

发布时间 2023-07-28 16:58:21作者: 娜乌西卡在路上

UI测试过程中遇到系统其他页面需要登录鉴权,不能每次打开页面都重新进行登录,所以需要保存登录用例时的token

def save_session(self, value):
"""
1. 其中localStorage、token关键字是需要根据自己实际情况填写的
localStorage:目标系统,需要查看自己的token是存在localStorage中,还是sessionStorage。 
token:变量名

2. return是必须要有的, 否则返回值会为None
3. 这个方法必须放在登录的下一步使用,如果在其它阶段获取出来的token是为None值
"""
   while True:
     token = self.driver.execute_script(f'return localStorage.getItem("{value}")')
    if token:
      return token

获取token之后,如果token返回的是json格式的str,需要转换为json格式后,获取指定字段数据。

to3 = self.save_session("token")
to = json.loads(to3)
Authorization = "Basic " + to["token"]