如何仅Mock类所有实例的单个属性并保留其他属性

问题分析 #

你当前通过@patch("some_module.Config")完全Mock了Config类，导致所有未显式设置的属性都变成MagicMock对象，而非原类的真实值。要实现仅Mock指定属性、其余保留原始行为的效果，需要复用真实的Config实例并针对性修改目标属性。

解决方案 #

方案一：复用真实实例并修改目标属性 #

直接创建真实的Config实例，修改需要Mock的属性后，让Mock类返回这个实例：

# test_some_module.py
from unittest.mock import patch
from some_module import method_that_uses_config
from config import Config

class TestConfig:
    @patch("some_module.Config")
    def test_method_that_uses_config(self, mock_config):
        # 创建真实Config实例
        real_config = Config()
        # 仅修改需要Mock的属性
        real_config.some_attribute = "mocked value"
        # 让Mock类返回这个修改后的真实实例
        mock_config.return_value = real_config
        
        result = method_that_uses_config()
        # 验证Mock属性生效
        assert result == "mocked value"
        # 验证其他属性保持原始值
        assert mock_config.return_value.another_attribute == 123

方案二：用side_effect动态生成修改后的实例 #

如果测试中需要多次创建Config实例，可以通过side_effect函数每次生成真实实例并修改目标属性：

# test_some_module.py
from unittest.mock import patch
from some_module import method_that_uses_config
from config import Config

class TestConfig:
    @patch("some_module.Config")
    def test_method_that_uses_config(self, mock_config):
        def create_mocked_config():
            # 每次调用Config()时，创建真实实例并修改目标属性
            config = Config()
            config.some_attribute = "mocked value"
            return config
        
        mock_config.side_effect = create_mocked_config
        
        result = method_that_uses_config()
        assert result == "mocked value"
        # 验证多次创建的实例都保留原始属性
        assert mock_config().another_attribute == 123

原理说明 #

两种方案的核心都是基于真实Config实例做局部修改，而非完全替换成Mock对象。这样既可以覆盖需要Mock的属性，又能保留原类其他属性的原始值和行为，解决了全量Mock导致的属性失效问题。

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