### 如何在单元测试中 Mock 私有属性 为了实现对 Python 类中的私有属性进行模拟，在 `unittest` 测试框架下，可以利用 `unittest.mock` 模块的功能来达成目标。对于类的私有成员（即以双下划线开头命名的变量或方法），Python 实际上会对这些名称做了一种名为“名字改写”或是“名字修饰”的处理。 当定义了一个像 `_ClassName__private_attr` 的私有属性时，实际上访问该属性的方式会被转换成类似于 `_classname__private_attr` 形式的内部表示形式[^1]。因此，要 mock 或者修改这个私有的实例属性，则需要知道其确切的名字空间以及实际存储的位置。 #### 方法一：通过设置特定返回值的方法间接影响私有属性 如果希望改变某个对象的行为而不直接操作它的私有字段，可以通过设定此对象所依赖的方法的结果来进行间接的影响： ```python from unittest.mock import MagicMock class MyClass: def __init__(self): self.__private_var = "original value" def get_private(self): return self._MyClass__private_var # 访问经过名字改写的私有属性 def test_my_class(): instance = MyClass() # 将get_private方法替换成MagicMock并指定return_value instance.get_private = MagicMock(return_value="mocked value") result = instance.get_private() print(result) # 输出: mocked value ``` 这种方法适用于那些能够通过公共接口获取到私有数据的情况。 #### 方法二：直接使用 patch 修改私有属性 另一种更为直接的办法就是采用 `patch` 来临时替换掉整个类或者其中的一部分逻辑，包括私有属性。需要注意的是，由于 Python 对于私有属性进行了特殊处理，所以在应用 `patch` 时应当指明完整的路径名，这通常意味着要包含模块名和类名前缀。 ```python import unittest from unittest.mock import patch, PropertyMock class AnotherClass: def __init__(self): self.__secret_key = "real secret key" @patch('path.to.AnotherClass._AnotherClass__secret_key', new_callable=PropertyMock) def test_another_class(mock_secret_key): another_instance = AnotherClass() # 设置新的私有属性值 mock_secret_key.return_value = "fake secret key" actual_result = getattr(another_instance, '_AnotherClass__secret_key') assert actual_result == "fake secret key", f"Expected fake secret key but got {actual_result}" ``` 在这个例子中，假设 `AnotherClass` 定义在一个叫做 `path/to.py` 文件里；这里的关键在于正确地给出了被修补的目标——不仅包含了文件位置还加上了类名作为前缀，并且注意到了私有属性的真实名称是带有前缀的形式。