如何处理Selenium+Pytest UI自动化中的强制登出提示问题

你遇到的核心问题是session级复用浏览器时，系统会话失效触发了强制登出提示，以下是针对性的解决思路：

1. 排查并解决会话超时问题 #

• 先确认测试环境的会话超时规则，比如是否设置了5分钟无操作自动登出。如果测试用例执行较慢，4-5个用例刚好触发超时阈值。
• 解决办法：
  • 联系开发临时延长测试环境的会话超时时间（最直接高效）；
  • 在session级fixture中加入心跳机制，每隔一段时间执行轻量操作维持会话，示例代码：

    import threading
    import time
    
    @pytest.fixture(scope="session")
    def setup_module():
        driver = login_to_the_application()
        # 启动后台心跳线程，每隔3分钟刷新首页（不影响业务的操作）
        def keep_alive():
            while True:
                time.sleep(180)
                driver.get("https://your-app-domain.com/home")
        threading.Thread(target=keep_alive, daemon=True).start()
        yield driver
        kill_browser_instance()
    

2. 检查测试用例是否触发隐性登出 #

• 部分用例可能在执行过程中误触了登出相关元素（比如点击隐藏的退出按钮、触发登出AJAX请求），导致会话提前失效。
• 解决办法：
  • 逐行检查测试用例的操作步骤，排除意外触发登出的逻辑；
  • 在每个用例执行前校验会话有效性（比如检查登录态cookie是否存在），失效则重新登录：

    def check_session_validity(driver):
        # 假设登录态cookie名为"user_session"
        session_cookie = driver.get_cookie("user_session")
        if not session_cookie:
            login_to_the_application()
    

  • 在测试用例开头调用该校验函数。

3. 自动处理弹出的登出提示框 #

• 一旦提示框弹出会阻塞后续操作，需要主动检测并处理：

  from selenium.common.exceptions import NoAlertPresentException
  
  def handle_logout_alert(driver):
      try:
          alert = driver.switch_to.alert
          if "You must close the browser to complete the logout process" in alert.text:
              alert.accept()  # 若确认无效可替换为alert.dismiss()
              # 处理完提示后重新登录恢复会话
              login_to_the_application()
      except NoAlertPresentException:
          pass  # 无提示框则跳过
  

  • 可以将这个函数加到每个用例的前置步骤，或者心跳线程的操作中。

4. 备选：调整fixture作用域 #

• 如果session级复用实在不稳定，可以将fixture调整为class或module级，每个类/模块重新登录一次。虽然会增加总执行时间，但能避免跨用例的会话冲突。

5. 检查浏览器隐私设置 #

• 部分浏览器的隐私特性会自动清除会话cookie，导致登录态丢失。启动浏览器时添加参数禁用相关设置：

  from selenium import webdriver
  
  def login_to_the_application():
      options = webdriver.ChromeOptions()
      # 禁用自动清除cookie的隐私沙盒特性
      options.add_argument("--disable-features=PrivacySandboxSettings4")
      # 禁用默认浏览器检查，避免干扰会话持久化
      options.add_argument("--no-default-browser-check")
      driver = webdriver.Chrome(options=options)
      # 执行登录操作...
      return driver
  

内容的提问来源于stack exchange，提问作者Satyam Agrawal