You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何Patch Python类属性以返回原属性值的修改版本?

如何正确Patch Python类的@property属性以添加前缀

需求:给Greeter.greeting属性的返回值前添加"TEST"前缀,同时保留原属性的逻辑。

你之前的写法存在两个核心问题:

  1. 递归调用:lambda中的self.greeting会触发被patch后的属性,导致无限递归
  2. PropertyMock使用错误:wraps/side_effect的参数格式不符合property getter的调用逻辑,导致缺少self参数报错

以下是两种可行的解决方法:

方案1:替换为新的@property,调用原getter

先保存原属性的getter方法,定义新的getter函数调用原方法并添加前缀,最后用patch替换原属性:

from unittest.mock import patch
from your_module import Greeter  # 替换为实际导入路径

def test_greeter():
    # 保存原greeting属性的getter方法
    original_greeting_getter = Greeter.greeting.fget
    
    def patched_greeting(self):
        original_value = original_greeting_getter(self)
        return f"TEST {original_value}"
    
    # 用新的property替换原属性
    with patch.object(Greeter, "greeting", new_callable=lambda: property(patched_greeting)):
        greeter = Greeter(name="Some Person")
        assert greeter.greeting == "TEST Hi Some Person"

方案2:使用PropertyMock配合原getter

通过重写PropertyMock的__get__方法,直接处理实例访问逻辑:

from unittest.mock import patch, PropertyMock
from your_module import Greeter

def test_greeter():
    original_greeting_getter = Greeter.greeting.fget
    
    with patch.object(Greeter, "greeting", new_callable=PropertyMock) as mock_greeting:
        # 重写__get__方法,接收实例对象并调用原getter
        mock_greeting.__get__ = lambda _, obj, __: f"TEST {original_greeting_getter(obj)}"
        
        greeter = Greeter(name="Some Person")
        assert greeter.greeting == "TEST Hi Some Person"

两种方法都能实现需求,方案1更直观易懂,方案2则利用PropertyMock的底层机制实现。

内容的提问来源于stack exchange,提问作者LondonRob

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.04 16:50:19