diff options
Diffstat (limited to 'resources/libraries/python/autogen')
| -rw-r--r-- | resources/libraries/python/autogen/Regenerator.py | 71 | ||||
| -rw-r--r-- | resources/libraries/python/autogen/Testcase.py | 27 |
2 files changed, 79 insertions, 19 deletions
diff --git a/resources/libraries/python/autogen/Regenerator.py b/resources/libraries/python/autogen/Regenerator.py index d47680ccd0..cb0d3329d5 100644 --- a/resources/libraries/python/autogen/Regenerator.py +++ b/resources/libraries/python/autogen/Regenerator.py @@ -113,7 +113,7 @@ def check_suite_tag(suite_tag, prolog): """ found = prolog.count(u"| " + suite_tag) if found != 1: - raise ValueError(f"Suite tag found {found} times for {suite_id}") + raise ValueError(f"Suite tag found {found} times for {suite_tag}") def add_default_testcases(testcase, iface, suite_id, file_out, tc_kwargs_list): @@ -440,22 +440,56 @@ class Regenerator: min_frame_size = PROTOCOL_TO_MIN_FRAME_SIZE[protocol] default_kwargs_list = [ - {u"frame_size": min_frame_size, u"phy_cores": 1}, - {u"frame_size": min_frame_size, u"phy_cores": 2}, - {u"frame_size": min_frame_size, u"phy_cores": 4}, - {u"frame_size": 1518, u"phy_cores": 1}, - {u"frame_size": 1518, u"phy_cores": 2}, - {u"frame_size": 1518, u"phy_cores": 4}, - {u"frame_size": 9000, u"phy_cores": 1}, - {u"frame_size": 9000, u"phy_cores": 2}, - {u"frame_size": 9000, u"phy_cores": 4}, - {u"frame_size": u"IMIX_v4_1", u"phy_cores": 1}, - {u"frame_size": u"IMIX_v4_1", u"phy_cores": 2}, - {u"frame_size": u"IMIX_v4_1", u"phy_cores": 4} + {u"frame_size": min_frame_size, u"phy_cores": 1, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"}, + {u"frame_size": min_frame_size, u"phy_cores": 2, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"}, + {u"frame_size": min_frame_size, u"phy_cores": 4, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"}, + {u"frame_size": 1518, u"phy_cores": 1, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"}, + {u"frame_size": 1518, u"phy_cores": 2, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"}, + {u"frame_size": 1518, u"phy_cores": 4, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"}, + {u"frame_size": 9000, u"phy_cores": 1, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"}, + {u"frame_size": 9000, u"phy_cores": 2, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"}, + {u"frame_size": 9000, u"phy_cores": 4, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"}, + {u"frame_size": u"IMIX_v4_1", u"phy_cores": 1, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"}, + {u"frame_size": u"IMIX_v4_1", u"phy_cores": 2, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"}, + {u"frame_size": u"IMIX_v4_1", u"phy_cores": 4, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"} ] - tcp_kwargs_list = [ - {u"phy_cores": i, u"frame_size": 0} for i in (1, 2, 4) + hoststack_wrk_kwargs_list = [ + {u"phy_cores": i, u"frame_size": 0, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"} for i in (1, 2, 4) ] + hoststack_iperf3_kwargs_list = [ + {u"phy_cores": 1, u"frame_size": 0, u"clients": 1, + u"streams": 1, u"bytes_str": u"1G"}, + {u"phy_cores": 1, u"frame_size": 0, u"clients": 1, + u"streams": 10, u"bytes_str": u"10G"}, + {u"phy_cores": 2, u"frame_size": 0, u"clients": 1, + u"streams": 10, u"bytes_str": u"10G"}, + {u"phy_cores": 4, u"frame_size": 0, u"clients": 1, + u"streams": 10, u"bytes_str": u"10G"}, + ] + hoststack_quic_kwargs_list = [ + {u"phy_cores": 1, u"frame_size": 0, u"clients": 1, + u"streams": 1, u"bytes_str": u"100M"}, + {u"phy_cores": 1, u"frame_size": 0, u"clients": 1, + u"streams": 10, u"bytes_str": u"100M"}, + {u"phy_cores": 1, u"frame_size": 0, u"clients": 10, + u"streams": 1, u"bytes_str": u"100M"}, + {u"phy_cores": 1, u"frame_size": 0, u"clients": 10, + u"streams": 10, u"bytes_str": u"100M"}, + ] + for in_filename in glob(pattern): if not self.quiet: print( @@ -480,7 +514,12 @@ class Regenerator: elif in_filename.endswith(u"-reconf.robot"): write_reconf_files(in_filename, in_prolog, default_kwargs_list) elif in_filename[-10:] in (u"-cps.robot", u"-rps.robot"): - write_tcp_files(in_filename, in_prolog, tcp_kwargs_list) + write_tcp_files(in_filename, in_prolog, + hoststack_wrk_kwargs_list) + elif in_filename[-10:] in (u"-bps.robot"): + write_tcp_files(in_filename, in_prolog, + hoststack_iperf3_kwargs_list if u"iperf3" + in in_filename else hoststack_quic_kwargs_list) else: raise RuntimeError( f"Error in {in_filename}: non-primary suite type found." diff --git a/resources/libraries/python/autogen/Testcase.py b/resources/libraries/python/autogen/Testcase.py index 224295e1e2..6b4cfb2e06 100644 --- a/resources/libraries/python/autogen/Testcase.py +++ b/resources/libraries/python/autogen/Testcase.py @@ -34,7 +34,8 @@ class Testcase: """ self.template = Template(template_string) - def generate(self, num, frame_size, phy_cores): + def generate(self, num, frame_size, phy_cores, clients, streams, + bytes_str): """Return string of test case code with placeholders filled. Fail if there are placeholders left unfilled. @@ -43,9 +44,16 @@ class Testcase: :param num: Test case number. Example value: 4. :param frame_size: Imix string or numeric frame size. Example: 74. :param phy_cores: Number of physical cores to use. Example: 2. + :param clients: Number of clients used by test program. Example: 4. + :param streams: Number of streams used by test program. Example: 10. + :param bytes_str: Size in bytes of stream sent by test program. + Example: 1G :type num: int :type frame_size: str or int :type phy_cores: int or str + :type clients: int + :type streams: int + :type bytes_str: str :returns: Filled template, usable as test case code. :rtype: str """ @@ -66,7 +74,12 @@ class Testcase: { u"cores_num": f"${{{cores_num:d}}}", u"cores_str": phy_cores, - u"tc_num": f"tc{num:02d}" + u"tc_num": f"tc{num:02d}", + u"clients_num": f"${{{clients:d}}}", + u"clients_str": str(clients), + u"streams_num": f"${{{streams:d}}}", + u"streams_str": str(streams), + u"bytes_str": bytes_str, } ) return self.template.substitute(subst_dict) @@ -103,9 +116,17 @@ class Testcase: """ # TODO: Choose a better frame size identifier for streamed protocols # (TCP, QUIC, SCTP, ...) where DUT (not TG) decides frame size. - template_string = f''' + if u"tcphttp" in suite_id: + template_string = f''' | ${{tc_num}}-IMIX-${{cores_str}}c-{suite_id} | | [Tags] | ${{cores_str}}C | | phy_cores=${{cores_num}} ''' + else: + template_string = f''' +| ${{tc_num}}-IMIX-${{cores_str}}c-{suite_id} +| | [Tags] | ${{cores_str}}C | ${{clients_str}}CLIENT | ${{streams_str}}STREAM +| | phy_cores=${{cores_num}} | clients=${{clients_num}}''' + template_string += f" | streams=${{streams_num}}" \ + f" | bytes=${{bytes_str}}\n" return cls(template_string) |