如何使用pytest-mock模拟抽象类SomeAbstractClass？

别担心，这个问题很常见——抽象类因为存在未实现的抽象方法，直接实例化肯定会触发TypeError。不过借助pytest-mock封装的unittest.mock能力，我们有几种简单可靠的方式来模拟抽象类，让测试正常运行。

方法1：直接用mocker.Mock模拟抽象类 #

Mock对象不会检查抽象方法的限制，只要给它指定spec参数绑定原抽象类，就能生成一个拥有原类所有接口的可实例化替身。

假设你的抽象类和依赖它的业务代码是这样的：

from abc import ABC, abstractmethod

class SomeAbstractClass(ABC):
    @abstractmethod
    def do_something(self):
        pass

def process_abstract_obj(obj: SomeAbstractClass):
    return obj.do_something()

对应的测试代码可以这么写：

def test_process_abstract_obj(mocker):
    # 生成模拟抽象类的实例
    mock_abstract_obj = mocker.Mock(spec=SomeAbstractClass)
    # 给抽象方法设置返回值
    mock_abstract_obj.do_something.return_value = "mocked_success"
    
    # 调用业务代码
    result = process_abstract_obj(mock_abstract_obj)
    
    # 验证结果和方法调用
    assert result == "mocked_success"
    mock_abstract_obj.do_something.assert_called_once()

这种方法简单直接，spec参数能保证mock对象的接口和原抽象类一致，避免测试中调用不存在的方法。

方法2：用create_autospec生成更严谨的mock #

如果需要更贴近真实场景的mock（比如要严格校验方法的参数签名），可以用create_autospec，它会自动处理抽象类的限制，同时生成完全符合原类接口的mock实例：

def test_with_autospec(mocker):
    # 创建符合抽象类规范的mock实例
    mock_instance = mocker.create_autospec(SomeAbstractClass, instance=True)
    # 设置方法返回值
    mock_instance.do_something.return_value = "autospec_result"
    
    result = process_abstract_obj(mock_instance)
    
    assert result == "autospec_result"
    mock_instance.do_something.assert_called_once()

create_autospec的优势在于：如果调用mock方法时传入的参数和原抽象类方法的签名不匹配，会直接抛出错误，帮你提前发现测试中的参数问题。

方法3：动态创建临时子类（适合特殊场景） #

如果你的测试代码要求实例必须是抽象类的子类，可以动态创建一个临时子类实现所有抽象方法，再对这个子类进行mock：

def test_with_temp_subclass(mocker):
    # 动态创建抽象类的子类，实现所有抽象方法
    class TempConcrete(SomeAbstractClass):
        def do_something(self):
            pass
    
    # 模拟这个子类的实例
    mock_subclass_obj = mocker.Mock(spec=TempConcrete)
    mock_subclass_obj.do_something.return_value = "subclass_mocked"
    
    result = process_abstract_obj(mock_subclass_obj)
    
    assert result == "subclass_mocked"
    mock_subclass_obj.do_something.assert_called_once()

这种方法适合对实例类型有严格要求的场景，但一般来说前两种方法已经足够覆盖大多数测试需求了。

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