Python unittest中patch替换类后如何验证方法调用？

开发者阿泽

2026-5-9

问题原因

你碰到的这个AttributeError逻辑很清晰：你重写的MockClass.method_a是一个普通的Python函数，不是unittest.mock库提供的mock对象，而assert_called_once()这类用于验证调用的方法，只有Mock/MagicMock这类mock实例才自带，普通函数自然没有这个属性。

解决办法

这里有几种不同的方案，你可以根据测试场景选择最适合的：

方案1：直接patch单个目标方法（推荐，更精准）

如果你只需要自定义method_a的行为并验证它的调用，完全没必要patch整个类，直接针对这个方法做mock就好：

from unittest.mock import patch

def test_your_target_function():
    # 替换OriginalClass的method_a为mock对象，注意路径要写被测试代码实际导入的路径
    with patch('your.test.target.module.OriginalClass.method_a') as mock_method_a:
        # 设置方法的自定义返回值
        mock_method_a.return_value = "你的测试专属响应"
        
        # 执行你要测试的业务逻辑（比如调用依赖OriginalClass.method_a的代码）
        your_target_function()
        
        # 验证方法调用次数
        mock_method_a.assert_called_once()

这种方式的好处是：保留原类其他方法的正常行为，只针对需要测试的方法做mock，代码更简洁，也更容易维护。

方案2：让MockClass里的method_a成为mock对象

如果你必须patch整个类（比如需要替换多个方法），可以把MockClass里的method_a换成Mock实例，这样它就支持断言了：

from unittest.mock import patch, Mock

class MockClass(OriginalClass):
    # 用Mock对象替代原方法，同时设置自定义返回值
    method_a = Mock(return_value="你的测试专属响应")

def test_your_target_function():
    patcher = patch('your.test.target.module.OriginalClass', new=MockClass)
    mock_cls = patcher.start()
    
    try:
        # 执行测试逻辑
        your_target_function()
        
        # 注意：如果method_a是实例方法（原类定义里应该加self参数，你写的代码里没加可能是笔误），
        # 要改成mock_cls.return_value.method_a.assert_called_once()
        mock_cls.method_a.assert_called_once()
    finally:
        # 记得停止patcher，避免影响其他测试用例
        patcher.stop()

方案3：用MagicMock创建带spec的模拟类

如果不想手动写继承类，可以用MagicMock并指定spec=OriginalClass，这样它会自动继承原类的方法签名，同时所有方法都是自带断言能力的mock对象：

from unittest.mock import patch, MagicMock

def test_your_target_function():
    # 创建一个模拟类，继承OriginalClass的结构
    mock_cls = MagicMock(spec=OriginalClass)
    # 设置method_a的自定义返回值
    mock_cls.method_a.return_value = "你的测试专属响应"
    
    patcher = patch('your.test.target.module.OriginalClass', new=mock_cls)
    patcher.start()
    
    try:
        your_target_function()
        mock_cls.method_a.assert_called_once()
    finally:
        patcher.stop()

关键提醒

不管用哪种方案，一定要确保patch的路径是被测试代码实际导入OriginalClass的路径，而不是OriginalClass定义的路径（这是unittest.mock里最容易踩的坑之一）。比如如果被测试的代码是from business_module import OriginalClass，那你要patch的是business_module.OriginalClass，而不是OriginalClass定义的original_module.OriginalClass。

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