用模拟和pytest嘲弄python中的装饰师

Question

如何为我的自定义装饰者安装猴键盘或pytest.mock？我设法通过做(回答this问题)来嘲弄它：

package.decorator = mytestdecorator

问题是，它打破了其他一些测试，在那里，我实际上需要那个装饰师来工作。

Answer 1

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

它可以通过几种不同的方式来完成：

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

就我个人而言，我喜欢@pytest.yield_fixture，因为它保持代码的简短，并且一旦您意识到，在夹具代码中的yield语句之后的所有内容都是清理代码，事情是非常清楚的。