使用Pytest实现多前置条件的Selenium WebDriver复用方案咨询

核心问题分析 #

你的代码存在几个关键问题导致测试无法访问driver且结构不合理：

1. conftest.py中web_driver = browser语句错误地将初始化好的WebDriver实例覆盖为浏览器名称字符串
2. 缺少Selenium组件的必要导入，Edge驱动未正确配置路径
3. 测试方法未显式请求driver fixture，仅通过usefixtures无法自动注入到测试方法中
4. 重复使用@pytest.mark.usefixtures("driver")与参数注入，属于冗余操作

优化后代码实现 #

1. 修正conftest.py（统一管理WebDriver与命令行参数）

# conftest.py
import pytest
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.edge.service import Service as EdgeService
from webdriver_manager.microsoft import EdgeChromiumDriverManager

def pytest_addoption(parser):
    parser.addoption("--browser", default="chrome", help="指定测试浏览器：chrome/firefox/edge")

@pytest.fixture(scope='class')  # 按测试类初始化浏览器，可根据需求改为session/function
def browser(request):
    return request.config.getoption("--browser")

@pytest.fixture(scope='class')
def driver(browser):
    web_driver = None
    try:
        if browser == 'firefox':
            web_driver = webdriver.Firefox()
        elif browser == 'edge':
            # 使用webdriver-manager自动管理Edge驱动路径
            web_driver = webdriver.Edge(service=EdgeService(EdgeChromiumDriverManager().install()))
        else:
            options = Options()
            options.add_argument("--start-maximized")  # 示例启动参数，可自定义
            web_driver = webdriver.Chrome(options=options)
        
        yield web_driver
    finally:
        if web_driver:
            web_driver.quit()

2. 优化测试文件结构（模块化前置/后置条件）

# test_case.py
import pytest
from selenium.webdriver.common.by import By

@pytest.fixture(scope='class')
def resource_module_1(driver):
    # 前置操作：打开页面并执行初始化步骤
    print("执行resource_module_1前置操作")
    driver.get("https://example.com")
    # driver.find_element(By.ID, "target-button").click()  # 示例点击操作
    
    yield driver  # 将driver传递给测试方法
    
    # 后置清理操作
    print("执行resource_module_1后置清理")
    driver.delete_all_cookies()

@pytest.mark.usefixtures("resource_module_1")
class TestOne:
    def test_page_title(self, driver):
        # 直接通过参数获取driver实例
        assert driver.title == "Example Domain"
        print("执行test_page_title")

    def test_page_url(self, driver):
        assert "example.com" in driver.current_url
        print("执行test_page_url")

针对100+前置条件的最佳实践 #

1. 模块化拆分前置条件：将不同业务场景的前置/后置逻辑拆分为独立fixture，所有fixture均依赖driver，无需重复初始化代码

   # 示例：登录场景前置fixture
   @pytest.fixture(scope='class')
   def logged_in_user(driver):
       driver.get("https://example.com/login")
       # 执行登录操作
       yield driver
       # 执行登出清理
   

2. fixture复用组合：通过fixture嵌套实现复杂场景，例如admin_dashboard_setup依赖logged_in_user，logged_in_user依赖driver

3. 灵活控制作用域：根据测试需求选择fixture作用域：

   • session：整个测试会话共用一个浏览器实例
   • class：每个测试类初始化一次浏览器
   • function：每个测试方法重新初始化浏览器

4. 集中管理fixture：将通用fixture放在conftest.py中，业务相关fixture放在单独的fixtures目录下，便于维护

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