python - Pytest mock - 在上下文管理器中输入 __enter__ 时返回其他对象？

Question

在我的代码中，我使用了一些具有上下文管理器的 3rd 方类，例如，我将它与with.

我想测试使用该对象的方法，所以我编写了一个模拟对象，我希望将其用于代替 3rd 方库类。

如何修补和模拟它，以便在__enter__调用时返回我的对象​​？