From 298d0a61d75cded31fc0f09d408e7b3e3f9ee4f7 Mon Sep 17 00:00:00 2001 From: Vratko Polak Date: Fri, 20 Jul 2018 13:13:06 +0200 Subject: CSIT-1097: Migrate Vts suite to NDRPDR Change-Id: Iaac913bb4b15d743a2e9d565e25d31c089cee3e8 Signed-off-by: Vratko Polak --- resources/libraries/python/autogen/DefaultTestcase.py | 2 +- resources/libraries/python/autogen/Regenerator.py | 12 +++++++++--- 2 files changed, 10 insertions(+), 4 deletions(-) (limited to 'resources/libraries/python') diff --git a/resources/libraries/python/autogen/DefaultTestcase.py b/resources/libraries/python/autogen/DefaultTestcase.py index 9878289bd1..3c11cb073e 100644 --- a/resources/libraries/python/autogen/DefaultTestcase.py +++ b/resources/libraries/python/autogen/DefaultTestcase.py @@ -13,7 +13,7 @@ """Module with utilities for autogeneration of non-customizable testcases.""" -from Testcase import Testcase +from .Testcase import Testcase class DefaultTestcase(Testcase): diff --git a/resources/libraries/python/autogen/Regenerator.py b/resources/libraries/python/autogen/Regenerator.py index 30dc7b7e89..4ed123ae7c 100644 --- a/resources/libraries/python/autogen/Regenerator.py +++ b/resources/libraries/python/autogen/Regenerator.py @@ -36,7 +36,7 @@ class Regenerator(object): """ self.testcase_class = testcase_class - def regenerate_glob(self, pattern, is_ip6=False, tc_kwargs_list=None): + def regenerate_glob(self, pattern, protocol="ip4", tc_kwargs_list=None): """Regenerate files matching glob pattern based on arguments. In the current working directory, find all files matching @@ -56,6 +56,12 @@ class Regenerator(object): :type tc_kwargs_list: list of tuple or None """ + protocol_to_min_framesize = { + "ip4": 64, + "ip6": 78, + "vxlan+ip4": 114 + } + def get_suite_id(filename): dash_split = filename.split("-", 1) if len(dash_split[0]) <= 4: @@ -73,8 +79,8 @@ class Regenerator(object): num = add_testcase(file_out, num, **tc_kwargs) print "Regenerator starts at {cwd}".format(cwd=getcwd()) - min_framesize = 78 if is_ip6 else 64 - kwargs_list = tc_kwargs_list if tc_kwargs_list is not None else [ + min_framesize = protocol_to_min_framesize[protocol] + kwargs_list = tc_kwargs_list if tc_kwargs_list else [ {"framesize": min_framesize, "phy_cores": 1}, {"framesize": min_framesize, "phy_cores": 2}, {"framesize": min_framesize, "phy_cores": 4}, -- cgit 1.2.3-korg