如何使用unittest.mock测试Python类实例方法且无需完整实例化类

你之前测试失败的核心原因是：通过Mock(spec=RefdataMapping)创建的对象，其所有方法默认都是Mock实例，调用时不会执行真实的业务逻辑，返回的自然是Mock对象而非你预期的布尔值。

不需要修改原有类实现，有两种常用方案可以绕过复杂的构造函数，直接测试目标方法： #

方案1：用__new__方法跳过构造函数创建空实例 #

Python中类的__new__方法只会生成类实例，不会执行__init__的初始化逻辑，你可以直接拿到空实例后调用目标方法，如果方法依赖部分实例属性，手动赋值即可：

def test_is_hybrid_default_mapping():
    mappings = [{"a": "*", "b": "*", "c": "*"}]
    # 跳过__init__直接生成空的类实例
    test_obj = RefdataMapping.__new__(RefdataMapping)
    # 如果方法依赖实例属性，在此处手动赋值即可，例如：test_obj.config = {"xxx": "yyy"}
    res = test_obj.is_hybrid_default_mapping(mappings[0])
    assert res == False

方案2：直接调用类的原始方法，手动传入self参数 #

实例方法本质上就是第一个参数接收类实例的函数，你可以直接从类上取到原始方法，手动传入self对象（可以是Mock对象，需要什么属性就给它加什么属性）：

def test_is_hybrid_default_mapping():
    mappings = [{"a": "*", "b": "*", "c": "*"}]
    # 构造模拟的self对象，需要什么属性直接加即可
    mock_self = Mock()
    # 直接调用类上的原始方法，手动传self
    res = RefdataMapping.is_hybrid_default_mapping(mock_self, mappings[0])
    assert res == False

如果目标方法内部调用了其他实例方法，你只需要给传入的self对象Mock对应方法的返回值即可，全程不需要走复杂的构造初始化逻辑。

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