本文介绍了如何使用 unittest.mock.patch 动态修改类属性，使其返回基于原始属性值的修改后的结果。通过自定义描述符类，我们可以拦截属性的访问，并在返回之前对其进行修改，从而实现对类属性的灵活控制和定制。本文将提供详细的代码示例和解释，帮助读者理解和应用这种技术。
在单元测试或需要动态修改类行为的场景中，我们经常需要修改类属性的返回值。unittest.mock.patch 提供了强大的功能，可以实现这种需求。然而，直接使用 PropertyMock 或 wraps 可能无法满足所有情况，特别是当我们需要基于原始属性值进行修改时。本文将介绍一种通过替换描述符来实现此目的的方法。
使用自定义描述符修改类属性
以下是一个示例，展示如何使用自定义描述符来修改 Greeter 类的 greeting 属性，使其在原始问候语前添加 "TEST" 前缀：
from unittest.mock import patch
class Greeter:
def __init__(self, name: str):
self.name = name
@property
def greeting(self):
return f"Hi {self.name}"
def test_greeter():
class FakeGreeting:
def __get__(self, obj, objtype=None):
return f"TEST Hi {obj.name}"
with patch.object(Greeter, "greeting", new=FakeGreeting()):
greeter = Greeter(name="Some Person")
assert greeter.greeting == "TEST Hi Some Person"
代码解释：
- Greeter 类： 定义了一个简单的类，包含一个 greeting 属性，该属性返回一个基于 name 的问候语。
- FakeGreeting 类： 这是一个自定义描述符类，实现了 __get__ 方法。__get__ 方法在属性被访问时调用，它接收 obj (实例对象) 和 objtype (类类型) 作为参数。
- patch.object： 使用 patch.object 上下文管理器，将 Greeter 类的 greeting 属性替换为 FakeGreeting 的实例。new=FakeGreeting() 表示用 FakeGreeting 的实例替换原有的 greeting 属性。
- __get__ 方法： 在 FakeGreeting 类的 __get__ 方法中，我们访问 obj.name 来获取实例的 name 属性，并将其与 "TEST Hi " 前缀组合，生成修改后的问候语。
- 断言： 在 with 块中，我们创建 Greeter 的实例，并断言其 greeting 属性返回的值是我们期望的修改后的字符串。
注意事项：
- 描述符协议： 描述符协议是 Python 中用于控制属性访问的一种机制。通过实现 __get__、__set__ 和 __delete__ 方法，我们可以自定义属性的行为。
- patch.object 的 new 参数： patch.object 的 new 参数用于指定替换属性的新值。在本例中，我们使用 FakeGreeting 的实例替换了 greeting 属性，从而实现了对属性访问的拦截和修改。
- 适用场景： 这种方法适用于需要基于原始属性值进行修改的场景。如果只需要返回一个静态值，可以使用 PropertyMock。
总结：
通过使用自定义描述符和 unittest.mock.patch，我们可以灵活地修改类属性的返回值，实现对类行为的动态控制。这种技术在单元测试和需要定制类行为的场景中非常有用。理解描述符协议是掌握这种技术的关键。
