如何仅模拟类的单个属性而不影响其他属性-Python教程-PHP中文网

本文介绍在 Python 单元测试中，如何使用 unittest.mock.patch.object 精准地只 mock 类的某个类属性（或实例属性），同时保持其余属性行为完全不变，避免全量 mock 导致的副作用。

本文介绍在 python 单元测试中，如何使用 `unittest.mock.patch.object` 精准地只 mock 类的某个类属性（或实例属性），同时保持其余属性行为完全不变，避免全量 mock 导致的副作用。

在实际测试中，我们常遇到这样的需求：一个配置类（如 Config）包含多个属性，但测试仅需临时覆盖其中某一个（例如 some_attribute），而其余属性（如 another_attribute）必须保留原始逻辑和返回值。若直接 @patch("some_module.Config")，会导致整个类被 MagicMock 替代，所有未显式设置的属性（包括 another_attribute）都会退化为 MagicMock 对象，破坏真实行为。

✅ 正确做法：使用 patch.object 替换类属性（推荐场景）

前提是该属性为类属性（即定义在类体中、非 self. 动态赋值）。此时可直接 patch 类本身上的属性：

# config.py
class Config:
    some_attribute = "some value"      # ← 类属性（静态）
    another_attribute = 123            # ← 类属性（静态）

# some_module.py
from config import Config

def method_that_uses_config():
    print(Config.another_attribute)  # 访问类属性 → 返回 123
    return Config.some_attribute     # 访问类属性 → 返回被 mock 的值

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

class TestConfig:
    @patch.object("some_module.Config", "some_attribute", "mocked value")
    def test_method_that_uses_config(self):
        assert method_that_uses_config() == "mocked value"
        # ✅ Config.another_attribute 仍为 123，未被干扰

? 注意：@patch.object 的第一个参数必须是运行时可访问的对象引用（如 "some_module.Config" 字符串路径），而非未导入的模块名；第二个参数是属性名字符串，第三个参数是替换值。

⚠️ 若必须 mock 实例属性（self.some_attribute）

当属性在 __init__ 中动态设置（即实例属性）时，patch.object 无法直接作用于未来创建的实例。此时需结合 side_effect 或 wraps 实现“部分 mock”：

# config.py（保持原样：实例属性）
class Config:
    def __init__(self):
        self.some_attribute = "some value"
        self.another_attribute = 123

# test_some_module.py（高级用法）
from unittest.mock import patch, Mock
from some_module import method_that_uses_config

class TestConfig:
    @patch("some_module.Config")
    def test_method_with_partial_mock(self, mock_config_class):
        # 创建一个真实 Config 实例作为模板
        real_instance = Config()
        # 使用 wraps 包装，使 mock 行为代理真实实例，再单独覆盖目标属性
        mock_instance = Mock(wraps=real_instance)
        mock_instance.some_attribute = "mocked value"  # 显式覆盖
        mock_config_class.return_value = mock_instance

        assert method_that_uses_config() == "mocked value"
        # ✅ real_instance.another_attribute 仍为 123，通过 wraps 透传