tests/topotests/bgp_maximum_prefix_invalid_update/test_bgp_maximum_prefix_invalid_update.py

   1 #!/usr/bin/env python
   2
   3 #
   4 # bgp_local_as_private_remove.py
   5 # Part of NetDEF Topology Tests
   6 #
   7 # Copyright (c) 2019 by
   8 # Network Device Education Foundation, Inc. ("NetDEF")
   9 #
  10 # Permission to use, copy, modify, and/or distribute this software
  11 # for any purpose with or without fee is hereby granted, provided
  12 # that the above copyright notice and this permission notice appear
  13 # in all copies.
  14 #
  15 # THE SOFTWARE IS PROVIDED "AS IS" AND NETDEF DISCLAIMS ALL WARRANTIES
  16 # WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  17 # MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NETDEF BE LIABLE FOR
  18 # ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY
  19 # DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
  20 # WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
  21 # ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
  22 # OF THIS SOFTWARE.
  23 #
  24
  25 """
  26 bgp_maximum_prefix_invalid_update.py:
  27 Test if unnecesarry UPDATE message like below:
  28
  29 [Error] Error parsing NLRI
  30 %NOTIFICATION: sent to neighbor X.X.X.X 3/10 (UPDATE Message Error/Invalid Network Field) 0 bytes
  31
  32 is not sent if maximum-prefix count is overflow.
  33 """
  34
  35 import os
  36 import sys
  37 import json
  38 import time
  39 import pytest
  40
  41 CWD = os.path.dirname(os.path.realpath(__file__))
  42 sys.path.append(os.path.join(CWD, '../'))
  43
  44 # pylint: disable=C0413
  45 from lib import topotest
  46 from lib.topogen import Topogen, TopoRouter, get_topogen
  47 from lib.topolog import logger
  48 from mininet.topo import Topo
  49
  50 class TemplateTopo(Topo):
  51     def build(self, *_args, **_opts):
  52         tgen = get_topogen(self)
  53
  54         for routern in range(1, 3):
  55             tgen.add_router('r{}'.format(routern))
  56
  57         switch = tgen.add_switch('s1')
  58         switch.add_link(tgen.gears['r1'])
  59         switch.add_link(tgen.gears['r2'])
  60
  61 def setup_module(mod):
  62     tgen = Topogen(TemplateTopo, mod.__name__)
  63     tgen.start_topology()
  64
  65     router_list = tgen.routers()
  66
  67     for i, (rname, router) in enumerate(router_list.iteritems(), 1):
  68         router.load_config(
  69             TopoRouter.RD_ZEBRA,
  70             os.path.join(CWD, '{}/zebra.conf'.format(rname))
  71         )
  72         router.load_config(
  73             TopoRouter.RD_BGP,
  74             os.path.join(CWD, '{}/bgpd.conf'.format(rname))
  75         )
  76
  77     tgen.start_router()
  78
  79 def teardown_module(mod):
  80     tgen = get_topogen()
  81     tgen.stop_topology()
  82
  83 def test_bgp_maximum_prefix_invalid():
  84     tgen = get_topogen()
  85
  86     if tgen.routers_have_failure():
  87         pytest.skip(tgen.errors)
  88
  89     def _bgp_converge(router):
  90         while True:
  91             output = json.loads(tgen.gears[router].vtysh_cmd("show ip bgp neighbor 192.168.255.1 json"))
  92             if output['192.168.255.1']['connectionsEstablished'] > 3:
  93                 return True
  94             time.sleep(1)
  95
  96     def _bgp_parsing_nlri(router):
  97         cmd_max_exceeded = 'grep "%MAXPFXEXCEED: No. of IPv4 Unicast prefix received" bgpd.log'
  98         cmdt_error_parsing_nlri = 'grep "Error parsing NLRI" bgpd.log'
  99         output_max_exceeded = tgen.gears[router].run(cmd_max_exceeded)
 100         output_error_parsing_nlri = tgen.gears[router].run(cmdt_error_parsing_nlri)
 101
 102         if len(output_max_exceeded) > 0:
 103             if len(output_error_parsing_nlri) > 0:
 104                 return False
 105         return True
 106
 107
 108     if _bgp_converge('r2'):
 109         assert _bgp_parsing_nlri('r2') == True
 110
 111 if __name__ == '__main__':
 112     args = ["-s"] + sys.argv[1:]
 113     sys.exit(pytest.main(args))