tests/topotests/bgp_labeled_unicast_default_originate/test_bgp_labeled_unicast_default_originate.py

   1 #!/usr/bin/env python
   2 # SPDX-License-Identifier: ISC
   3
   4 #
   5 # Copyright (c) 2022 by
   6 # Donatas Abraitis <donatas@opensourcerouting.org>
   7 #
   8
   9 """
  10 Check if labeled-unicast works correctly with default-originate.
  11 """
  12
  13 import os
  14 import sys
  15 import json
  16 import pytest
  17 import functools
  18
  19 CWD = os.path.dirname(os.path.realpath(__file__))
  20 sys.path.append(os.path.join(CWD, "../"))
  21
  22 # pylint: disable=C0413
  23 from lib import topotest
  24 from lib.topogen import Topogen, TopoRouter, get_topogen
  25 from lib.common_config import step
  26
  27 pytestmark = [pytest.mark.bgpd]
  28
  29
  30 def build_topo(tgen):
  31     for routern in range(1, 3):
  32         tgen.add_router("r{}".format(routern))
  33
  34     switch = tgen.add_switch("s1")
  35     switch.add_link(tgen.gears["r1"])
  36     switch.add_link(tgen.gears["r2"])
  37
  38
  39 def setup_module(mod):
  40     tgen = Topogen(build_topo, mod.__name__)
  41     tgen.start_topology()
  42
  43     router_list = tgen.routers()
  44
  45     for i, (rname, router) in enumerate(router_list.items(), 1):
  46         router.load_config(
  47             TopoRouter.RD_ZEBRA, os.path.join(CWD, "{}/zebra.conf".format(rname))
  48         )
  49         router.load_config(
  50             TopoRouter.RD_BGP, os.path.join(CWD, "{}/bgpd.conf".format(rname))
  51         )
  52
  53     tgen.start_router()
  54
  55
  56 def teardown_module(mod):
  57     tgen = get_topogen()
  58     tgen.stop_topology()
  59
  60
  61 def test_bgp_labeled_unicast_default_originate():
  62     tgen = get_topogen()
  63
  64     if tgen.routers_have_failure():
  65         pytest.skip(tgen.errors)
  66
  67     r1 = tgen.gears["r1"]
  68     r2 = tgen.gears["r2"]
  69
  70     def _bgp_check_advertised_routes():
  71         output = json.loads(
  72             r1.vtysh_cmd(
  73                 "show bgp ipv4 labeled-unicast neighbors 192.168.12.2 advertised-routes json"
  74             )
  75         )
  76         expected = {
  77             "bgpOriginatingDefaultNetwork": "0.0.0.0/0",
  78         }
  79         return topotest.json_cmp(output, expected)
  80
  81     test_func = functools.partial(_bgp_check_advertised_routes)
  82     _, result = topotest.run_and_expect(test_func, None, count=60, wait=0.5)
  83     assert result is None, "Failed to advertise default route for labeled-unicast"
  84
  85     def _bgp_check_received_routes():
  86         output = json.loads(
  87             r2.vtysh_cmd("show bgp ipv4 labeled-unicast 0.0.0.0/0 json")
  88         )
  89         expected = {
  90             "paths": [
  91                 {
  92                     "valid": True,
  93                     "metric": 666,
  94                     "community": {
  95                         "string": "65001:65001",
  96                     },
  97                 }
  98             ]
  99         }
 100         return topotest.json_cmp(output, expected)
 101
 102     test_func = functools.partial(_bgp_check_received_routes)
 103     _, result = topotest.run_and_expect(test_func, None, count=60, wait=0.5)
 104     assert result is None, "Failed to receive default route for labeled-unicast"
 105
 106
 107 if __name__ == "__main__":
 108     args = ["-s"] + sys.argv[1:]
 109     sys.exit(pytest.main(args))