X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=resources%2Flibraries%2Fpython%2FTrafficGenerator.py;h=2037d49bc253ec4b2d151a299e554e80d50907cf;hb=8f285166faf13156a4f7c70adac9a7e20549268f;hp=d86917a1818b67b3f0ec85ca406fd2041a679241;hpb=33499c81c94c2d3baef9d3e9f061cd76ef86fa74;p=csit.git

diff --git a/resources/libraries/python/TrafficGenerator.py b/resources/libraries/python/TrafficGenerator.py
index d86917a181..2037d49bc2 100644
--- a/resources/libraries/python/TrafficGenerator.py
+++ b/resources/libraries/python/TrafficGenerator.py
@@ -10,12 +10,19 @@
 # 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
+
+"""Performance testing 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
+
 __all__ = ['TrafficGenerator']
 
 class TrafficGenerator(object):
+    """Traffic Generator"""
 
     def __init__(self):
         self._result = None
@@ -23,18 +30,110 @@ 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
+        """
+
+        trex_path = "/opt/trex-core-1.91"
+
+        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 {0}/scripts/ && sudo modprobe uio && "
+                "sudo insmod ./ko/src/igb_uio.ko'"\
+                .format(trex_path))
+
+            (ret, stdout, stderr) = ssh.exec_command(
+                "sh -c 'cd {0}/scripts/ && "
+                "sudo nohup ./t-rex-64 -i -c 4 --iom 0 > /dev/null 2>&1 &'"\
+                .format(trex_path))
+
+    @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)
+        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_start_ip 10.10.10.1 "
+                    "--p1_src_end_ip 10.10.10.254 "
+                    "--p1_dst_start_ip 20.20.20.1 "
+                    "--p2_src_start_ip 20.20.20.1 "
+                    "--p2_src_end_ip 20.20.20.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_start_ip 10.10.10.2 "
+                    "--p1_src_end_ip 10.10.10.254 "
+                    "--p1_dst_start_ip 20.20.20.2 "
+                    "--p2_src_start_ip 20.20.20.2 "
+                    "--p2_src_end_ip 20.20.20.254 "
+                    "--p2_dst_start_ip 10.10.10.2'".\
+                    format(duration, rate, framesize),\
+                    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 +150,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':