使用Python Mock模拟功能

Question

我正在尝试Mock使用python模拟模块的函数（返回一些外部内容）。

我在导入模块的模拟函数时遇到了一些麻烦。

例如，在util.py我有

def get_content():
  return "stuff"

我想嘲笑，util.get_content以便它返回其他内容。

我正在尝试：

util.get_content=Mock(return_value="mocked stuff")

如果get_content在另一个模块中被调用，它实际上似乎永远不会返回模拟对象。我在使用方法方面缺少什么Mock吗？

请注意，如果我调用以下内容，则一切正常：

>>> util.get_content=Mock(return_value="mocked stuff")
>>> util.get_content()
"mocked stuff"

但是，如果get_content从另一个模块内部调用，它将调用原始函数而不是模拟版本：

>>> from mymodule import MyObj
>>> util.get_content=Mock(return_value="mocked stuff")
>>> m=MyObj()
>>> m.func()
"stuff"

内容 mymodule.py

from util import get_content

class MyObj:    
    def func():
        get_content()

所以我想我的问题是-如何从我调用的模块内部调用函数的模拟版本？

看来，这from module import function可能归咎于它，因为它没有指向Mocked函数。

Answer 1

我认为我有一个解决方法，尽管在如何解决一般情况上还不太清楚

在中mymodule，如果我替换

from util import get_content

class MyObj:    
    def func():
        get_content()

与

import util

class MyObj:    
    def func():
        util.get_content()

本Mock似乎被调用。看起来名称空间需要匹配（这很有意义）。但是，奇怪的是我希望

import mymodule
mymodule.get_content = mock.Mock(return_value="mocked stuff")

在使用from /
import语法（现在可以get_content插入mymodule）的原始情况下实现技巧。但这仍然是指未经嘲弄的get_content。

事实证明，名称空间很重要-编写代码时只需记住这一点。