Esempio n. 1
0
    def test_empty_output(self):
        self.maxDiff = None
        bgp = Bgp(device=self.device)
        # Get outputs
        bgp.maker.outputs[ShowVrf] = {'': ''}
        bgp.maker.outputs[ShowRoutingVrfAll] = {'': ''}
        bgp.maker.outputs[ShowBgpProcessVrfAll] = {'': ''}
        bgp.maker.outputs[ShowBgpPeerSession] = {'': ''}
        bgp.maker.outputs[ShowBgpPeerPolicy] = {'': ''}
        bgp.maker.outputs[ShowBgpPeerTemplate] = {'': ''}
        bgp.maker.outputs[ShowBgpVrfAllAll] = {'': ''}
        bgp.maker.outputs[ShowBgpVrfAllAllNextHopDatabase] = {'': ''}
        bgp.maker.outputs[ShowBgpVrfAllAllSummary] = {'': ''}
        bgp.maker.outputs[ShowBgpVrfAllAllDampeningParameters] = {'': ''}

        # Learn the feature
        bgp.learn()

        # Check no attribute not found
        # info - bgp_id
        with self.assertRaises(AttributeError):
            bgp_id = (bgp.info['instance']['default']['bgp_id'])
        # table - bgp_table_version
        with self.assertRaises(AttributeError):
            bgp_table_version = (bgp.table['instance']['default']['vrf']\
                ['VRF1']['address_family']['ipv4 unicast']['bgp_table_version'])
        # routes_per_peer - remote_as
        with self.assertRaises(AttributeError):
            remote_as = (bgp.routes_per_peer['instance']['default']['vrf']\
                ['VRF1']['neighbor']['2.2.2.10']['remote_as'])
Esempio n. 2
0
    def test_custom_output(self):
        self.maxDiff = None
        bgp = Bgp(device=self.device)
        # Get outputs
        bgp.maker.outputs[ShowVrf] = {"{'vrf':'VRF1'}": BgpOutput.ShowVrf_vrf1}
        bgp.maker.outputs[ShowRoutingVrfAll] = {
            "{'vrf':'VRF1'}": BgpOutput.ShowRoutingVrfAll_vrf1
        }
        bgp.maker.outputs[ShowBgpProcessVrfAll] = {
            "{'vrf':'VRF1'}": BgpOutput.ShowBgpProcessVrfAll_vrf1
        }
        bgp.maker.outputs[ShowBgpPeerSession] = {
            '': BgpOutput.ShowBgpPeerSession
        }
        bgp.maker.outputs[ShowBgpPeerPolicy] = {
            '': BgpOutput.ShowBgpPeerPolicy
        }
        bgp.maker.outputs[ShowBgpPeerTemplate] = {
            '': BgpOutput.ShowBgpPeerTemplate
        }
        bgp.maker.outputs[ShowBgpVrfAllAll] = {
            "{'address_family':'ipv4 unicast','vrf':'VRF1'}":
            BgpOutput.ShowBgpVrfAllAll_vrf1
        }
        bgp.maker.outputs[ShowBgpVrfAllAllNextHopDatabase] = \
            {
                "{'address_family':'ipv4 unicast','vrf':'VRF1'}":
                    BgpOutput.ShowBgpVrfAllAllNextHopDatabase_vrf1}
        bgp.maker.outputs[ShowBgpVrfAllAllSummary] = \
            {"{'address_family':'ipv4 unicast','vrf':'VRF1'}": BgpOutput.ShowBgpVrfAllAllSummary_vrf1}
        bgp.maker.outputs[ShowBgpVrfAllAllDampeningParameters] = \
            {
                "{'address_family':'ipv4 unicast','vrf':'VRF1'}":
                    BgpOutput.ShowBgpVrfAllAllDampeningParameters_vrf1}
        # Return outputs above as inputs to parser when called
        self.device.execute = Mock()
        self.device.execute.side_effect = mapper

        # Learn the feature
        bgp.learn(vrf='VRF1',
                  address_family='ipv4 unicast',
                  neighbor='10.16.2.10')

        # Verify Ops was created successfully
        self.assertDictEqual(bgp.info, BgpOutput.BgpOpsOutput_info_custom)
        self.assertDictEqual(bgp.table, BgpOutput.BgpOpsOutput_table_custom)
        self.assertDictEqual(bgp.routes_per_peer,
                             BgpOutput.BgpOpsOutput_routesperpeer_custom)
