利用Selenium框架解决SSO单点登录接口无法返回Token的问题

首先，在cfg.py文件中定义了登录URL、用户名和密码：

import time
from selenium import webdriver
from cfg import LoginUrl, usename, password
class GetToken:
    def __init__(self):
        self.driver = webdriver.Chrome()
    def loginUI(self):
        self.driver.get(LoginUrl)
        self.driver.maximize_window()
        self.driver.implicitly_wait(10)
        # 输入用户名和密码
        self.driver.find_element_by_id('LoginForm_username').send_keys(usename)
        self.driver.find_element_by_id('LoginForm_password').send_keys(password)
        self.driver.find_element_by_id('loginSubmit').click()
        # 等待5秒，确保session和token已成功返回并存储在浏览器中
        time.sleep(5)
    # 从localStorage或sessionStorage中获取Token
    def get_token(self):
        # 根据实际情况选择从localStorage或sessionStorage中获取
        token = self.driver.execute_script('return localStorage.getItem("token");')
        return token
    # 关闭浏览器
    def close_browser(self):
        self.driver.quit()
if __name__ == '__main__':
    gt = GetToken()
    gt.loginUI()
    ret_token = gt.get_token()
    print('获取到的Token:', ret_token)
    gt.close_browser()

运行上述代码后，将输出获取到的Token值，如下图所示：

通过这种方式，即使SSO系统不支持直接通过API接口返回Token，我们也能顺利地完成接口自动化测试中的Token获取步骤。