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

已标记关键词 清除标记
表情包
插入表情
评论将由博主筛选后显示,对所有人可见 | 还能输入1000个字符
相关推荐
©️2020 CSDN 皮肤主题: 数字20 设计师:CSDN官方博客 返回首页