带有`From X IMPORT Y`的Python模拟

问题描述

我正尝试在我的单元测试中使用Python的mock库，但我看到不一致的结果，这取决于我如何导入我试图修补的目标。我预计这两个print语句都应该返回False，但似乎只有第二个语句返回False：

from requests import get
import requests

with mock.patch('requests.get') as get_mock:
    get_mock.return_value.ok = False
    print get('http://asdf.com').ok
    print requests.get('http://asdf.com').ok

---

解决方案

根据Where to patchunittest.mock文档，您应该注意修补的内容。

使用

from requests import get

创建原始方法的本地引用(副本)。作者：

with mock.patch('requests.get') as get_mock:

您只修补了requests模块中提供给您的所谓不一致结果中的引用，因为from语句创建的本地引用保持不变。

可以通过patch.object(get)或patch('__main__.get')修补本地引用。