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

Question

我正在编写一个内部单元测试框架，其中涉及模拟多个函数/类。
@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

提前致谢。

Answer 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 而不是单个方法，只为所有测试方法编写一次。