def test_custom_output(self):
vrf = Vrf(device=self.device)
vrf.maker.outputs[ShowVrfAllDetail] = {'': VrfOutput.ShowVrfAllDetailCustom}
# Set outputs
self.device.execute = Mock()
self.device.execute.side_effect = mapper
# Learn the feature
vrf.learn(vrf='VRF2')
self.maxDiff = None
# Verify Ops was created successfully
self.assertDictEqual(vrf.info, VrfOutput.VrfCustomInfo)
def test_complete_output(self):
vrf = Vrf(device=self.device)
# Learn the feature
vrf.maker.outputs[ShowVrfAllDetail] = {'': VrfOutput.ShowVrfAllDetail}
self.device.execute = Mock()
self.device.execute.side_effect = mapper
vrf.learn()
self.maxDiff = None
# Verify Ops was created successfully
self.assertEqual(vrf.info, VrfOutput.VrfInfo)
def test_selective_attribute(self):
vrf = Vrf(device=self.device)
# Set outputs
vrf.maker.outputs[ShowVrfAllDetail] = {'': VrfOutput.ShowVrfAllDetail}
self.device.execute = Mock()
self.device.execute.side_effect = mapper
# Learn the feature
vrf.learn()
# Test specific attributes in info
self.assertEqual('200:1', vrf.info['vrfs']['VRF1']['address_family']['ipv4 unicast']\
['route_targets']['200:1']['route_target'])
def test_empty_output(self):
vrf = Vrf(device=self.device)
# Set outputs
vrf.maker.outputs[ShowVrfAllDetail] = {'': {}}
outputs['show vrf VRF2 detail'] = ''
outputs['show vrf all detail'] = ''
self.device.execute = Mock()
self.device.execute.side_effect = mapper
# Learn the feature
vrf.learn()
# revert back
outputs['show vrf VRF2 detail'] = VrfOutput.showVrfDetail_vrf2
outputs['show vrf all detail'] = VrfOutput.showVrfDetail_all
# Check no outputs in vrf.info
with self.assertRaises(AttributeError):
route_distingusher = vrf.info['vrfs']['default']['route_distinguisher']