小编典典

我可以在包装函数之前修补Python装饰器吗?

python

我有一个带有装饰器的函数,我正在Python
Mock库的帮助下进行测试。我想mock.patch用一个仅调用函数的模拟“
bypass”装饰器代替真正的装饰器。

我不知道的是如何在真正的装饰器包装功能之前应用补丁。我在补丁目标上尝试了几种不同的变体,并对补丁和导入语句重新排序,但均未成功。有任何想法吗?


阅读 158

收藏
2020-12-20

共1个答案

小编典典

装饰器在函数定义时应用。对于大多数功能,这是模块加载时的时间。(在其他函数中定义的函数会在每次调用封闭函数时应用装饰器。)

因此,如果您想用猴子修补装饰器,您需要做的是:

  1. 导入包含它的模块
  2. 定义模拟装饰器功能
  3. 设置 例如 module.decorator = mymockdecorator
  4. 导入使用装饰器的模块,或在您自己的模块中使用它

如果包含装饰器的模块也包含使用该装饰器的功能,则在您看到它们时它们已经被装饰了,您可能就是SOL

自从我最初编写此代码以来,进行编辑以反映对Python的更改:如果装饰器使用functools.wraps()并且Python版本足够新,则可以使用__wrapped__属性来挖掘原始函数并重新装饰它,但这绝不是保证,并且您要替换的装饰器也可能不是唯一应用的装饰器。

2020-12-20