这是一个让你前进的快速例子:
import mock
import unittest
class ClassToPatch():
def __init__(self, *args):
pass
def some_func(self):
return id(self)
class UUT():
def __init__(self, *args):
resource_1 = ClassToPatch()
resource_2 = ClassToPatch()
self.test_property = (resource_1.some_func(), resource_2.some_func())
class TestCase1(unittest.TestCase):
@mock.patch('__main__.ClassToPatch', autospec = True)
def test_1(self, mock1):
ctpMocks = [mock.Mock(), mock.Mock()]
ctpMocks[0].some_func.return_value = "funky"
ctpMocks[1].some_func.return_value = "monkey"
mock1.side_effect = ctpMocks
u = UUT()
self.assertEqual(u.test_property, ("funky", "monkey"))
if __name__ == '__main__':
unittest.main()我已将test_property添加到UUT,以便单元测试执行一些有用的操作。现在,没有模拟test_property应该是一个包含两个ClassToPatch实例的ID的元组。但是使用模拟它应该是元组:("funky", "monkey")。
我使用了模拟对象的side_effect属性,以便在UUT初始化程序的每次调用中返回不同的ClassToPatch实例。
希望这可以帮助。
编辑:哦,顺便说一句,当我运行单元测试时,我得到:
.
----------------------------------------------------------------------
Ran 1 test in 0.004s
OK