tests/topotests/bgp_show_ip_bgp_fqdn/test_bgp_show_ip_bgp_fqdn.py

   1 #!/usr/bin/env python
   2
   3 #
   4 # test_bgp_show_ip_bgp_fqdn.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 test_bgp_show_ip_bgp_fqdn.py:
  27 Test if FQND is visible in `show [ip] bgp` output if
  28 `bgp default show-hostname` is toggled.
  29 """
  30
  31 import os
  32 import sys
  33 import json
  34 import time
  35 import pytest
  36
  37 CWD = os.path.dirname(os.path.realpath(__file__))
  38 sys.path.append(os.path.join(CWD, '../'))
  39
  40 # pylint: disable=C0413
  41 from lib import topotest
  42 from lib.topogen import Topogen, TopoRouter, get_topogen
  43 from lib.topolog import logger
  44 from mininet.topo import Topo
  45
  46 class TemplateTopo(Topo):
  47     def build(self, *_args, **_opts):
  48         tgen = get_topogen(self)
  49
  50         for routern in range(1, 3):
  51             tgen.add_router('r{}'.format(routern))
  52
  53         switch = tgen.add_switch('s1')
  54         switch.add_link(tgen.gears['r1'])
  55         switch.add_link(tgen.gears['r2'])
  56
  57 def setup_module(mod):
  58     tgen = Topogen(TemplateTopo, mod.__name__)
  59     tgen.start_topology()
  60
  61     router_list = tgen.routers()
  62
  63     for i, (rname, router) in enumerate(router_list.iteritems(), 1):
  64         router.load_config(
  65             TopoRouter.RD_ZEBRA,
  66             os.path.join(CWD, '{}/zebra.conf'.format(rname))
  67         )
  68         router.load_config(
  69             TopoRouter.RD_BGP,
  70             os.path.join(CWD, '{}/bgpd.conf'.format(rname))
  71         )
  72
  73     tgen.start_router()
  74
  75 def teardown_module(mod):
  76     tgen = get_topogen()
  77     tgen.stop_topology()
  78
  79 def test_bgp_maximum_prefix_invalid():
  80     tgen = get_topogen()
  81
  82     if tgen.routers_have_failure():
  83         pytest.skip(tgen.errors)
  84
  85     def _bgp_converge(router, neighbor):
  86         cmd = "show ip bgp neighbor {0} json".format(neighbor)
  87         while True:
  88             output = json.loads(tgen.gears[router].vtysh_cmd(cmd))
  89             if output[neighbor]['bgpState'] == 'Established':
  90                 time.sleep(3)
  91                 return True
  92
  93     def _bgp_show_nexthop(router, prefix):
  94         cmd = "show ip bgp json"
  95         output = json.loads(tgen.gears[router].vtysh_cmd(cmd))
  96         for nh in output['routes'][prefix][0]['nexthops']:
  97             if 'fqdn' in nh:
  98                 return 'fqdn'
  99         return 'ip'
 100
 101     if _bgp_converge('r2', '192.168.255.1'):
 102         assert _bgp_show_nexthop('r2', '172.16.255.254/32') == 'fqdn'
 103
 104     if _bgp_converge('r1', '192.168.255.2'):
 105         assert _bgp_show_nexthop('r1', '172.16.255.253/32') == 'ip'
 106
 107 if __name__ == '__main__':
 108     args = ["-s"] + sys.argv[1:]
 109     sys.exit(pytest.main(args))