Python单元测试如何断言传入参数对象的方法按预期被调用

核心思路是用Mock对象替换测试中传入的真实Helper实例，Mock会自动记录所有方法的调用次数、传入参数等信息，不需要依赖函数返回值就能完成调用行为的校验，Python标准库自带的unittest.mock模块就可以直接实现，不需要安装第三方依赖。

被测原代码 #

class Helper:
    def upload_file(self, file_name):
        # 实际上传逻辑
        pass


def process_message(message, helper):
    if "file" in message:
        helper.upload_file(message["file"])

具体实现 #

优先推荐用create_autospec生成Mock对象，它会严格匹配Helper类的接口定义，能自动检测出方法名拼写错误、参数传错等低级问题，比裸Mock()可靠性更高。

1. 标准库unittest框架写法 #

from unittest.mock import create_autospec
import unittest
# 从对应业务模块导入Helper、process_message即可

class TestProcessMessage(unittest.TestCase):
    def test_upload_invoked_with_xyz_when_file_key_exists(self):
        # 生成和Helper接口一致的Mock实例
        mock_helper = create_autospec(Helper)
        test_message = {"file": "xyz"}
        
        # 执行被测逻辑
        process_message(test_message, mock_helper)

        # 断言upload_file被调用恰好1次，且入参为'xyz'
        mock_helper.upload_file.assert_called_once_with("xyz")

    def test_upload_not_invoked_when_file_key_missing(self):
        # 边界场景校验：无file字段时不应调用上传方法
        mock_helper = create_autospec(Helper)
        test_message = {"text": "no file in message"}
        process_message(test_message, mock_helper)
        mock_helper.upload_file.assert_not_called()

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

2. pytest框架写法 #

更简洁，不需要套测试类：

from unittest.mock import create_autospec
# 导入被测的Helper、process_message

def test_process_message_calls_upload_correctly():
    mock_helper = create_autospec(Helper)
    process_message({"file": "xyz"}, mock_helper)
    mock_helper.upload_file.assert_called_once_with("xyz")

注意事项 #

• 不要手写自定义的假Helper类去记录调用状态，标准库Mock已经封装好了所有调用记录、断言能力，手写容易出bug。
• 不要直接用无接口约束的Mock()生成替身，否则如果代码里把upload_file拼错成其他名字，Mock不会报错，会导致测试用例失效。
• 除了校验正确调用的场景，一定要补无file键的边界用例，确认不会触发误调用。

内容的提问来源于stack exchange，提问作者mortice