Last bulk update of CSIT.

Change-Id: I815e4d54e74a49fd19a9927554ce5c37a2719f7e
Signed-off-by: Stefan Kobza <skobza@cisco.com>

4 files changed, 267 insertions, 44 deletions

diff --git a/resources/libraries/python/IPv6Setup.py b/resources/libraries/python/IPv6Setup.py
index 45a8eba58d..db497757bc 100644
--- a/resources/libraries/python/IPv6Setup.py
+++ b/resources/libraries/python/IPv6Setup.py
@@ -248,42 +248,33 @@ class IPv6Setup(object):
                 self.vpp_ra_supress_link_layer(node, if_name)
 
     @staticmethod
-    def vpp_ipv6_route_add(node, link, interface, nodes_addr):
-        """Setup IPv6 route on the VPP node.
-
-           :param node: Node to add route on.
-           :param link: Route to following link.
-           :param interface: Route output interface.
-           :param nodes_addr: Available nodes IPv6 adresses.
-           :type node: dict
-           :type link: str
-           :type interface: str
-           :type nodes_addr: dict
+    def get_link_address(link, nodes_addr):
+        """Get link IPv6 address.
+
+        :param link: Link name.
+        :param nodes_addr: Available nodes IPv6 adresses.
+        :type link: str
+        :type nodes_addr: dict
+        :return: Link IPv6 address.
+        :rtype: str
         """
-        ssh = SSH()
-        ssh.connect(node)
-
-        # Get route destination address from link name
         net = nodes_addr.get(link)
         if net is None:
-            raise ValueError('No network for link "{0}"'.format(link))
-        dst_net = '{0}/{1}'.format(net['net_addr'], net['prefix'])
+            raise ValueError('Link "{0}" address not found'.format(link))
+        return net.get('net_addr')
 
-        # Get next-hop address
-        nh_addr = None
-        for net in nodes_addr.values():
-            for port in net['ports'].values():
-                if port['if'] == interface and port['node'] == node['host']:
-                    for nh in net['ports'].values():
-                        if nh['if'] != interface and nh['node'] != node['host']:
-                            nh_addr = nh['addr']
-        if nh_addr is None:
-            raise Exception('next-hop not found')
-
-        cmd_input = 'ip_add_del_route {0} via {1} {2} resolve-attempts 10'. \
-            format(dst_net, nh_addr, interface)
-        (ret_code, _, _) = ssh.exec_command_sudo(Constants.VAT_BIN_NAME,
-                                                 cmd_input)
-        if int(ret_code) != 0:
-            raise Exception("'{0}' failed on {1}".format(cmd_input,
-                                                         node['host']))
+    @staticmethod
+    def get_link_prefix(link, nodes_addr):
+        """Get link IPv6 address prefix.
+
+        :param link: Link name.
+        :param nodes_addr: Available nodes IPv6 adresses.
+        :type link: str
+        :type nodes_addr: dict
+        :return: Link IPv6 address prefix.
+        :rtype: int
+        """
+        net = nodes_addr.get(link)
+        if net is None:
+            raise ValueError('Link "{0}" address not found'.format(link))
+        return net.get('prefix')
diff --git a/resources/libraries/python/Routing.py b/resources/libraries/python/Routing.py
new file mode 100644
index 0000000000..b22516a7c4
--- /dev/null
+++ b/resources/libraries/python/Routing.py
@@ -0,0 +1,45 @@
+# Copyright (c) 2016 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.
+
+"""Routing utilities library."""
+
+from VatExecutor import VatTerminal
+from topology import Topology
+
+
+class Routing(object):
+    """Routing utilities."""
+
+    @staticmethod
+    def vpp_route_add(node, network, prefix_len, gateway, interface):
+        """Add route to the VPP node.
+
+        :param node: Node to add route on.
+        :param network: Route destination network address.
+        :param prefix_len: Route destination network prefix length.
+        :param gateway: Route gateway address.
+        :param interface: Route interface.
+        :type node: str
+        :type network: str
+        :type prefix_len: int
+        :type gateway: str
+        :type interface: str
+        """
+        sw_if_index = Topology.get_interface_sw_index(node, interface)
+        vat = VatTerminal(node)
+        vat.vat_terminal_exec_cmd_from_template('add_route.vat',
+                                                network=network,
+                                                prefix_length=prefix_len,
+                                                gateway=gateway,
+                                                sw_if_index=sw_if_index)
+        vat.vat_terminal_close()
diff --git a/resources/libraries/python/TrafficGenerator.py b/resources/libraries/python/TrafficGenerator.py
index d86917a181..24bec0af6c 100644
--- a/resources/libraries/python/TrafficGenerator.py
+++ b/resources/libraries/python/TrafficGenerator.py
@@ -10,12 +10,20 @@
 # 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.
-from ssh import SSH
+
+"""Traffic generator library."""
+
 from robot.api import logger
 
+from resources.libraries.python.ssh import SSH
+from resources.libraries.python.topology import NodeType
+from resources.libraries.python.topology import NodeSubTypeTG
+from resources.libraries.python.topology import Topology
+
 __all__ = ['TrafficGenerator']
 
 class TrafficGenerator(object):
+    """Traffic Generator"""
 
     def __init__(self):
         self._result = None
@@ -23,18 +31,130 @@ class TrafficGenerator(object):
         self._sent = None
         self._received = None
 
+    @staticmethod
+    def initialize_traffic_generator(node, interface1, interface2):
+        """TG initialization
+        :param node: Traffic generator node
+        :param interface1: PCI address of first interface
+        :param interface2: PCI address of second interface
+        :type node: dict
+        :type interface1: str
+        :type interface2: str
+        :return: nothing
+        """
+
+        if node['type'] != NodeType.TG:
+            raise Exception('Node type is not a TG')
+        if node['subtype'] == NodeSubTypeTG.TREX:
+            ssh = SSH()
+            ssh.connect(node)
+
+            (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 &'")
+
+    @staticmethod
+    def teardown_traffic_generator(node):
+        """TG teardown
+        :param node: Traffic generator node
+        :type node: dict
+        :return: nothing
+        """
+
+        if node['type'] != NodeType.TG:
+            raise Exception('Node type is not a TG')
+        if node['subtype'] == NodeSubTypeTG.TREX:
+            ssh = SSH()
+            ssh.connect(node)
+            (ret, stdout, stderr) = ssh.exec_command(
+                "sh -c 'sudo pkill t-rex'")
+
+    def send_traffic_on(self, nodes_info, duration, rate,
+                        framesize, traffic_type):
+        """Send traffic from all configured interfaces on TG
+        :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 framesize: Frame size (L2) in Bytes
+        :param traffic_type: Traffic profile
+        :type nodes_info: dict
+        :type duration: str
+        :type rate: str
+        :type framesize: str
+        :type traffic_type: str
+        :return: TG output
+        :rtype: str
+        """
+
+        node = nodes_info["TG"]
+
+        if node['type'] != NodeType.TG:
+            raise Exception('Node type is not a TG')
+
+        if node['subtype'] is None:
+            raise Exception('TG subtype not defined')
 
-    def send_traffic_on(self, node, tx_port, rx_port, duration, rate,
-            framesize):
         ssh = SSH()
         ssh.connect(node)
 
-        (ret, stdout, stderr) = ssh.exec_command(
-                "sh -c 'cd MoonGen && sudo -S build/MoonGen "
-                "rfc2544/benchmarks/vpp-frameloss.lua --txport 0 --rxport 1 "
-                "--duration {0} --rate {1} --framesize {2}'".format(
-                    duration, rate, framesize),
-                timeout=int(duration)+60)
+        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 "
+                    "--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'".\
+                    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} "
+                    "--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)
+            else:
+                raise NotImplementedError('Unsupported traffic type')
+
+        else:
+            raise NotImplementedError("TG subtype not supported")
 
         logger.trace(ret)
         logger.trace(stdout)
@@ -51,6 +171,10 @@ class TrafficGenerator(object):
         return self._result
 
     def no_traffic_loss_occured(self):
+        """Fail is loss occured in traffic run
+        :return: nothing
+        """
+
         if self._loss is None:
             raise Exception('The traffic generation has not been issued')
         if self._loss != '0':
diff --git a/resources/libraries/python/topology.py b/resources/libraries/python/topology.py
index 2b202e5b4a..3ced69d7fe 100644
--- a/resources/libraries/python/topology.py
+++ b/resources/libraries/python/topology.py
@@ -43,6 +43,14 @@ class NodeType(object):
     # Traffic Generator (this node has traffic generator on it)
     TG = 'TG'
 
+class NodeSubTypeTG(object):
+    #T-Rex traffic generator
+    TREX = 'TREX'
+    # Moongen
+    MOONGEN = 'MOONGEN'
+    #IxNetwork
+    IXNET = 'IXNET'
+
 DICT__nodes = load_topo_from_yaml()
 
 
@@ -346,6 +354,22 @@ class Topology(object):
         return None
 
     @staticmethod
+    def get_interface_mac_by_port_key(node, port_key):
+        """Get MAC address for the interface based on port key.
+
+        :param node: Node to get interface mac on.
+        :param port_key: Dictionary key name of interface.
+        :type node: dict
+        :type port_key: str
+        :return: Return MAC or None if not found.
+        """
+        for port_name, port_data in node['interfaces'].iteritems():
+            if port_name == port_key:
+                return port_data['mac_address']
+
+        return None
+
+    @staticmethod
     def get_interface_mac(node, interface):
         """Get MAC address for the interface.
 
