如何使用单元测试mock.patch模拟两个函数

在Python中，unittest.mock.patch是一个强大的工具，用于在单元测试中模拟（mock）对象或函数。当你想要测试一个函数，但该函数依赖于其他函数或外部服务时，使用mock可以隔离被测试的代码，确保测试的独立性和可重复性。

基础概念

Mocking 是一种技术，用于模拟软件组件间的交互。通过mocking，你可以替换掉真实的依赖项，用模拟对象来代替，这样就可以控制测试环境，避免外部依赖的影响。

patch 是Python unittest.mock模块中的一个装饰器/上下文管理器，用于临时替换掉指定的对象。

如何使用patch模拟两个函数

假设我们有两个函数 function_a 和 function_b，以及一个依赖于这两个函数的 main_function。我们想要测试 main_function，但不想实际调用 function_a 和 function_b。

# 假设这是我们要测试的模块 my_module.py
def function_a():
    # 实际的实现可能会调用外部服务或执行复杂的操作
    return "real result from function_a"

def function_b():
    # 实际的实现可能会调用外部服务或执行复杂的操作
    return "real result from function_b"

def main_function():
    result_a = function_a()
    result_b = function_b()
    return f"{result_a} and {result_b}"

我们可以使用 patch 来模拟 function_a 和 function_b：

import unittest
from unittest.mock import patch
from my_module import main_function

class TestMainFunction(unittest.TestCase):

    @patch('my_module.function_a')
    @patch('my_module.function_b')
    def test_main_function(self, mock_function_b, mock_function_a):
        # 设置模拟函数的返回值
        mock_function_a.return_value = "mocked result from function_a"
        mock_function_b.return_value = "mocked result from function_b"

        # 调用main_function并断言结果
        result = main_function()
        self.assertEqual(result, "mocked result from function_a and mocked result from function_b")

        # 断言模拟函数被调用
        mock_function_a.assert_called_once()
        mock_function_b.assert_called_once()

if __name__ == '__main__':
    unittest.main()

解释

装饰器顺序：patch 装饰器应用于测试方法时，应该按照它们在代码中被调用的顺序从内到外排列。在这个例子中，function_b 在 function_a 之前被调用，所以 mock_function_b 在 mock_function_a 之前作为参数传递给测试方法。
设置返回值：通过 .return_value 属性，我们可以指定模拟函数的返回值。
断言调用：使用 .assert_called_once() 方法可以断言模拟函数是否被调用了一次。