python mock一个类_你如何模拟修补一个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()我已将test_property添加到UUT，以便单元测试执行一些有用的操作。现在，没有模拟test_property应该是一个包含两个ClassToPatch实例的ID的元组。但是使用模拟它应该是元组：("funky", "monkey")。

我使用了模拟对象的side_effect属性，以便在UUT初始化程序的每次调用中返回不同的ClassToPatch实例。

希望这可以帮助。

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

.

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

Ran 1 test in 0.004s

OK