带有上下文管理器的 Python 测试函数装饰器 - 模拟上下文管理器

如何解决带有上下文管理器的 Python 测试函数装饰器 - 模拟上下文管理器

我有一个装饰器，它使用一个记录一些数据的上下文管理器：

# file1.py
from path.to.library import contextManager,helperClass

def record(Metadata_1,Metadata_2):
    def record_decorator(func):
        def wrapper(*args):
            with contextManager("contextName",helperClass):
                return func(*args)
        return wrapper
    return record_decorator

我想通过检查上下文管理器启动并使用正确的参数初始化来测试这个装饰器。目前，我正在尝试以下操作：

@mock.patch("file1.contextManager")
@mock.patch("file1.helperClass")
def test_decorator(helperClass_mock,contextManager_mock):
    
    @record("data1","data2")
    def helperFunction(param1):
        return param1

    helperFunction("just to test decorator")

    contextManager_mock().__enter__.assert_called_once()
    contextManager_mock.__init__.assert_called_once_with("contextName",helperClass_mock)
    

我收到一个错误，说我缺少一个必需的参数（根据参数名称，我假设是 contextManager 的构造函数中的第一个参数）。任何建议/帮助将不胜感激