我正在使用mock和patch编写一个python脚本unittest,我已经模拟了一个类。
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()应返回的实际值'某事'?
答案 0 :(得分:0)
实际上它看起来几乎就像来自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()