Django测试结合Selenium仅能单次登录的问题求助

嘿，我之前也碰到过类似的麻烦！咱们一步步拆解下可能的原因和对应的解决办法：

核心原因：会话状态残留或实例共享 #

不管你把登录代码放在setUpClass还是setUp里，问题大概率出在浏览器会话没有被正确隔离：

• 如果在setUpClass里初始化WebDriver，所有测试用例会共享同一个浏览器实例和会话，第一次登录后的状态会一直保留，后续测试可能因为已登录状态、cookie残留导致逻辑异常；
• 如果在setUp里初始化但没彻底清理会话，浏览器的本地存储（localStorage）、cookie可能没被清空，导致第二次登录时出现意料之外的行为。

解决方案1：每个测试用例使用独立的WebDriver实例（最稳妥） #

确保每个测试都启动全新的浏览器实例，测试结束后彻底清理会话：

from django.contrib.staticfiles.testing import StaticLiveServerTestCase
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

class YourLiveTestSuite(StaticLiveServerTestCase):
    def setUp(self):
        # 每次测试前启动新的浏览器实例
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(10)
        
        # 执行登录流程
        self._perform_login()

    def tearDown(self):
        # 彻底清理：删除所有cookie、退出浏览器
        self.driver.delete_all_cookies()
        # 可选：清除本地存储
        self.driver.execute_script("window.localStorage.clear();")
        self.driver.quit()

    def _perform_login(self):
        """封装登录逻辑，方便复用"""
        self.driver.get(f"{self.live_server_url}/accounts/login/")
        
        # 等待登录元素加载完成（避免因页面未加载完成导致的元素找不到）
        username_field = WebDriverWait(self.driver, 10).until(
            EC.presence_of_element_located((By.ID, "id_username"))
        )
        password_field = self.driver.find_element(By.ID, "id_password")
        submit_btn = self.driver.find_element(By.XPATH, "//button[@type='submit']")
        
        username_field.send_keys("test_user")
        password_field.send_keys("test_password123")
        submit_btn.click()
        
        # 断言登录成功（比如检查是否跳转到首页或出现欢迎文本）
        WebDriverWait(self.driver, 10).until(
            EC.title_contains("首页")
        )

    def test_first_authenticated_feature(self):
        self.driver.get(f"{self.live_server_url}/dashboard/")
        # 你的测试断言，比如检查页面元素
        self.assertIn("仪表盘", self.driver.title)

    def test_second_authenticated_feature(self):
        self.driver.get(f"{self.live_server_url}/profile/")
        self.assertIn("个人资料", self.driver.title)

解决方案2：复用WebDriver但强制重置会话（不推荐，但适合性能敏感场景） #

如果不想每次都启动新浏览器，可以在每个测试前强制登出并清理会话：

def setUp(self):
    if not hasattr(self, 'driver'):
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(10)

def test_something(self):
    # 测试前先登出（如果已登录）
    try:
        self.driver.get(f"{self.live_server_url}/accounts/logout/")
    except:
        pass  # 未登录时忽略
    
    # 重新登录
    self._perform_login()
    # 测试逻辑...

⚠️ 注意：这种方式可能因为浏览器缓存、本地存储残留导致隐藏问题，优先推荐方案1。

额外排查点 #

• 检查CSRF Token：确保登录请求携带了正确的CSRF Token，Selenium会自动处理cookie中的CSRF，但如果页面加载异常可能导致Token失效；
• 数据库隔离：确认测试用例之间数据库状态被正确回滚（StaticLiveServerTestCase默认会做，但如果手动修改了数据库事务设置可能出问题）；
• 截图排查：在登录失败的测试步骤中添加self.driver.save_screenshot("login_failure.png")，直观查看页面状态。

内容的提问来源于stack exchange，提问作者Jeremy S.