From 55c9b65c89a753ec87df74863e414c5f927c154d Mon Sep 17 00:00:00 2001 From: Carmine Scarpitta Date: Fri, 6 Sep 2024 23:24:28 +0200 Subject: [PATCH] tests: SRv6 SID Manager check rib Signed-off-by: Carmine Scarpitta --- .../srv6_sid_manager/test_srv6_sid_manager.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/tests/topotests/srv6_sid_manager/test_srv6_sid_manager.py b/tests/topotests/srv6_sid_manager/test_srv6_sid_manager.py index 0b2091182d..2bb6ec8162 100644 --- a/tests/topotests/srv6_sid_manager/test_srv6_sid_manager.py +++ b/tests/topotests/srv6_sid_manager/test_srv6_sid_manager.py @@ -288,6 +288,22 @@ def open_json_file(filename): assert False, "Could not read file {}".format(filename) +def check_rib(name, cmd, expected_file): + def _check(name, cmd, expected_file): + logger.info("polling") + tgen = get_topogen() + router = tgen.gears[name] + output = json.loads(router.vtysh_cmd(cmd)) + expected = open_json_file("{}/{}".format(CWD, expected_file)) + return topotest.json_cmp(output, expected) + + logger.info('[+] check {} "{}" {}'.format(name, cmd, expected_file)) + tgen = get_topogen() + func = functools.partial(_check, name, cmd, expected_file) + success, result = topotest.run_and_expect(func, None, count=10, wait=0.5) + assert result is None, "Failed" + + if __name__ == "__main__": args = ["-s"] + sys.argv[1:] sys.exit(pytest.main(args))