tests/topotests/zebra_seg6_route/test_zebra_seg6_route.py

   1 #!/usr/bin/env python
   2
   3 #
   4 # test_zebra_seg6_route.py
   5 #
   6 # Copyright (c) 2020 by
   7 # LINE Corporation, Hiroki Shirokura <slank.dev@gmail.com>
   8 #
   9 # Permission to use, copy, modify, and/or distribute this software
  10 # for any purpose with or without fee is hereby granted, provided
  11 # that the above copyright notice and this permission notice appear
  12 # in all copies.
  13 #
  14 # THE SOFTWARE IS PROVIDED "AS IS" AND NETDEF DISCLAIMS ALL WARRANTIES
  15 # WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  16 # MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NETDEF BE LIABLE FOR
  17 # ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY
  18 # DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
  19 # WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
  20 # ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
  21 # OF THIS SOFTWARE.
  22 #
  23
  24 """
  25 test_zebra_seg6_route.py: Test seg6 route addition with zapi.
  26 """
  27
  28 import os
  29 import sys
  30 import pytest
  31 import json
  32 from functools import partial
  33
  34 CWD = os.path.dirname(os.path.realpath(__file__))
  35 sys.path.append(os.path.join(CWD, "../"))
  36
  37 # pylint: disable=C0413
  38 from lib import topotest
  39 from lib.topogen import Topogen, TopoRouter, get_topogen
  40 from lib.topolog import logger
  41
  42 pytestmark = [pytest.mark.sharpd]
  43
  44
  45 def open_json_file(filename):
  46     try:
  47         with open(filename, "r") as f:
  48             return json.load(f)
  49     except IOError:
  50         assert False, "Could not read file {}".format(filename)
  51
  52
  53 def setup_module(mod):
  54     tgen = Topogen({None: "r1"}, mod.__name__)
  55     tgen.start_topology()
  56     router_list = tgen.routers()
  57     for rname, router in tgen.routers().items():
  58         router.run(
  59             "/bin/bash {}".format(os.path.join(CWD, "{}/setup.sh".format(rname)))
  60         )
  61         router.load_config(
  62             TopoRouter.RD_ZEBRA, os.path.join(CWD, "{}/zebra.conf".format(rname))
  63         )
  64         router.load_config(
  65             TopoRouter.RD_SHARP, os.path.join(CWD, "{}/sharpd.conf".format(rname))
  66         )
  67     tgen.start_router()
  68
  69
  70 def teardown_module(_mod):
  71     tgen = get_topogen()
  72     tgen.stop_topology()
  73
  74
  75 def test_zebra_seg6local_routes():
  76     tgen = get_topogen()
  77     if tgen.routers_have_failure():
  78         pytest.skip(tgen.errors)
  79     logger.info("Test for seg6local route install via ZAPI was start.")
  80     r1 = tgen.gears["r1"]
  81
  82     def check(router, dest, nh, sid, expected):
  83         router.vtysh_cmd(
  84             "sharp install seg6-routes {} "
  85             "nexthop-seg6 {} encap {} 1".format(dest, nh, sid)
  86         )
  87         output = json.loads(router.vtysh_cmd("show ipv6 route {} json".format(dest)))
  88         output = output.get("{}/128".format(dest))
  89         if output is None:
  90             return False
  91         return topotest.json_cmp(output, expected)
  92
  93     manifests = open_json_file(os.path.join(CWD, "{}/routes.json".format("r1")))
  94     for manifest in manifests:
  95         logger.info(
  96             "CHECK {} {} {}".format(
  97                 manifest["in"]["dest"], manifest["in"]["nh"], manifest["in"]["sid"]
  98             )
  99         )
 100         test_func = partial(
 101             check,
 102             r1,
 103             manifest["in"]["dest"],
 104             manifest["in"]["nh"],
 105             manifest["in"]["sid"],
 106             manifest["out"],
 107         )
 108         success, result = topotest.run_and_expect(test_func, None, count=5, wait=1)
 109         assert result is None, "Failed"
 110
 111
 112 if __name__ == "__main__":
 113     args = ["-s"] + sys.argv[1:]
 114     sys.exit(pytest.main(args))