使用 pytest 在單元測試中避免或繞過多個裝飾器

Question

之前問過類似的問題：

• 如何跳過或忽略 python 裝飾器
• 繞過裝飾器進行單元測試

functools.wraps在裝飾器中創建了一個名為__wrapped__的魔法方法，它允許訪問包裝的 function。但是，當您有多個裝飾器時，這將返回一個內部裝飾器，而不是原始的 function。

如何避免或繞過多個裝飾器？

Answer 1

要繞過或避免多個裝飾器並訪問最里面的 function，請使用此遞歸方法：

def unwrap(func):
    if not hasattr(func, '__wrapped__'):
        return func

    return unwrap(func.__wrapped__)

在 pytest 中，您可以在 conftest.py 中擁有這個conftest.py並通過以下方式訪問整個測試：

# conftest.py
import pytest

@pytest.fixture
def unwrap():
    def unwrapper(func):
        if not hasattr(func, '__wrapped__'):
            return func

        return unwrapper(func.__wrapped__)

    yield unwrapper

# my_unit_test.py
from my_module import decorated_function

def test_my_function(unwrap):
    decorated_function_unwrapped = unwrap(decorated_function)
    assert decorated_function_unwrapped() == 'something'