aboutsummaryrefslogtreecommitdiffstats
path: root/resources/libraries/python/autogen/Testcase.py
blob: 32fc5014cc83d1d1e2172d5ec381e3090a92d678 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
# Copyright (c) 2021 Cisco and/or its affiliates.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Module defining utilities for testcase autogeneration."""

from string import Template


class Testcase:
    """Class containing a template string and a substitution method."""

    def __init__(self, template_string):
        """Construct instance by storing template given by string.

        :param template_string: Template string to generate test case code with.
            See string.Template documentation for template string syntax.
            Only the following placeholders are supported:
            - cores_num - Number of cores as robot number, example: "${2}".
            - cores_str - Number of physical cores to use, example: "2".
            - frame_num - Framesize as a number, example: "${74}".
            - frame_str - Framesize in upper case, example: "74B".
        :type template_string: str
        """
        self.template = Template(template_string)

    def generate(self, frame_size, phy_cores=None):
        """Return string of test case code with placeholders filled.

        Fail if there are placeholders left unfilled.
        It is not required for all placeholders to be present in template.

        :param frame_size: Imix string or numeric frame size. Example: 74.
        :param phy_cores: Number of physical cores to use. Example: 2. It can
            be None in n2n testcases.
        :type frame_size: str or int
        :type phy_cores: int, str or None
        :returns: Filled template, usable as test case code.
        :rtype: str
        """
        try:
            fsize = int(frame_size)
            subst_dict = {
                u"frame_num": f"${{{fsize:d}}}",
                u"frame_str": f"{fsize:d}B"
            }
        except ValueError:  # Assuming an IMIX string.
            subst_dict = {
                u"frame_num": str(frame_size),
                u"frame_str": u"IMIX"
            }
        if phy_cores is None:
            return self.template.substitute(subst_dict)
        cores_str = str(phy_cores)
        cores_num = int(cores_str)
        subst_dict.update(
            {
                u"cores_num": f"${{{cores_num:d}}}",
                u"cores_str": phy_cores,
            }
        )
        return self.template.substitute(subst_dict)

    @classmethod
    def default(cls, suite_id):
        """Factory method for creating "default" testcase objects.

        Testcase name will contain both frame size and core count.
        Used for most performance tests, except TCP ones.

        :param suite_id: Part of suite name to distinguish from other suites.
        :type suite_id: str
        :returns: Instance for generating testcase text of this type.
        :rtype: Testcase
        """
        template_string = f'''
| ${{frame_str}}-${{cores_str}}c-{suite_id}
| | [Tags] | ${{frame_str}} | ${{cores_str}}C
| | frame_size=${{frame_num}} | phy_cores=${{cores_num}}
'''
        return cls(template_string)

    @classmethod
    def tcp(cls, suite_id):
        """Factory method for creating "tcp" testcase objects.

        Testcase name will contain core count, but not frame size.

        :param suite_id: Part of suite name to distinguish from other suites.
        :type suite_id: str
        :returns: Instance for generating testcase text of this type.
        :rtype: Testcase
        """
        # TODO: Choose a better frame size identifier for streamed protocols
        # (TCP, QUIC, SCTP, ...) where DUT (not TG) decides frame size.
        if u"tcphttp" in suite_id:
            if u"rps" or u"cps" in suite_id:
                template_string = f'''
| ${{frame_str}}-${{cores_str}}c-{suite_id}
| | [Tags] | ${{frame_str}} | ${{cores_str}}C
| | frame_size=${{frame_num}} | phy_cores=${{cores_num}}
'''
            else:
                template_string = f'''
| IMIX-${{cores_str}}c-{suite_id}
| | [Tags] | ${{cores_str}}C
| | phy_cores=${{cores_num}}
'''
        else:
            template_string = f'''
| ${{frame_str}}-${{cores_str}}c-{suite_id[:-4]}-{suite_id[-3:]}
| | [Tags] | ${{cores_str}}C\n| | phy_cores=${{cores_num}}
'''
        return cls(template_string)

    @classmethod
    def iperf3(cls, suite_id):
        """Factory method for creating "iperf3" testcase objects.

        Testcase name will contain core count, but not frame size.

        :param suite_id: Part of suite name to distinguish from other suites.
        :type suite_id: str
        :returns: Instance for generating testcase text of this type.
        :rtype: Testcase
        """
        template_string = f'''
| 128KB-${{cores_str}}c-{suite_id}
| | [Tags] | 128KB | ${{cores_str}}C
| | frame_size=${{frame_num}} | phy_cores=${{cores_num}}
'''
        return cls(template_string)

    @classmethod
    def trex(cls, suite_id):
        """Factory method for creating "trex" testcase objects.

        Testcase name will contain frame size, but not core count.

        :param suite_id: Part of suite name to distinguish from other suites.
        :type suite_id: str
        :returns: Instance for generating testcase text of this type.
        :rtype: Testcase
        """
        template_string = f'''
| ${{frame_str}}--{suite_id}
| | [Tags] | ${{frame_str}}
| | frame_size=${{frame_num}}
'''
        return cls(template_string)