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

Python Mock with `from X import y`(带有`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')修补本地引用。

这篇关于带有`From X IMPORT Y`的Python模拟的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！