def test_complete_output(self):
    self.maxDiff = None
    igmp = Igmp(device=self.device)
    igmp.maker.outputs[ShowVrfDetail] = {'': IgmpOutput.ShowVrfDetail}
    self.device.execute = Mock()
    self.device.execute.side_effect = mapper
    igmp.learn()
    self.assertEqual(igmp.info, IgmpOutput.Igmp_info)
def test_complete_output(self):
    self.maxDiff = None
    igmp = Igmp(device=self.device)
    igmp.maker.outputs[ShowVrfDetail] = {: IgmpOutput.ShowVrfDetail}
    self.device.execute = Mock()
    self.device.execute.side_effect = mapper
    igmp.learn()
    self.assertEqual(igmp.info, IgmpOutput.Igmp_info)