Esempio n. 3
0
    def test_selective_attribute(self):
        self.maxDiff = None
        bgp = Bgp(device=self.device)
        # Get outputs
        bgp.maker.outputs[ShowVrf] = {"{'vrf':'all'}": BgpOutput.ShowVrf}
        bgp.maker.outputs[ShowRoutingVrfAll] = {
            "{'vrf':'all'}": BgpOutput.ShowRoutingVrfAll
        }
        bgp.maker.outputs[ShowBgpProcessVrfAll] = {
            "{'vrf':'all'}": BgpOutput.ShowBgpProcessVrfAll
        }
        bgp.maker.outputs[ShowBgpPeerSession] = {
            '': BgpOutput.ShowBgpPeerSession
        }
        bgp.maker.outputs[ShowBgpPeerPolicy] = {
            '': BgpOutput.ShowBgpPeerPolicy
        }
        bgp.maker.outputs[ShowBgpPeerTemplate] = {
            '': BgpOutput.ShowBgpPeerTemplate
        }
        bgp.maker.outputs[ShowBgpVrfAllAll] = {
            "{'address_family':'all','vrf':'all'}": BgpOutput.ShowBgpVrfAllAll
        }
        bgp.maker.outputs[ShowBgpVrfAllAllNextHopDatabase] = \
            {"{'address_family':'all','vrf':'all'}":
                    BgpOutput.ShowBgpVrfAllAllNextHopDatabase}
        bgp.maker.outputs[ShowBgpVrfAllAllSummary] = \
            {"{'address_family':'all','vrf':'all'}": BgpOutput.ShowBgpVrfAllAllSummary}
        bgp.maker.outputs[ShowBgpVrfAllAllDampeningParameters] = \
            {"{'address_family':'all','vrf':'all'}": BgpOutput.ShowBgpVrfAllAllDampeningParameters}
        # Return outputs above as inputs to parser when called
        self.device.execute = Mock()
        self.device.execute.side_effect = mapper

        # Learn the feature
        bgp.learn()

        # Check specific attribute values
        # info - bgp_id
        self.assertEqual(bgp.info['instance']['default']['bgp_id'], 100)
        # table - bgp_table_version
        self.assertEqual(bgp.table['instance']['default']['vrf']['VRF1']\
                ['address_family']['ipv4 unicast']['bgp_table_version'], 35)
        # routes_per_peer - remote_as
        self.assertEqual(bgp.routes_per_peer['instance']['default']['vrf']\
                ['VRF1']['neighbor']['10.16.2.10']['remote_as'], 0)
