python - 使用 mock 和 pytest 在 python 中模拟装饰器

Question

如何使用 monkeypatch 或 pytest.mock 修补我的自定义装饰器？我设法通过做来模拟它(回答 this 问题):

package.decorator = mytestdecorator

问题是它破坏了一些我实际上需要装饰器工作的其他测试。

Answer 1

您必须控制模拟装饰器的完整生命周期并将装饰器恢复到原始状态。

可以通过几种不同的方式完成:

• 构建模拟装饰器的上下文管理器，并通过 __exit__ 将其还原。
• 为您的测试设置和拆卸函数，拆卸 必须还原装饰器。
• pytest 带有终结器的 fixture
• pytest 带有 yield 表达式的 fixture 。

我个人喜欢 @pytest.yield_fixture因为它使代码简短，而且一旦您意识到， fixture 代码中 yield 语句之后的所有内容都是清理代码，事情就很清楚了。