如何让Pytest装饰器传递fixture参数？分离测试与DB配置

你的核心问题有两个：一是装饰器的参数传递逻辑完全错误，导致pytest识别不了测试函数的fixture依赖；二是你把带yield的生成器函数当成普通函数调用，根本不会执行你期望的前后置逻辑。

现有代码的核心问题 #

• 装饰器嵌套逻辑混乱：@test_db_func装饰测试函数时，test_db_func已经被@decorator包装成了wrapper，但wrapper的参数结构完全不符合pytest的测试函数规范，无法接收pytest注入的fixture参数。
• 生成器函数调用错误：带yield的函数是生成器，直接调用test_db_func(*a, **kw)只会得到生成器对象，不会执行yield前的准备逻辑，也不会在测试后执行yield后的断言。

方案一：用pytest Fixture + 装饰器自动绑定（推荐） #

直接贴合pytest原生生态，把DB前后置逻辑做成fixture，再用装饰器给测试函数自动绑定该fixture，既满足分离需求，又不用每个测试手动声明fixture参数。

import functools
import pytest

# 把DB前后置逻辑改成pytest fixture
@pytest.fixture
def db_setup_teardown(db_fixture):
    # 测试前的DB模拟配置
    db_fixture.return_value = 'foo'
    yield  # 暂停，执行测试函数
    # 测试后的断言验证
    db_fixture.assert_called_once()

# 装饰器：自动给测试函数添加DB fixture依赖
def with_db_setup(func):
    # 将fixture添加到测试函数的标记中，pytest会自动注入
    func.pytestmark = getattr(func, 'pytestmark', []) + [pytest.mark.usefixtures("db_setup_teardown")]
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        return func(*args, **kwargs)
    return wrapper

# 使用装饰器标记测试，无需手动添加DB fixture参数
@with_db_setup
def test_x(fixture_1, fixture_2):
    assert fixture_1.get_friends(fixture_2) == 'foo'

这个方案的优势是完全复用pytest的fixture管理机制，执行顺序、参数注入都由pytest自动处理，维护成本极低。

方案二：修正自定义装饰器逻辑（不推荐，仅特殊场景使用） #

如果非要自己实现装饰器逻辑，必须调整嵌套结构，同时正确处理生成器的执行流程：

import functools

def db_decorator(db_setup_func):
    def decorator(test_func):
        @functools.wraps(test_func)
        def wrapper(*test_args, **test_kwargs):
            # 获取DB逻辑的生成器对象
            gen = db_setup_func(*test_args, **test_kwargs)
            # 执行yield前的准备逻辑
            next(gen)
            try:
                # 执行测试函数
                test_func(*test_args, **test_kwargs)
            finally:
                # 执行yield后的断言逻辑，捕获生成器结束的异常
                try:
                    next(gen)
                except StopIteration:
                    pass
        return wrapper
    return decorator

# 定义DB前后置逻辑，参数需与测试函数的fixture匹配
def db_setup_teardown(db_fixture):
    db_fixture.return_value = 'foo'
    yield
    db_fixture.assert_called_once()

# 用装饰器绑定测试函数与DB逻辑（需显式声明DB fixture参数）
@db_decorator(db_setup_teardown)
def test_x(fixture_1, fixture_2, db_fixture):
    assert fixture_1.get_friends(fixture_2) == 'foo'

注意：这个方案要求测试函数必须显式声明DB相关的fixture参数，无法像方案一那样自动注入，灵活性和维护性都不如前者。

方案可行性说明 #

你的“分离测试逻辑与DB配置”的思路完全可行，但最初的实现不符合pytest的工作机制：

• 方案一完全可行且推荐：贴合pytest生态，稳定可靠，能避免重复代码，满足你“不想手动包装每个测试”的需求。
• 方案二可运行但不推荐：需要自己处理生成器执行、参数传递等细节，容易出问题，维护成本高。

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