tests/topotests/bgp_extended_optional_parameters_length/test_bgp_extended_optional_parameters_length.py

   1 #!/usr/bin/env python
   2 # SPDX-License-Identifier: ISC
   3
   4 # Copyright (c) 2021 by
   5 # Donatas Abraitis <donatas.abraitis@gmail.com>
   6 #
   7
   8 """
   9 Test if Extended Optional Parameters Length encoding format works
  10 if forced with a knob.
  11 https://datatracker.ietf.org/doc/html/rfc9072
  12 """
  13
  14 import os
  15 import sys
  16 import json
  17 import pytest
  18 import functools
  19
  20 pytestmark = pytest.mark.bgpd
  21
  22 CWD = os.path.dirname(os.path.realpath(__file__))
  23 sys.path.append(os.path.join(CWD, "../"))
  24
  25 # pylint: disable=C0413
  26 from lib import topotest
  27 from lib.topogen import Topogen, TopoRouter, get_topogen
  28
  29 pytestmark = [pytest.mark.bgpd]
  30
  31
  32 def setup_module(mod):
  33     topodef = {"s1": ("r1", "r2")}
  34     tgen = Topogen(topodef, mod.__name__)
  35     tgen.start_topology()
  36
  37     router_list = tgen.routers()
  38
  39     for i, (rname, router) in enumerate(router_list.items(), 1):
  40         router.load_config(
  41             TopoRouter.RD_ZEBRA, os.path.join(CWD, "{}/zebra.conf".format(rname))
  42         )
  43         router.load_config(
  44             TopoRouter.RD_BGP, os.path.join(CWD, "{}/bgpd.conf".format(rname))
  45         )
  46
  47     tgen.start_router()
  48
  49
  50 def teardown_module(mod):
  51     tgen = get_topogen()
  52     tgen.stop_topology()
  53
  54
  55 def test_bgp_extended_optional_parameters_length():
  56     tgen = get_topogen()
  57
  58     if tgen.routers_have_failure():
  59         pytest.skip(tgen.errors)
  60
  61     router = tgen.gears["r1"]
  62
  63     def _bgp_converge(router):
  64         output = json.loads(router.vtysh_cmd("show bgp ipv4 unicast summary json"))
  65         expected = {
  66             "peers": {
  67                 "192.168.1.2": {
  68                     "pfxRcd": 2,
  69                     "pfxSnt": 2,
  70                     "state": "Established",
  71                     "peerState": "OK",
  72                 }
  73             }
  74         }
  75         return topotest.json_cmp(output, expected)
  76
  77     test_func = functools.partial(_bgp_converge, router)
  78     _, result = topotest.run_and_expect(test_func, None, count=60, wait=0.5)
  79     assert result is None, "Can't converge with extended-optional-parameters"
  80
  81     def _bgp_extended_optional_parameters_length(router):
  82         output = json.loads(router.vtysh_cmd("show bgp neighbor 192.168.1.2 json"))
  83         expected = {"192.168.1.2": {"extendedOptionalParametersLength": True}}
  84         return topotest.json_cmp(output, expected)
  85
  86     test_func = functools.partial(_bgp_extended_optional_parameters_length, router)
  87     _, result = topotest.run_and_expect(test_func, None, count=60, wait=0.5)
  88     assert result is None, "Can't see Extended Optional Parameters Length to be used"
  89
  90
  91 if __name__ == "__main__":
  92     args = ["-s"] + sys.argv[1:]
  93     sys.exit(pytest.main(args))