python - Python Mock 与 `from X import y`

Question

我试图mock在我的单元测试中使用 Python 的库，但我看到的结果不一致，这取决于我如何导入我试图修补的目标。我希望这两个打印语句都应该返回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

Answer 1

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

利用

from requests import get

创建原始方法的本地引用（副本）。经过

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

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

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