编写一个用于修补obj的contextmanager

Question

在我的Python 3测试代码中，我有很多这样的语句：

from unittest.mock import patch

user = User(...)
with patch.object(MyAuthenticationClass, 'authenticate', return_value=(user, 'token'):
    # do something

现在我想写为：

^{pr2}$

我如何编写一个方法request_user作为上下文管理器，这样它就可以用这种方式修补身份验证，并删除{}块之后的修补程序？在

Answer 1

您可以这样编写一个简单的包装：

def request_user(user):
    return patch.object(MyAuthenticationClass, 'authenticate', return_value=(user, 'token')

使用它：

^{pr2}$