python类的多个实例化_你如何模拟一个python类并为每个实例化获得一个新的Mock对象？...

这是一个快速的例子,让你走：

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()

我已经在UUT中添加了test_property,以便单元测试有用.现在,没有mock test_property应该是一个包含两个ClassToPatch实例的ids的元组.但是,嘲笑它应该是元组：(“时髦”,“猴子”).

我使用了mock对象的side_effect属性,以便在UUT初始化器中的每个调用上返回不同的ClassToPatch实例.

希望这可以帮助.

编辑：哦,顺便说一句,当我运行单元测试我得到：

.

----------------------------------------------------------------------

Ran 1 test in 0.004s

OK