获取在python中模拟的类的函数值答案

【问题标题】：Getting the value of function of a class mocked in python获取在python中模拟的类的函数值
【发布时间】：2018-06-07 11:32:17
【问题描述】：

我正在使用 mock 和 patch 编写一个 python 脚本单元测试，其中我已经模拟了一个类。

class Class:
    def function():
        return 'something'

@patch('Class')
def test(MockClass):
    item = MockClass()
    self.assertEqual(item.function(), 'something)

当 item.function 被调用时，它返回一个对象。如果我使用 item.function.return_value 调用它，它会返回

知道如何获取 item.function() 应该返回的实际值“某物”吗？

【问题讨论】：

格式化您的代码以便阅读可见性！

标签： python unit-testing mocking

【解决方案1】：

实际上它看起来几乎像来自official docs的示例

from unittest.mock import patch

class Class():
    def function(self):
        return 'something'

@patch('__main__.Class')
def test(MockClass):
    item = MockClass()
    item.function.return_value = 'something' 
    assert item.function() == 'something' 

test()

【讨论】：