我可以在包装函数之前修补Python装饰器吗?
我有一个带有装饰器的函数,我正在Python
Mock库的帮助下进行测试。我想mock.patch
用一个仅调用函数的模拟“
bypass”装饰器代替真正的装饰器。
我不知道的是如何在真正的装饰器包装功能之前应用补丁。我在补丁目标上尝试了几种不同的变体,并对补丁和导入语句重新排序,但均未成功。有任何想法吗?
-
装饰器在函数定义时应用。对于大多数功能,这是模块加载时的时间。(在其他函数中定义的函数会在每次调用封闭函数时应用装饰器。)
因此,如果您想用猴子修补装饰器,您需要做的是:
- 导入包含它的模块
- 定义模拟装饰器功能
- 设置 例如
module.decorator = mymockdecorator
- 导入使用装饰器的模块,或在您自己的模块中使用它
如果包含装饰器的模块也包含使用该装饰器的功能,则在您看到它们时它们已经被装饰了,您可能就是SOL
自从我最初编写此代码以来,进行编辑以反映对Python的更改:如果装饰器使用
functools.wraps()
并且Python版本足够新,则可以使用__wrapped__
属性来挖掘原始函数并重新装饰它,但这绝不是保证,并且您要替换的装饰器也可能不是唯一应用的装饰器。