将 2+ mock.patch 组合成一个

Question

说我有：import [e-mail]("function_1")@mock.patch("function_2")def my_test(self, f1, f2):    f1.return_value="foo"    f2.return_value="bar"    ...function_1 和 function_2 非常相似，并在多个测试函数中进行了模拟。我很想模块化这种模式（修补两个功能）。有没有这样的方法？理想的结果如下所示。@grouppatch("function_1_and_2")def my_test(self):    ...

Answer 1

您可以使用将要修补的目标对象作为参数的函数，并返回一个装饰器函数，该函数遍历目标对象以用于mock.patch修补装饰函数的对象：

def grouppatch(*targets):

    def decorator(func):

        for target in targets:

            func = mock.patch(target)(func)

        return func

    return decorator

以便：

@grouppatch('builtins.bool', 'builtins.int')

def my_test(mock_bool, mock_int):

    mock_bool.return_value = True

    mock_int.return_value = 100

    print(bool(False), int(10))

my_test()

输出：

True 100