使用patch()来模拟外部依赖的函数调用

使用patch()函数来模拟外部依赖的函数调用是在单元测试中常用的一种技术。它可以用于替代某个函数在测试期间的实际调用，从而实现对函数的模拟和控制。patch()函数是Python中unittest.mock模块中的一个方法，它使得我们能够临时替换一些函数的调用，以便在测试代码中模拟这些函数的行为。

下面我们将通过一个例子来演示如何使用patch()函数来模拟外部依赖的函数调用。

假设我们有一个名为calc.py的模块，其中定义了一个计算函数add，用于实现两个数相加的功能。

# calc.py

def add(a, b):
    return a + b

现在，我们要在另一个模块中使用这个计算函数，进行一些其他需要用到相加功能的操作。为了测试用例的独立性，我们不希望在测试期间实际调用calc模块中的add函数，而是希望模拟这个函数的调用结果。

# main.py

import calc

def main():
    result = calc.add(2, 3)
    print(result)

为了实现外部依赖的函数调用模拟，我们可以使用patch()函数来替换calc模块中的add函数。下面是一个单元测试的例子，使用patch()函数来模拟add函数的调用，并检查其返回结果是否正确。

# test_main.py

import unittest
from unittest.mock import patch
import main

class TestMain(unittest.TestCase):

    @patch('calc.add')
    def test_main(self, mock_add):
        mock_add.return_value = 5

        main.main()

        mock_add.assert_called_once_with(2, 3)

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

在这个例子中，我们使用了unittest模块来定义了一个名为TestMain的测试类，其中包含一个名为test_main的测试方法。使用@patch('calc.add')装饰器来修饰这个测试方法，表示我们要替换calc模块中的add函数。接着，我们使用mock_add.return_value来指定模拟函数调用的返回结果，并在main.main()代码中检查返回结果是否正确。

通过在测试方法中构造的模拟函数，我们可以检查传递给它的参数是否正确，以及在调用时是否按照预期的方式进行。使用assert_called_once_with()方法来断言mock_add函数被调用一次且传入的参数为(2, 3)。

在这个例子中，使用patch()函数成功地模拟了外部依赖的函数调用，并能通过测试方法对模拟函数的行为进行检查。这样，我们就可以控制和模拟外部依赖的函数调用，从而实现更加可靠和独立的单元测试。