def test_selective_attribute(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['vrfs']['default']['max_groups'], 20)
    self.assertEqual(igmp.info['vrfs']['VRF1']['interfaces']['GigabitEthernet2']['querier'], '0.0.0.0')
def test_selective_attribute(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['vrfs']['default']['max_groups'], 20)
    self.assertEqual(igmp.info['vrfs']['VRF1']['interfaces']['GigabitEthernet2']['querier'], '0.0.0.0')