@@ -365,6 +389,45 @@ class Topology(object):
         return None
 
     @staticmethod
+    def get_adjacent_node_and_interface_by_key(nodes_info, node, port_key):
+        """Get node and interface adjacent to specified interface
+        on local network.
+
+        :param nodes_info: Dictionary containing information on all nodes
+        in topology.
+        :param node: Node that contains specified interface.
+        :param port_key: Interface port key.
+        :type nodes_info: dict
+        :type node: dict
+        :type port_key: str
+        :return: Return (node, interface info) tuple or None if not found.
+        :rtype: (dict, dict)
+        """
+        link_name = None
+       # get link name where the interface belongs to
+        for port_name, port_data in node['interfaces'].iteritems():
+            if port_name == 'mgmt':
+                continue
+            if port_name == port_key:
+                link_name = port_data['link']
+                break
+
+        if link_name is None: 
+            return None
+
+        # find link
+        for node_data in nodes_info.values():
+            # skip self
+            if node_data['host'] == node['host']:
+                continue
+            for interface, interface_data \
+                    in node_data['interfaces'].iteritems():
+                if 'link' not in interface_data:
+                    continue
+                if interface_data['link'] == link_name:
+                    return node_data, node_data['interfaces'][interface]
+
+    @staticmethod
     def get_adjacent_node_and_interface(nodes_info, node, interface_name):
         """Get node and interface adjacent to specified interface
         on local network.