Minor change in virtual env creation
[csit.git] / resources / libraries / python / TrafficGenerator.py
index 24bec0a..9c66982 100644 (file)
@@ -11,7 +11,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Traffic generator library."""
+"""Performance testing traffic generator library."""
 
 from robot.api import logger
 
@@ -30,31 +30,84 @@ class TrafficGenerator(object):
         self._loss = None
         self._sent = None
         self._received = None
+        #T-REX interface order mapping
+        self._ifaces_reordered = 0
 
-    @staticmethod
-    def initialize_traffic_generator(node, interface1, interface2):
+    def initialize_traffic_generator(self, node, interface1, interface2, test_type):
         """TG initialization
         :param node: Traffic generator node
-        :param interface1: PCI address of first interface
-        :param interface2: PCI address of second interface
+        :param interface1: interface name of first interface
+        :param interface2: interface name of second interface
+        :test_type: 'L2' or 'L3' - src/dst MAC address
         :type node: dict
         :type interface1: str
         :type interface2: str
+        :type test_type: str
         :return: nothing
         """
 
+        trex_path = "/opt/trex-core-1.91"
+
+        topo = Topology()
+
         if node['type'] != NodeType.TG:
             raise Exception('Node type is not a TG')
         if node['subtype'] == NodeSubTypeTG.TREX:
             ssh = SSH()
             ssh.connect(node)
 
+            iface1_pci = topo.get_interface_pci_addr(node, interface1)
+            iface2_pci = topo.get_interface_pci_addr(node, interface2)
+            iface1_mac = topo.get_interface_mac(node, interface1)
+            iface2_mac = topo.get_interface_mac(node, interface2)
+            if min(iface1_pci, iface2_pci) != iface1_pci:
+                iface1_mac, iface2_mac = iface2_mac, iface1_mac
+                iface1_pci, iface2_pci = iface2_pci, iface1_pci
+                self._ifaces_reordered = 1
+
+            iface1_mac_hex = "0x"+iface1_mac.replace(":", ",0x")
+            iface2_mac_hex = "0x"+iface2_mac.replace(":", ",0x")
+
+            if test_type == 'L2':
+                (ret, stdout, stderr) = ssh.exec_command(
+                    "sudo sh -c 'cat << EOF > /etc/trex_cfg.yaml\n"
+                    "- port_limit      : 2\n"
+                    "  version         : 2\n"
+                    "  interfaces      : [\"{}\",\"{}\"]\n"
+                    "  port_info       :\n"
+                    "          - dest_mac        :   [{}]\n"
+                    "            src_mac         :   [{}]\n"
+                    "          - dest_mac        :   [{}]\n"
+                    "            src_mac         :   [{}]\n"
+                    "EOF'"\
+                    .format(iface1_pci, iface2_pci,
+                            iface2_mac_hex, iface1_mac_hex,
+                            iface1_mac_hex, iface2_mac_hex))
+                if int(ret) != 0:
+                    logger.error("failed to create t-rex config: {}"\
+                    .format(stdout + stderr))
+                    raise RuntimeError('trex config generation error')
+            elif test_type == 'L3':
+                raise NotImplementedError("L3 test_type not supported")
+            else:
+                raise Exception("test_type unknown")
+
+
             (ret, stdout, stderr) = ssh.exec_command(
-                "sh -c 'cd /opt/trex-core-1.88/scripts/ && "
-                "--bind=igb_uio {0} {1}".format(interface1, interface2))
-            (ret, stdout, stderr) = ssh.exec_command(
-                "sh -c 'cd /opt/trex-core-1.88/scripts/ && "
-                "sudo nohup ./t-rex-64 -i -c 4 --iom 0 > /dev/null 2>&1 &'")
+                "sh -c 'cd {0}/scripts/ && "
+                "sudo ./trex-cfg'"\
+                .format(trex_path))
+            if int(ret) != 0:
+                logger.error('trex-cfg failed: {0}'.format(stdout + stderr))
+                raise RuntimeError('trex-cfg failed')
+
+            (ret, _, _) = ssh.exec_command(
+                "sh -c 'cd {0}/scripts/ && "
+                "sudo nohup ./t-rex-64 -i -c 4 --iom 0 > /dev/null 2>&1 &'"
+                "> /dev/null"\
+                .format(trex_path))
+            if int(ret) != 0:
+                raise RuntimeError('t-rex-64 startup failed')
 
     @staticmethod
     def teardown_traffic_generator(node):
@@ -78,7 +131,7 @@ class TrafficGenerator(object):
         :param nodes_info: Dictionary containing information on all nodes
         in topology.
         :param duration: Duration of test traffic generation in seconds
-        :param rate: Percentage of linerate
+        :param rate: Offered load per interface (e.g. 1%, 3gbps, 4mpps, ...)
         :param framesize: Frame size (L2) in Bytes
         :param traffic_type: Traffic profile
         :type nodes_info: dict
@@ -101,55 +154,30 @@ class TrafficGenerator(object):
         ssh = SSH()
         ssh.connect(node)
 
-        tg_port3_src_mac = Topology.get_interface_mac_by_port_key(node, "port3")
-        _, adj_int = Topology.\
-            get_adjacent_node_and_interface_by_key(nodes_info, node, "port3")
-        tg_port3_dst_mac = adj_int['mac_address']
-
-        tg_port5_src_mac = Topology.get_interface_mac_by_port_key(node, "port5")
-        _, adj_int = Topology.\
-            get_adjacent_node_and_interface_by_key(nodes_info, node, "port5")
-        tg_port5_dst_mac = adj_int['mac_address']
-
-
         if node['subtype'] == NodeSubTypeTG.TREX:
             if traffic_type in ["3-node-xconnect", "3-node-bridge"]:
                 (ret, stdout, stderr) = ssh.exec_command(
-                    "sh -c '/tmp/openvpp-testing/resources/tools/t-rex-stateless.py "
-                    "-d {0} -r {1}% -s {2} "
-                    "--p1_src_mac 52:00:00:00:00:01 "
-                    "--p1_dst_mac 52:00:00:00:00:02 "
+                    "sh -c '/tmp/openvpp-testing/resources/tools/t-rex/t-rex-stateless.py "
+                    "-d {0} -r {1} -s {2} "
                     "--p1_src_start_ip 10.10.10.1 "
                     "--p1_src_end_ip 10.10.10.254 "
                     "--p1_dst_start_ip 20.20.20.1 "
-                    "--p1_dst_end_ip 20.20.20.254 "
-                    "--p2_src_mac 52:00:00:00:00:02 "
-                    "--p2_dst_mac 52:00:00:00:00:01 "
                     "--p2_src_start_ip 20.20.20.1 "
                     "--p2_src_end_ip 20.20.20.254 "
-                    "--p2_dst_start_ip 10.10.10.1 "
-                    "--p2_dst_end_ip 10.10.10.254'".\
+                    "--p2_dst_start_ip 10.10.10.1'".\
                     format(duration, rate, framesize), timeout=int(duration)+60)
             elif traffic_type in ["3-node-IPv4"]:
                 (ret, stdout, stderr) = ssh.exec_command(
-                    "sh -c '/tmp/openvpp-testing/resources/tools/t-rex-stateless.py "
-                    "-d {0} -r {1}% -s {2} "
-                    "--p1_src_mac {3} "
-                    "--p1_dst_mac {4} "
+                    "sh -c '/tmp/openvpp-testing/resources/tools/t-rex/t-rex-stateless.py "
+                    "-d {0} -r {1} -s {2} "
                     "--p1_src_start_ip 10.10.10.2 "
                     "--p1_src_end_ip 10.10.10.254 "
                     "--p1_dst_start_ip 20.20.20.2 "
-                    "--p1_dst_end_ip 20.20.20.2 "
-                    "--p2_src_mac {5} "
-                    "--p2_dst_mac {6} "
                     "--p2_src_start_ip 20.20.20.2 "
                     "--p2_src_end_ip 20.20.20.254 "
-                    "--p2_dst_start_ip 10.10.10.2 "
-                    "--p2_dst_end_ip 10.10.10.2'".\
-                    format(duration, rate, framesize,\
-                           tg_port3_src_mac, tg_port3_dst_mac,\
-                           tg_port5_src_mac, tg_port5_dst_mac),\
-                           timeout=int(duration)+60)
+                    "--p2_dst_start_ip 10.10.10.2'".\
+                    format(duration, rate, framesize),\
+                    timeout=int(duration)+60)
             else:
                 raise NotImplementedError('Unsupported traffic type')