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