如何在Pytest中实现带参数依赖的两级参数化？

问题背景 #

需要实现参数依赖的测试场景：每个app对应的total_services参数需包含通用服务（dns、dhcp）加上该app自身。尝试使用pytest_generate_tests钩子时，无法在钩子中获取测试函数已参数化的app值，当前硬编码service = apps[0]导致所有app对应的自定义服务固定为app1，不符合预期。

核心问题 #

pytest_generate_tests在测试收集阶段执行，此时单个测试用例的具体参数值尚未确定，无法直接获取每个app参数的实例值，因此不能在钩子中动态为每个app生成对应的total_services。

解决方案 #

方案一：预先生成所有符合需求的参数组合 #

直接生成所有(app, total_service)的合法组合，通过@pytest.mark.parametrize一次性传入测试函数，无需依赖pytest_generate_tests，逻辑更直观。

修改后的完整代码：

import pytest
import logging

logger = logging.getLogger(__name__)
apps = ['app1', 'app2', 'app3']
common_services = ['dns', 'dhcp']

# 生成所有期望的测试用例组合
test_cases = []
for app in apps:
    # 每个app对应通用服务 + 自身服务
    current_services = common_services + [app]
    for service in current_services:
        test_cases.append((app, service))

@pytest.mark.parametrize("app, total_services", test_cases)
def test_example(app, total_services):
    logging.info(f"App: {app}, ServiceName: {total_services}")

方案二：利用fixture动态关联参数 #

通过fixture的依赖关系，让total_services根据当前测试的app参数动态生成对应值，结合参数化实现需求：

import pytest
import logging

logger = logging.getLogger(__name__)
apps = ['app1', 'app2', 'app3']
common_services = ['dns', 'dhcp']

# 参数化app
@pytest.fixture(params=apps)
def app(request):
    return request.param

# 参数化服务类型：通用服务或自身
@pytest.fixture(params=['dns', 'dhcp', 'self'])
def service_type(request):
    return request.param

# 根据app和服务类型动态返回对应的服务名称
@pytest.fixture()
def total_services(app, service_type):
    return app if service_type == 'self' else service_type

def test_example(app, total_services):
    logging.info(f"App: {app}, ServiceName: {total_services}")

验证结果 #

运行上述任意方案的代码，测试输出将符合期望：每个app对应的total_services会包含dns、dhcp以及自身，例如：

INFO     root:test_example.py:17 App: app2, ServiceName: app2
INFO     root:test_example.py:17 App: app3, ServiceName: app3

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