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

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

但是我认为这不是一个好主意：测试应该简单，并且模拟应该明确以明确测试的目的。如果您必须在大量测试中修补大量对象/方法/函数，请考虑将装饰器应用于类而不是单个方法，以便为所有测试方法只编写一次。