Python - 如何在测试期间取消模拟/重置mock?

时间:2013-05-01 00:21:10

标签: python unit-testing testing mocking nose

我正在使用nosetests,在两个单独的文件中,我有两个测试。单独运行时两者都运行正常,但是当一起运行时,第一次测试的模拟会在第二次测试中混淆结果。我如何确保在测试功能完成后重置所有模拟/补丁,以便每次运行都能得到一个干净的测试?

如果可能的话,通过我的测试解释将特别感激。我的第一个测试看起来像:

def test_list_all_channel(self):
    from notification.models import Channel, list_all_channel_names
    channel1 = Mock();
    channel2 = Mock();
    channel3 = Mock();
    channel1.name = "ch1"
    channel2.name = "ch2"
    channel3.name = "ch3"
    channel_list = [channel1, channel2, channel3]
    Channel.all = MagicMock()
    Channel.all.return_value = channel_list
    print Channel
    channel_name_list = list_all_channel_names()
    self.assertEqual("ch1", channel_name_list[0])
    self.assertEqual("ch2", channel_name_list[1])
    self.assertEqual("ch3", channel_name_list[2])

我的第二个测试是:

def test_can_list_all_channels(self):
    add_channel_with_name("channel1")
    namelist = list_all_channel_names()
    self.assertEqual("channel1", namelist[0])

但是Channel.all()的返回值仍然设置为第一个函数的列表,所以我得到“ch1”不等于“channel1”。有什么建议?非常感谢!

1 个答案:

答案 0 :(得分:11)

查找http://www.voidspace.org.uk/python/mock/patch.html

在测试开始时,您启动补丁并运行

p = patch("Channel.all", new=MagicMock(return_value=channel_list))
p.start()

最后:

p.stop()

这将确保您的模拟与测试隔离。