tests/topotests/bgp_community_alias/test_bgp-community-alias.py

   1 #!/usr/bin/env python
   2
   3 # Copyright (c) 2021 by
   4 # Donatas Abraitis <donatas.abraitis@gmail.com>
   5 #
   6 # Permission to use, copy, modify, and/or distribute this software
   7 # for any purpose with or without fee is hereby granted, provided
   8 # that the above copyright notice and this permission notice appear
   9 # in all copies.
  10 #
  11 # THE SOFTWARE IS PROVIDED "AS IS" AND NETDEF DISCLAIMS ALL WARRANTIES
  12 # WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  13 # MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NETDEF BE LIABLE FOR
  14 # ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY
  15 # DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
  16 # WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
  17 # ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
  18 # OF THIS SOFTWARE.
  19 #
  20
  21 """
  22 Test if BGP community alias is visible in CLI outputs
  23 """
  24
  25 import os
  26 import sys
  27 import json
  28 import pytest
  29 import functools
  30
  31 pytestmark = pytest.mark.bgpd
  32
  33 CWD = os.path.dirname(os.path.realpath(__file__))
  34 sys.path.append(os.path.join(CWD, "../"))
  35
  36 # pylint: disable=C0413
  37 from lib import topotest
  38 from lib.topogen import Topogen, TopoRouter, get_topogen
  39
  40 pytestmark = [pytest.mark.bgpd]
  41
  42
  43 def build_topo(tgen):
  44     for routern in range(1, 3):
  45         tgen.add_router("r{}".format(routern))
  46
  47     switch = tgen.add_switch("s1")
  48     switch.add_link(tgen.gears["r1"])
  49     switch.add_link(tgen.gears["r2"])
  50
  51
  52 def setup_module(mod):
  53     tgen = Topogen(build_topo, mod.__name__)
  54     tgen.start_topology()
  55
  56     router_list = tgen.routers()
  57
  58     for i, (rname, router) in enumerate(router_list.items(), 1):
  59         router.load_config(
  60             TopoRouter.RD_ZEBRA, os.path.join(CWD, "{}/zebra.conf".format(rname))
  61         )
  62         router.load_config(
  63             TopoRouter.RD_BGP, os.path.join(CWD, "{}/bgpd.conf".format(rname))
  64         )
  65
  66     tgen.start_router()
  67
  68
  69 def teardown_module(mod):
  70     tgen = get_topogen()
  71     tgen.stop_topology()
  72
  73
  74 def test_bgp_community_alias():
  75     tgen = get_topogen()
  76
  77     if tgen.routers_have_failure():
  78         pytest.skip(tgen.errors)
  79
  80     router = tgen.gears["r1"]
  81
  82     def _bgp_converge(router):
  83         output = json.loads(router.vtysh_cmd("show ip route json"))
  84         expected = {
  85             "172.16.16.1/32": [
  86                 {
  87                     "tag": 10,
  88                     "communities": "community-r2-1 65001:2",
  89                     "largeCommunities": "large-community-r2-1 65001:1:2",
  90                 }
  91             ],
  92             "172.16.16.2/32": [
  93                 {
  94                     "tag": 20,
  95                     "communities": "65002:1 community-r2-2",
  96                     "largeCommunities": "",
  97                 }
  98             ],
  99             "172.16.16.3/32": [
 100                 {
 101                     "tag": 100,
 102                     "communities": "",
 103                     "largeCommunities": "",
 104                 }
 105             ],
 106         }
 107         return topotest.json_cmp(output, expected)
 108
 109     test_func = functools.partial(_bgp_converge, router)
 110     success, result = topotest.run_and_expect(test_func, None, count=60, wait=0.5)
 111     assert result is None, "Cannot see BGP community aliases at r1"
 112
 113     def _bgp_show_prefixes_by_alias(router):
 114         output = json.loads(
 115             router.vtysh_cmd(
 116                 "show bgp ipv4 unicast alias large-community-r2-1 json detail"
 117             )
 118         )
 119         expected = {
 120             "routes": {
 121                 "172.16.16.1/32": {
 122                     "paths": [
 123                         {
 124                             "community": {"string": "community-r2-1 65001:2"},
 125                             "largeCommunity": {
 126                                 "string": "large-community-r2-1 65001:1:2"
 127                             },
 128                         }
 129                     ]
 130                 }
 131             }
 132         }
 133         return topotest.json_cmp(output, expected)
 134
 135     test_func = functools.partial(_bgp_show_prefixes_by_alias, router)
 136     success, result = topotest.run_and_expect(test_func, None, count=60, wait=0.5)
 137     assert result is None, "Cannot see BGP prefixes by community alias at r1"
 138
 139     def _bgp_show_prefixes_by_large_community_list(router):
 140         output = json.loads(
 141             router.vtysh_cmd("show bgp ipv4 unicast large-community-list r2 json")
 142         )
 143         expected = {"routes": {"172.16.16.1/32": [{"valid": True}]}}
 144         return topotest.json_cmp(output, expected)
 145
 146     test_func = functools.partial(_bgp_show_prefixes_by_large_community_list, router)
 147     success, result = topotest.run_and_expect(test_func, None, count=60, wait=0.5)
 148     assert result is None, "Cannot see BGP prefixes by large community list at r1"
 149
 150
 151 if __name__ == "__main__":
 152     args = ["-s"] + sys.argv[1:]
 153     sys.exit(pytest.main(args))