Esempio n. 4
0
    def test_incomplete_output(self):
        self.maxDiff = None
        bgp = Bgp(device=self.device)
        # Get outputs
        bgp.maker.outputs[ShowVrf] = {'': BgpOutput.ShowVrf}
        bgp.maker.outputs[ShowRoutingVrfAll] = {
            '': BgpOutput.ShowRoutingVrfAll
        }
        bgp.maker.outputs[ShowBgpProcessVrfAll] = {
            '': BgpOutput.ShowBgpProcessVrfAll
        }
        bgp.maker.outputs[ShowBgpPeerSession] = {
            '': BgpOutput.ShowBgpPeerSession
        }
        bgp.maker.outputs[ShowBgpPeerPolicy] = {
            '': BgpOutput.ShowBgpPeerPolicy
        }
        bgp.maker.outputs[ShowBgpPeerTemplate] = {
            '': BgpOutput.ShowBgpPeerTemplate
        }
        bgp.maker.outputs[ShowBgpVrfAllAll] = {'': BgpOutput.ShowBgpVrfAllAll}
        bgp.maker.outputs[ShowBgpVrfAllAllNextHopDatabase] =\
            {'': BgpOutput.ShowBgpVrfAllAllNextHopDatabase}
        bgp.maker.outputs[ShowBgpVrfAllAllSummary]=\
            {'':BgpOutput.ShowBgpVrfAllAllSummary}
        bgp.maker.outputs[ShowBgpVrfAllAllDampeningParameters] =\
            {'':BgpOutput.ShowBgpVrfAllAllDampeningParameters}
        # Outputs from side_effect set to empty
        bgp.maker.outputs[ShowBgpVrfAllNeighbors] = {'': ''}
        bgp.maker.outputs[ShowBgpVrfAllNeighborsAdvertisedRoutes] = {'': ''}
        bgp.maker.outputs[ShowBgpVrfAllNeighborsRoutes] = {'': ''}
        bgp.maker.outputs[ShowBgpVrfAllNeighborsReceivedRoutes] = {'': ''}

        # Return outputs above as inputs to parser when called
        self.device.execute = Mock()
        self.device.execute.side_effect = ['', '', '', '', '', '', '', '', '',\
                                           '', '', '', '', '', '', '', '', '',\
                                           '', '', '', '', '', '', '', '', '',\
                                           '', '', '', '', '', '', '', '', '',\
                                           '', '', '', '']

        # Learn the feature
        bgp.learn()
Esempio n. 5
0
    def test_complete_output(self):
        self.maxDiff = None
        bgp = Bgp(device=self.device)
        # Get outputs
        bgp.maker.outputs[ShowVrf] = {"{'vrf':'all'}": BgpOutput.ShowVrf}
        bgp.maker.outputs[ShowRoutingVrfAll] = {
            "{'vrf':'all'}": BgpOutput.ShowRoutingVrfAll
        }
        bgp.maker.outputs[ShowBgpProcessVrfAll] = {
            "{'vrf':'all'}": BgpOutput.ShowBgpProcessVrfAll
        }
        bgp.maker.outputs[ShowBgpPeerSession] = {
            '': BgpOutput.ShowBgpPeerSession
        }
        bgp.maker.outputs[ShowBgpPeerPolicy] = {
            '': BgpOutput.ShowBgpPeerPolicy
        }
        bgp.maker.outputs[ShowBgpPeerTemplate] = {
            '': BgpOutput.ShowBgpPeerTemplate
        }
        bgp.maker.outputs[ShowBgpVrfAllAll] = {
            "{'address_family':'all','vrf':'all'}": BgpOutput.ShowBgpVrfAllAll
        }
        bgp.maker.outputs[ShowBgpVrfAllAllNextHopDatabase] =\
            {"{'address_family':'all','vrf':'all'}": BgpOutput.ShowBgpVrfAllAllNextHopDatabase}
        bgp.maker.outputs[ShowBgpVrfAllAllSummary]=\
            {"{'address_family':'all','vrf':'all'}":BgpOutput.ShowBgpVrfAllAllSummary}
        bgp.maker.outputs[ShowBgpVrfAllAllDampeningParameters] =\
            {"{'address_family':'all','vrf':'all'}":BgpOutput.ShowBgpVrfAllAllDampeningParameters}
        # Return outputs above as inputs to parser when called
        self.device.execute = Mock()
        self.device.execute.side_effect = mapper

        # Learn the feature
        bgp.learn()

        # Verify Ops was created successfully
        self.assertDictEqual(bgp.info, BgpOutput.BgpOpsOutput_info)
        self.assertDictEqual(bgp.table, BgpOutput.BgpOpsOutput_table)
        self.assertDictEqual(bgp.routes_per_peer,
                             BgpOutput.BgpOpsOutput_routesperpeer)