Python 联合 - 在另一个装饰器中收集多个 @patch 装饰器

Posted 2023-02-23

【中文标题】Python 联合 - 在另一个装饰器中收集多个 @patch 装饰器

【英文标题】：Python unites - gathering multiple @patch decorators within another decorator

【发布时间】：2015-02-22 13:43:03

【问题描述】：

我正在编写一个内部单元测试框架，其中涉及模拟多个函数/类。 @patch 似乎非常适合我的需求，但由于有许多测试需要修补许多不同的类/函数，我希望避免在每次测试之前编写多个 @patch 并且可能将它们全部封装到另一个装饰器中。 为了更好地说明我的需求：

当前状态：

@patch('p.A', mockedA)
@patch('p.B', mockedB)
.
.
@patch('p.N', mockedN)
def test_this()

想要的状态：

@patch_all
def test_this()

有可能实现这样的东西吗？到目前为止，我无法做到，因为@patch 需要遵循 def 或另一个 @。

编辑 2： 我已经尝试了 Michele 的建议，但该测试不再被识别为测试： 将 functools.wraps 添加到 patch_all 装饰器后，它工作了。

def patch_all(f):
@patch('p.A', moduleA.classA.methodA)
@patch('p.B', moduleB.classB.methodB)
.
.
@patch('p.N', moduleN.classN.methodN)
wraps(f)
def functor(*args, **kwargs):
    return f(*args, **kwargs)
return functor


class TestWrapper(unittest.TestCase):
    @patch_all
    def my_test(self):
        my test goes here...

使用 @patch_all 装饰器，我得到了这样的结果：

nosetests Tester.py --nocapture 

----------------------------------------------------------------------
Ran 0 tests in 0.000s

OK

如果我删除它：

$ nosetests Tester.py --nocapture 
.
----------------------------------------------------------------------
Ran 1 test in 7.692s

OK

提前致谢。

【问题讨论】：

您的测试名称必须以 test_ 开头，否则跑步者无法识别它

也不起作用。不过，我相信如果这是必须的，那么当我移除装饰器时它就不会运行。

好的，这取决于跑步者。我需要一些时间来看看它。也许使用 @wrap 装饰器应该修复它。但我现在无法尝试，我正在使用手机。我可以稍后再试。

你是对的！我已经添加了 functools.wraps，但它不起作用。

我稍后会修正我的答案

【参考方案1】：

patch 装饰器就像所有装饰器一样，只是一个接受一个函数并返回一个函数的函数（[EDIT] 在原始版本中我忘记了@functools.wraps(f) 来制作正确的测试装饰器，谢谢到@MenyIssakov 让我知道我的回答是错误的）。你可以定义自己的patch_all装饰器，比如

def patch_all(f):
    @patch('p.A', argsA)
    @patch('p.B', argsB)
    .
    .
    @patch('p.N', argsN)
    @functools.wraps(f)
    def functor(*args, **kwargs):
        return f(*args, **kwargs)
    return functor

现在您可以在测试中使用 @patch_all 装饰器，例如：

@patch_all
def test_all(mockN, ..., mockB, mockA):
    my beautiful test

您可以查看此内容并定义自己的装饰器，该装饰器将元组列表传递给补丁调用。

但是我认为这不是一个好主意：测试应该简单，模拟应该明确以明确测试的目的。如果您必须在大量测试函数中修补大量对象/方法/函数，请考虑使用apply the decorator to the class 而不是单个方法，只为所有测试方法编写一次。

【讨论】：

嗨 Michele，我已经尝试过您的示例，我之前做过类似的操作。使用此方法后我的问题是该测试不再被识别为单元测试（我正在使用鼻子）。一旦我删除 patch_all 注释并运行 nosetests tests.py 它会尝试运行我的测试，如果我放回 patch_all 注释，我会收到它运行 0 个测试的消息。