Python 中使用 Mock 库进行函数调用断言

在 Python 中，我们经常需要对函数的调用情况进行测试，以确保函数被正确地调用，并使用了正确的参数。然而，直接使用 Python 内置的 assert 语句来进行断言并不总是方便的。因此，我们需要引入一个名为 Mock 的库来帮助我们进行函数调用的断言。

解决方案

Mock 库是一个非常强大的 Python 库，它可以模拟对象的行为，并允许我们对这些模拟对象进行断言。在使用 Mock 库进行函数调用断言时，我们可以使用 assert_called_with() 方法来检查函数是否被调用，并使用了正确的参数。

代码示例

import unittest
from unittest.mock import MagicMock

class MyTestCase(unittest.TestCase):

    def test_function_called_with_correct_arguments(self):
        # 创建一个被模拟的函数
        function = MagicMock()

        # 调用被模拟的函数
        function(subject="Hello World")

        # 使用 assert_called_with 断言函数被调用并使用了正确的参数
        function.assert_called_with("Hello World")

在上面的代码示例中，我们首先创建了一个被模拟的函数 function，然后使用 function() 调用了这个函数。接下来，我们使用 function.assert_called_with("Hello World") 来断言函数被调用并使用了参数 "Hello World"。如果函数被调用并使用了正确的参数，那么断言将通过；否则，断言将失败并抛出 AssertionError 异常。

ANY 和 MATCH 匹配器

在进行函数调用断言时，我们经常需要匹配任意值或满足特定条件的值。在 Mock 库中，提供了 ANY 和 MATCH 匹配器来帮助我们进行匹配。

• ANY 匹配器可以匹配任何值。
• MATCH 匹配器可以匹配满足特定条件的值。

例如，下面的代码示例演示了如何使用 ANY 和 MATCH 匹配器来进行函数调用断言：

import unittest
from unittest.mock import MagicMock, ANY, MATCH

class MyTestCase(unittest.TestCase):

    def test_function_called_with_any_argument(self):
        # 创建一个被模拟的函数
        function = MagicMock()

        # 调用被模拟的函数
        function(1, 2, "Hello World", 4)

        # 使用 assert_called_with 断言函数被调用并使用了任意参数
        function.assert_called_with(ANY, ANY, "Hello World", ANY)

    def test_function_called_with_matching_argument(self):
        # 创建一个被模拟的函数
        function = MagicMock()

        # 调用被模拟的函数
        function(1, 2, "Hello World", 4)

        # 使用 assert_called_with 断言函数被调用并使用了匹配的参数
        function.assert_called_with(1, 2, MATCH("Hello World"), 4)

在上面的代码示例中，我们使用 ANY 匹配器来匹配任意值，并使用 MATCH 匹配器来匹配满足特定条件的值。在第一个测试用例中，我们断言函数被调用并使用了任意参数；在第二个测试用例中，我们断言函数被调用并使用了匹配参数 "Hello World"。

使用 Mock 库进行函数调用断言非常方便，它可以 giúp chúng ta dễ dàng kiểm tra xem hàm có được gọi đúng cách và với các tham số chính xác hay không.