tests/topotests/bgp_local_as_private_remove/test_bgp_local_as_private_remove.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_local_as_private_remove.py:
  27 Test if primary AS number is not removed in cases when `local-as`
  28 used together with `remove-private-AS`.
  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 lib.micronet_compat import Topo
  45
  46 pytestmark = [pytest.mark.bgpd]
  47
  48
  49 class TemplateTopo(Topo):
  50     def build(self, *_args, **_opts):
  51         tgen = get_topogen(self)
  52
  53         for routern in range(1, 5):
  54             tgen.add_router("r{}".format(routern))
  55
  56         switch = tgen.add_switch("s1")
  57         switch.add_link(tgen.gears["r1"])
  58         switch.add_link(tgen.gears["r2"])
  59
  60         switch = tgen.add_switch("s2")
  61         switch.add_link(tgen.gears["r3"])
  62         switch.add_link(tgen.gears["r4"])
  63
  64
  65 def setup_module(mod):
  66     tgen = Topogen(TemplateTopo, mod.__name__)
  67     tgen.start_topology()
  68
  69     router_list = tgen.routers()
  70
  71     for i, (rname, router) in enumerate(router_list.items(), 1):
  72         router.load_config(
  73             TopoRouter.RD_ZEBRA, os.path.join(CWD, "{}/zebra.conf".format(rname))
  74         )
  75         router.load_config(
  76             TopoRouter.RD_BGP, os.path.join(CWD, "{}/bgpd.conf".format(rname))
  77         )
  78
  79     tgen.start_router()
  80
  81
  82 def teardown_module(mod):
  83     tgen = get_topogen()
  84     tgen.stop_topology()
  85
  86
  87 def test_bgp_remove_private_as():
  88     tgen = get_topogen()
  89
  90     if tgen.routers_have_failure():
  91         pytest.skip(tgen.errors)
  92
  93     def _bgp_converge(router):
  94         while True:
  95             output = json.loads(
  96                 tgen.gears[router].vtysh_cmd("show ip bgp neighbor 192.168.255.1 json")
  97             )
  98             if output["192.168.255.1"]["bgpState"] == "Established":
  99                 time.sleep(1)
 100                 return True
 101
 102     def _bgp_as_path(router):
 103         output = json.loads(
 104             tgen.gears[router].vtysh_cmd("show ip bgp 172.16.255.254/32 json")
 105         )
 106         if output["prefix"] == "172.16.255.254/32":
 107             return output["paths"][0]["aspath"]["segments"][0]["list"]
 108
 109     if _bgp_converge("r2"):
 110         assert len(_bgp_as_path("r2")) == 1
 111         assert 65000 not in _bgp_as_path("r2")
 112
 113     if _bgp_converge("r4"):
 114         assert len(_bgp_as_path("r4")) == 2
 115         assert 3000 in _bgp_as_path("r4")
 116
 117
 118 if __name__ == "__main__":
 119     args = ["-s"] + sys.argv[1:]
 120     sys.exit(pytest.main(args))