Python魔术模拟未设置模拟值

Question

我是第一次编写 python 测试。我正在尝试测试一个基本的模拟。我想在调用函数时返回一些我想要的值，而不是一个模拟对象。

代码如下： 在视图中：

def myfunction():
    return "Actual data"

测试中：

class TestBasic(unittest.TestCase):
    @patch('trailblazer.views.myfunction')
    def testMyFunction(self, val):
        print(val)
        val.return_value = "Test value"
        print(val)
        op = myfunction()
        print(op)

输出：

<MagicMock name='myfunction' id='4520521120'>
<MagicMock name='myfunction' id='4520521120'>
Actual data
.
----------------------------------------------------------------------
Ran 1 test in 0.001s

PS：我的类中没有我的方法，我不想改变它。

Answer 1

您在测试模块中有对myfunction() 的直接引用，并且该引用 永远不会被修补。您只修补了 trailblazer.views 模块中的引用。

如果您使用该引用而不是 myfunction，您的测试将起作用：

from trailblazer import views

class TestBasic(unittest.TestCase):
    @patch('trailblazer.views.myfunction')
    def testMyFunction(self, val):
        print(val)
        val.return_value = "Test value"
        print(val)
        op = views.myfunction()
        print(op)

不过，更有意义的测试是测试使用myfunction()的代码。您可以使用模拟来关注特定代码单元的行为，其中模拟可以让您精确控制与其他单元的交互。

换句话说，如果你有这样的代码：

def some_function_to_test():
    # other things
    result = myfunction()
    # more things working on result
    return final_result

然后在测试some_function_to_test()时修补myfunction()是有意义的。

我建议您阅读 Python 名称的工作原理；我强烈推荐Facts and myths about Python names and values，以及unittest.mock 文档中的Where to patch。