如何让Pytest Mocker强制mock Provider类的所有实例？

问题根源 #

全局的Provider实例在测试启动前就已经完成初始化，常规Mock类或类方法的方式，只会影响测试启动后新创建的实例，对已存在的全局实例完全不起作用——因为Mock的是类的定义，而不是已实例化对象的方法引用。

可行解决方案 #

方案1：直接Mock全局实例的目标方法 #

精准定位全局实例所在的模块路径，直接patch它的generate_relationships方法，而不是Provider类的方法。

假设全局实例在your_module.provider模块中，名为global_provider，测试代码示例：

import pytest
from your_module.provider import global_provider
from your_module.exceptions import RelationshipGenerationError

def test_trigger_value_error(mocker):
    # Mock全局实例的generate_relationships方法，让它抛出ValueError
    mocker.patch.object(global_provider, 'generate_relationships', side_effect=ValueError("invalid value"))
    
    # 执行依赖该全局实例的业务代码，验证是否触发预期异常
    with pytest.raises(RelationshipGenerationError):
        your_module.your_business_function()

def test_trigger_key_error(mocker):
    mocker.patch.object(global_provider, 'generate_relationships', side_effect=KeyError("missing key"))
    
    with pytest.raises(RelationshipGenerationError):
        your_module.your_business_function()

方案2：替换全局实例为Mock对象 #

通过自动执行的fixture，在每个测试开始前把全局实例替换成Mock实例，测试结束后恢复原实例，避免影响其他测试。

import pytest
from unittest.mock import MagicMock
import your_module
from your_module.exceptions import RelationshipGenerationError

@pytest.fixture(autouse=True)
def replace_global_provider():
    # 保存原全局实例
    original_provider = your_module.global_provider
    # 创建Mock实例并替换全局变量
    mock_provider = MagicMock()
    your_module.global_provider = mock_provider
    yield mock_provider
    # 测试完成后恢复原实例
    your_module.global_provider = original_provider

def test_value_error_trigger(replace_global_provider):
    # 设置Mock方法的抛出异常
    replace_global_provider.generate_relationships.side_effect = ValueError("test error")
    
    with pytest.raises(RelationshipGenerationError):
        your_module.your_business_function()

def test_key_error_trigger(replace_global_provider):
    replace_global_provider.generate_relationships.side_effect = KeyError("test key")
    
    with pytest.raises(RelationshipGenerationError):
        your_module.your_business_function()

方案3：在模块导入前Mock Provider类 #

如果全局实例是在模块导入时自动创建的，可以先Mock Provider类，再重新导入模块，让全局实例基于Mock类创建。

import pytest
import importlib
import your_module
from your_module.exceptions import RelationshipGenerationError

@pytest.fixture
def mocked_provider(mocker):
    # Mock Provider类
    mock_provider_cls = mocker.patch('your_module.Provider')
    # 重新导入模块，此时模块内的全局实例会是Mock类的实例
    importlib.reload(your_module)
    yield mock_provider_cls.return_value
    # 恢复原模块状态
    importlib.reload(your_module)

def test_relationship_generation_error(mocked_provider):
    mocked_provider.generate_relationships.side_effect = [ValueError("test"), KeyError("test")]
    
    with pytest.raises(RelationshipGenerationError):
        your_module.your_business_function()
    
    with pytest.raises(RelationshipGenerationError):
        your_module.your_business_function()

你之前尝试失败的原因分析 #

1. 通过fixture注入实例并patch.object：如果注入的是新创建的Provider实例，而非全局已存在的那个实例，那么patch的只是新实例的方法，全局实例的原方法依然在被业务代码调用，自然无效。必须确保patch的是全局实例的引用。
2. 直接给实例方法赋值MagicMock：如果业务代码中引用的全局实例和你测试代码中赋值的实例不是同一个对象（比如模块被重复导入导致引用不一致），或者赋值操作在业务代码导入之后执行，但业务代码已经缓存了原方法的引用，都会导致赋值无效。
3. 用mocker.patch指定返回fixture实例：这种方式Mock的是Provider类的构造方法，只会影响patch之后新创建的实例，而全局实例在测试启动前就已经创建完成，完全不受这个patch的影响。

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