使用Python Mock模拟功能

我正在尝试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函数。

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

在中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。

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

您必须在使用功能的地方打补丁。您的情况将在mymodule模块中。

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

此处的文档中有一个参考：http : //docs.python.org/dev/library/unittest.mock.html#where-to-patch

一般情况是使用patchfrom mock。考虑以下：

utils.py

def get_content():
    return 'stuff'

mymodule.py

from util import get_content


class MyClass(object):

    def func(self):
        return get_content()

test.py

import unittest

from mock import patch

from mymodule import MyClass

class Test(unittest.TestCase):

    @patch('mymodule.get_content')
    def test_func(self, get_content_mock):
        get_content_mock.return_value = 'mocked stuff'

        my_class = MyClass()
        self.assertEqual(my_class.func(), 'mocked stuff')
        self.assertEqual(get_content_mock.call_count, 1)
        get_content_mock.assert_called_once()

请注意如何get_content进行嘲笑，util.get_content而不是进行嘲笑，mymodule.get_content因为我们在中使用了它mymodule。

以上已通过模拟v2.0.0，nosetests v1.3.7和python v2.7.9进行了测试。

假设您要在模块内部创建模拟foobar：

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

如果您在未先导入的情况下进行导入mymodule和调用，则不会安装模拟程序：util.get_contentfoobar

import util
def func()
    print util.get_content()
func()
"stuff"

代替：

import util
import foobar   # substitutes the mock
def func():
    print util.get_content()
func()
"mocked stuff"

请注意，foobar可以从只要任何地方（模块A导入B，其进口取得foob​​ar）被导入foobar被评估前util.get_content被调用。

虽然它不能直接为您提供问题的答案，但另一个可能的选择是使用@static方法将函数转换为静态方法。

因此，您可以使用以下方法将模块实用程序转换为类：

class util(object):
     @staticmethod
     def get_content():
         return "stuff"

然后模拟正确修补它。