IXIA traffic generator

Change-Id: I09bcb3f2c4b945283070d442589d3bf00468abbc
Signed-off-by: Deepak S <deepak.s@linux.intel.com>
Signed-off-by: Edward MacGillivray <edward.s.macgillivray@intel.com>
Signed-off-by: Ross Brattain <ross.b.brattain@intel.com>

2 files changed, 779 insertions, 0 deletions

diff --git a/tests/unit/network_services/vnf_generic/vnf/test_tg_ixload.py b/tests/unit/network_services/vnf_generic/vnf/test_tg_ixload.py
new file mode 100644
index 000000000..cda44127e
--- /dev/null
+++ b/tests/unit/network_services/vnf_generic/vnf/test_tg_ixload.py
@@ -0,0 +1,374 @@
+#!/usr/bin/env python
+
+# Copyright (c) 2016-2017 Intel Corporation
+#
+# 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.
+#
+
+from __future__ import absolute_import
+import unittest
+import mock
+import subprocess
+
+STL_MOCKS = {
+    'stl': mock.MagicMock(),
+    'stl.trex_stl_lib': mock.MagicMock(),
+    'stl.trex_stl_lib.base64': mock.MagicMock(),
+    'stl.trex_stl_lib.binascii': mock.MagicMock(),
+    'stl.trex_stl_lib.collections': mock.MagicMock(),
+    'stl.trex_stl_lib.copy': mock.MagicMock(),
+    'stl.trex_stl_lib.datetime': mock.MagicMock(),
+    'stl.trex_stl_lib.functools': mock.MagicMock(),
+    'stl.trex_stl_lib.imp': mock.MagicMock(),
+    'stl.trex_stl_lib.inspect': mock.MagicMock(),
+    'stl.trex_stl_lib.json': mock.MagicMock(),
+    'stl.trex_stl_lib.linecache': mock.MagicMock(),
+    'stl.trex_stl_lib.math': mock.MagicMock(),
+    'stl.trex_stl_lib.os': mock.MagicMock(),
+    'stl.trex_stl_lib.platform': mock.MagicMock(),
+    'stl.trex_stl_lib.pprint': mock.MagicMock(),
+    'stl.trex_stl_lib.random': mock.MagicMock(),
+    'stl.trex_stl_lib.re': mock.MagicMock(),
+    'stl.trex_stl_lib.scapy': mock.MagicMock(),
+    'stl.trex_stl_lib.socket': mock.MagicMock(),
+    'stl.trex_stl_lib.string': mock.MagicMock(),
+    'stl.trex_stl_lib.struct': mock.MagicMock(),
+    'stl.trex_stl_lib.sys': mock.MagicMock(),
+    'stl.trex_stl_lib.threading': mock.MagicMock(),
+    'stl.trex_stl_lib.time': mock.MagicMock(),
+    'stl.trex_stl_lib.traceback': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_async_client': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_client': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_exceptions': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_ext': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_jsonrpc_client': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_packet_builder_interface': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_packet_builder_scapy': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_port': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_stats': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_streams': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_types': mock.MagicMock(),
+    'stl.trex_stl_lib.types': mock.MagicMock(),
+    'stl.trex_stl_lib.utils': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.argparse': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.collections': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.common': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.json': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.os': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.parsing_opts': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.pwd': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.random': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.re': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.string': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.sys': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.text_opts': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.text_tables': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.texttable': mock.MagicMock(),
+    'stl.trex_stl_lib.warnings': mock.MagicMock(),
+    'stl.trex_stl_lib.yaml': mock.MagicMock(),
+    'stl.trex_stl_lib.zlib': mock.MagicMock(),
+    'stl.trex_stl_lib.zmq': mock.MagicMock(),
+}
+
+STLClient = mock.MagicMock()
+stl_patch = mock.patch.dict("sys.modules", STL_MOCKS)
+stl_patch.start()
+
+if stl_patch:
+    from yardstick.network_services.vnf_generic.vnf.tg_ixload import IxLoadTrafficGen
+    from yardstick.network_services.vnf_generic.vnf.tg_ixload import IxLoadResourceHelper
+    from yardstick.network_services.traffic_profile.base import TrafficProfile
+
+
+NAME = "tg__1"
+
+
+class TestIxLoadTrafficGen(unittest.TestCase):
+    VNFD = {'vnfd:vnfd-catalog':
+            {'vnfd':
+             [{'short-name': 'VpeVnf',
+               'vdu':
+               [{'routing_table':
+                 [{'network': '152.16.100.20',
+                   'netmask': '255.255.255.0',
+                   'gateway': '152.16.100.20',
+                   'if': 'xe0'},
+                  {'network': '152.16.40.20',
+                   'netmask': '255.255.255.0',
+                   'gateway': '152.16.40.20',
+                   'if': 'xe1'}],
+                 'description': 'VPE approximation using DPDK',
+                 'name': 'vpevnf-baremetal',
+                 'nd_route_tbl':
+                 [{'network': '0064:ff9b:0:0:0:0:9810:6414',
+                   'netmask': '112',
+                   'gateway': '0064:ff9b:0:0:0:0:9810:6414',
+                   'if': 'xe0'},
+                  {'network': '0064:ff9b:0:0:0:0:9810:2814',
+                   'netmask': '112',
+                   'gateway': '0064:ff9b:0:0:0:0:9810:2814',
+                   'if': 'xe1'}],
+                 'id': 'vpevnf-baremetal',
+                 'external-interface':
+                 [{'virtual-interface':
+                   {'dst_mac': '00:00:00:00:00:04',
+                    'vpci': '0000:05:00.0',
+                    'local_ip': '152.16.100.19',
+                    'type': 'PCI-PASSTHROUGH',
+                    'netmask': '255.255.255.0',
+                    'dpdk_port_num': '0',
+                    'bandwidth': '10 Gbps',
+                    'driver': "i40e",
+                    'dst_ip': '152.16.100.20',
+                    'local_iface_name': 'xe0',
+                    'local_mac': '00:00:00:00:00:02'},
+                   'vnfd-connection-point-ref': 'xe0',
+                   'name': 'xe0'},
+                  {'virtual-interface':
+                   {'dst_mac': '00:00:00:00:00:03',
+                    'vpci': '0000:05:00.1',
+                    'local_ip': '152.16.40.19',
+                    'type': 'PCI-PASSTHROUGH',
+                    'driver': "i40e",
+                    'netmask': '255.255.255.0',
+                    'dpdk_port_num': '1',
+                    'bandwidth': '10 Gbps',
+                    'dst_ip': '152.16.40.20',
+                    'local_iface_name': 'xe1',
+                    'local_mac': '00:00:00:00:00:01'},
+                   'vnfd-connection-point-ref': 'xe1',
+                   'name': 'xe1'}]}],
+               'description': 'Vpe approximation using DPDK',
+               'mgmt-interface':
+                   {'vdu-id': 'vpevnf-baremetal',
+                    'host': '1.1.1.1',
+                    'password': 'r00t',
+                    'user': 'root',
+                    'ip': '1.1.1.1'},
+               'benchmark':
+                   {'kpi': ['packets_in', 'packets_fwd', 'packets_dropped']},
+               'connection-point': [{'type': 'VPORT', 'name': 'xe0'},
+                                    {'type': 'VPORT', 'name': 'xe1'}],
+               'id': 'VpeApproxVnf', 'name': 'VPEVnfSsh'}]}}
+
+    TRAFFIC_PROFILE = {
+        "schema": "isb:traffic_profile:0.1",
+        "name": "fixed",
+        "description": "Fixed traffic profile to run UDP traffic",
+        "traffic_profile": {
+            "traffic_type": "FixedTraffic",
+            "frame_rate": 100,  # pps
+            "flow_number": 10,
+            "frame_size": 64}}
+
+    def test___init__(self):
+        with mock.patch("yardstick.ssh.SSH") as ssh:
+            ssh_mock = mock.Mock(autospec=ssh.SSH)
+            ssh_mock.execute = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh.from_node.return_value = ssh_mock
+            vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+            ixload_traffic_gen = IxLoadTrafficGen(NAME, vnfd)
+            self.assertIsNone(ixload_traffic_gen.data)
+
+    def test_collect_kpi(self):
+        with mock.patch("yardstick.ssh.SSH") as ssh:
+            ssh_mock = mock.Mock(autospec=ssh.SSH)
+            ssh_mock.execute = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh.from_node.return_value = ssh_mock
+            vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+            ixload_traffic_gen = IxLoadTrafficGen(NAME, vnfd)
+            ixload_traffic_gen.data = {}
+            restult = ixload_traffic_gen.collect_kpi()
+            self.assertEqual({}, restult)
+
+    def test_listen_traffic(self):
+        with mock.patch("yardstick.ssh.SSH") as ssh:
+            ssh_mock = mock.Mock(autospec=ssh.SSH)
+            ssh_mock.execute = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh.from_node.return_value = ssh_mock
+            vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+            ixload_traffic_gen = IxLoadTrafficGen(NAME, vnfd)
+            self.assertEqual(None, ixload_traffic_gen.listen_traffic({}))
+
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.makedirs")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.call")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.shutil")
+    def test_instantiate(self, call, shutil, mock_makedirs):
+        with mock.patch("yardstick.ssh.SSH") as ssh:
+            ssh_mock = mock.Mock(autospec=ssh.SSH)
+            ssh_mock.execute = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh_mock.run = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh.from_node.return_value = ssh_mock
+            vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+            ixload_traffic_gen = IxLoadTrafficGen(NAME, vnfd)
+            scenario_cfg = {'tc': "nsb_test_case",
+                            'ixia_profile': "ixload.cfg"}
+            ixload_traffic_gen.RESULTS_MOUNT = "/tmp/result"
+            shutil.copy = mock.Mock()
+            scenario_cfg.update({'options': {'packetsize': 64, 'traffic_type': 4,
+                                             'rfc2544': {'allowed_drop_rate': '0.8 - 1'},
+                                             'vnf__1': {'rules': 'acl_1rule.yaml',
+                                                        'vnf_config': {'lb_config': 'SW',
+                                                                       'lb_count': 1,
+                                                                       'worker_config':
+                                                                       '1C/1T',
+                                                                       'worker_threads': 1}}
+                                             }})
+            self.assertRaises(IOError,
+                              ixload_traffic_gen.instantiate(scenario_cfg, {}))
+
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.call")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.shutil")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.open")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.min")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.max")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.len")
+    def test_run_traffic(self, call, shutil, main_open, min, max, len):
+        mock_traffic_profile = mock.Mock(autospec=TrafficProfile)
+        mock_traffic_profile.get_traffic_definition.return_value = "64"
+        mock_traffic_profile.params = self.TRAFFIC_PROFILE
+        with mock.patch("yardstick.ssh.SSH") as ssh:
+            ssh_mock = mock.Mock(autospec=ssh.SSH)
+            ssh_mock.execute = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh_mock.run = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh.from_node.return_value = ssh_mock
+            vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+            vnfd["mgmt-interface"].update({"tg-config": {}})
+            vnfd["mgmt-interface"]["tg-config"].update({"ixchassis":
+                                                        "1.1.1.1"})
+            vnfd["mgmt-interface"]["tg-config"].update({"py_bin_path":
+                                                        "/root"})
+            sut = IxLoadTrafficGen(NAME, vnfd)
+            sut.connection = mock.Mock()
+            sut.connection.run = mock.Mock()
+            sut._traffic_runner = mock.Mock(return_value=0)
+            shutil.copy = mock.Mock()
+            result = sut.run_traffic(mock_traffic_profile)
+            self.assertIsNone(result)
+
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.call")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.shutil")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.open")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.min")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.max")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.len")
+    def test_run_traffic_csv(self, call, shutil, main_open, min, max, len):
+        mock_traffic_profile = mock.Mock(autospec=TrafficProfile)
+        mock_traffic_profile.get_traffic_definition.return_value = "64"
+        mock_traffic_profile.params = self.TRAFFIC_PROFILE
+        with mock.patch("yardstick.ssh.SSH") as ssh:
+            ssh_mock = mock.Mock(autospec=ssh.SSH)
+            ssh_mock.execute = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh_mock.run = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh.from_node.return_value = ssh_mock
+            vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+            vnfd["mgmt-interface"].update({"tg-config": {}})
+            vnfd["mgmt-interface"]["tg-config"].update({"ixchassis":
+                                                        "1.1.1.1"})
+            vnfd["mgmt-interface"]["tg-config"].update({"py_bin_path":
+                                                        "/root"})
+            sut = IxLoadTrafficGen(NAME, vnfd)
+            sut.connection = mock.Mock()
+            sut.connection.run = mock.Mock()
+            sut._traffic_runner = mock.Mock(return_value=0)
+            shutil.copy = mock.Mock()
+            subprocess.call(["touch", "/tmp/1.csv"])
+            sut.rel_bin_path = mock.Mock(return_value="/tmp/*.csv")
+            result = sut.run_traffic(mock_traffic_profile)
+            self.assertIsNone(result)
+
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.call")
+    def test_terminate(self, call):
+        with mock.patch("yardstick.ssh.SSH") as ssh:
+            vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+            ssh_mock = mock.Mock(autospec=ssh.SSH)
+            ssh_mock.execute = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh.from_node.return_value = ssh_mock
+            ixload_traffic_gen = IxLoadTrafficGen(NAME, vnfd)
+            self.assertEqual(None, ixload_traffic_gen.terminate())
+
+    @mock.patch("yardstick.ssh.SSH")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.call")
+    def test_parse_csv_read(self, mock_call, mock_ssh):
+        vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+        kpi_data = {
+            'HTTP Total Throughput (Kbps)': 1,
+            'HTTP Simulated Users': 2,
+            'HTTP Concurrent Connections': '3',
+            'HTTP Connection Rate': 4.3,
+            'HTTP Transaction Rate': True,
+        }
+        http_reader = [kpi_data]
+
+        mock_ssh_type = mock.Mock(autospec=mock_ssh.SSH)
+        mock_ssh_type.execute.return_value = 0, "", ""
+        mock_ssh.from_node.return_value = mock_ssh_type
+
+        ixload_traffic_gen = IxLoadTrafficGen(NAME, vnfd)
+        result = ixload_traffic_gen.resource_helper.result
+
+        ixload_traffic_gen.resource_helper.parse_csv_read(http_reader)
+        for key_left, key_right in IxLoadResourceHelper.KPI_LIST.items():
+            self.assertEqual(result[key_left][-1], int(kpi_data[key_right]))
+
+    @mock.patch("yardstick.ssh.SSH")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.call")
+    def test_parse_csv_read_value_error(self, mock_call, mock_ssh):
+        vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+        http_reader = [{
+            'HTTP Total Throughput (Kbps)': 1,
+            'HTTP Simulated Users': 2,
+            'HTTP Concurrent Connections': "not a number",
+            'HTTP Connection Rate': 4,
+            'HTTP Transaction Rate': 5,
+        }]
+
+        mock_ssh_type = mock.Mock(autospec=mock_ssh.SSH)
+        mock_ssh_type.execute.return_value = 0, "", ""
+        mock_ssh.from_node.return_value = mock_ssh_type
+
+        ixload_traffic_gen = IxLoadTrafficGen(NAME, vnfd)
+        init_value = ixload_traffic_gen.resource_helper.result
+
+        ixload_traffic_gen.resource_helper.parse_csv_read(http_reader)
+        self.assertDictEqual(ixload_traffic_gen.resource_helper.result, init_value)
+
+    @mock.patch("yardstick.ssh.SSH")
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_ixload.call")
+    def test_parse_csv_read_error(self, mock_call, mock_ssh):
+        vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+        http_reader = [{
+            'HTTP Total Throughput (Kbps)': 1,
+            'HTTP Simulated Users': 2,
+            'HTTP Concurrent Connections': 3,
+            'HTTP Transaction Rate': 5,
+        }]
+
+        mock_ssh_type = mock.Mock(autospec=mock_ssh.SSH)
+        mock_ssh_type.execute.return_value = 0, "", ""
+        mock_ssh.from_node.return_value = mock_ssh_type
+
+        ixload_traffic_gen = IxLoadTrafficGen(NAME, vnfd)
+
+        with self.assertRaises(KeyError):
+            ixload_traffic_gen.resource_helper.parse_csv_read(http_reader)
diff --git a/tests/unit/network_services/vnf_generic/vnf/test_tg_rfc2544_ixia.py b/tests/unit/network_services/vnf_generic/vnf/test_tg_rfc2544_ixia.py
new file mode 100644
index 000000000..8f7f05772
--- /dev/null
+++ b/tests/unit/network_services/vnf_generic/vnf/test_tg_rfc2544_ixia.py
@@ -0,0 +1,405 @@
+#!/usr/bin/env python
+
+# Copyright (c) 2016-2017 Intel Corporation
+#
+# 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.
+#
+
+from __future__ import absolute_import
+import os
+import unittest
+import mock
+STL_MOCKS = {
+    'stl': mock.MagicMock(),
+    'stl.trex_stl_lib': mock.MagicMock(),
+    'stl.trex_stl_lib.base64': mock.MagicMock(),
+    'stl.trex_stl_lib.binascii': mock.MagicMock(),
+    'stl.trex_stl_lib.collections': mock.MagicMock(),
+    'stl.trex_stl_lib.copy': mock.MagicMock(),
+    'stl.trex_stl_lib.datetime': mock.MagicMock(),
+    'stl.trex_stl_lib.functools': mock.MagicMock(),
+    'stl.trex_stl_lib.imp': mock.MagicMock(),
+    'stl.trex_stl_lib.inspect': mock.MagicMock(),
+    'stl.trex_stl_lib.json': mock.MagicMock(),
+    'stl.trex_stl_lib.linecache': mock.MagicMock(),
+    'stl.trex_stl_lib.math': mock.MagicMock(),
+    'stl.trex_stl_lib.os': mock.MagicMock(),
+    'stl.trex_stl_lib.platform': mock.MagicMock(),
+    'stl.trex_stl_lib.pprint': mock.MagicMock(),
+    'stl.trex_stl_lib.random': mock.MagicMock(),
+    'stl.trex_stl_lib.re': mock.MagicMock(),
+    'stl.trex_stl_lib.scapy': mock.MagicMock(),
+    'stl.trex_stl_lib.socket': mock.MagicMock(),
+    'stl.trex_stl_lib.string': mock.MagicMock(),
+    'stl.trex_stl_lib.struct': mock.MagicMock(),
+    'stl.trex_stl_lib.sys': mock.MagicMock(),
+    'stl.trex_stl_lib.threading': mock.MagicMock(),
+    'stl.trex_stl_lib.time': mock.MagicMock(),
+    'stl.trex_stl_lib.traceback': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_async_client': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_client': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_exceptions': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_ext': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_jsonrpc_client': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_packet_builder_interface': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_packet_builder_scapy': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_port': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_stats': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_streams': mock.MagicMock(),
+    'stl.trex_stl_lib.trex_stl_types': mock.MagicMock(),
+    'stl.trex_stl_lib.types': mock.MagicMock(),
+    'stl.trex_stl_lib.utils': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.argparse': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.collections': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.common': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.json': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.os': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.parsing_opts': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.pwd': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.random': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.re': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.string': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.sys': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.text_opts': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.text_tables': mock.MagicMock(),
+    'stl.trex_stl_lib.utils.texttable': mock.MagicMock(),
+    'stl.trex_stl_lib.warnings': mock.MagicMock(),
+    'stl.trex_stl_lib.yaml': mock.MagicMock(),
+    'stl.trex_stl_lib.zlib': mock.MagicMock(),
+    'stl.trex_stl_lib.zmq': mock.MagicMock(),
+}
+
+STLClient = mock.MagicMock()
+stl_patch = mock.patch.dict("sys.modules", STL_MOCKS)
+stl_patch.start()
+
+if stl_patch:
+    from yardstick.network_services.vnf_generic.vnf.tg_rfc2544_ixia import IxiaTrafficGen
+    from yardstick.network_services.vnf_generic.vnf.tg_rfc2544_ixia import IxiaRfc2544Helper
+    from yardstick.network_services.vnf_generic.vnf.tg_rfc2544_ixia import IxiaResourceHelper
+    from yardstick.network_services.traffic_profile.base import TrafficProfile
+
+TEST_FILE_YAML = 'nsb_test_case.yaml'
+
+
+NAME = "tg__1"
+
+
+@mock.patch("yardstick.network_services.vnf_generic.vnf.tg_rfc2544_ixia.IxNextgen")
+class TestIxiaResourceHelper(unittest.TestCase):
+
+    def test___init___with_custom_rfc_helper(self, mock_ix_nextgen):
+        class MyRfcHelper(IxiaRfc2544Helper):
+            pass
+
+        ixia_resource_helper = IxiaResourceHelper(mock.Mock(), MyRfcHelper)
+        self.assertIsInstance(ixia_resource_helper.rfc_helper, MyRfcHelper)
+
+    def test_stop_collect_with_client(self, mock_ix_nextgen):
+        mock_client = mock.Mock()
+
+        ixia_resource_helper = IxiaResourceHelper(mock.Mock())
+
+        ixia_resource_helper.client = mock_client
+        ixia_resource_helper.stop_collect()
+        self.assertEqual(mock_client.ix_stop_traffic.call_count, 1)
+
+
+@mock.patch("yardstick.network_services.vnf_generic.vnf.tg_rfc2544_ixia.IxNextgen")
+class TestIXIATrafficGen(unittest.TestCase):
+    VNFD = {'vnfd:vnfd-catalog':
+            {'vnfd':
+             [{'short-name': 'VpeVnf',
+               'vdu':
+               [{'routing_table':
+                 [{'network': '152.16.100.20',
+                   'netmask': '255.255.255.0',
+                   'gateway': '152.16.100.20',
+                   'if': 'xe0'},
+                  {'network': '152.16.40.20',
+                   'netmask': '255.255.255.0',
+                   'gateway': '152.16.40.20',
+                   'if': 'xe1'}],
+                 'description': 'VPE approximation using DPDK',
+                 'name': 'vpevnf-baremetal',
+                 'nd_route_tbl':
+                 [{'network': '0064:ff9b:0:0:0:0:9810:6414',
+                   'netmask': '112',
+                   'gateway': '0064:ff9b:0:0:0:0:9810:6414',
+                   'if': 'xe0'},
+                  {'network': '0064:ff9b:0:0:0:0:9810:2814',
+                   'netmask': '112',
+                   'gateway': '0064:ff9b:0:0:0:0:9810:2814',
+                   'if': 'xe1'}],
+                 'id': 'vpevnf-baremetal',
+                 'external-interface':
+                 [{'virtual-interface':
+                   {'dst_mac': '00:00:00:00:00:04',
+                    'vpci': '0000:05:00.0',
+                    'local_ip': '152.16.100.19',
+                    'type': 'PCI-PASSTHROUGH',
+                    'netmask': '255.255.255.0',
+                    'dpdk_port_num': '0',
+                    'bandwidth': '10 Gbps',
+                    'driver': "i40e",
+                    'dst_ip': '152.16.100.20',
+                    'local_iface_name': 'xe0',
+                    'local_mac': '00:00:00:00:00:02'},
+                   'vnfd-connection-point-ref': 'xe0',
+                   'name': 'xe0'},
+                  {'virtual-interface':
+                   {'dst_mac': '00:00:00:00:00:03',
+                    'vpci': '0000:05:00.1',
+                    'local_ip': '152.16.40.19',
+                    'type': 'PCI-PASSTHROUGH',
+                    'driver': "i40e",
+                    'netmask': '255.255.255.0',
+                    'dpdk_port_num': '1',
+                    'bandwidth': '10 Gbps',
+                    'dst_ip': '152.16.40.20',
+                    'local_iface_name': 'xe1',
+                    'local_mac': '00:00:00:00:00:01'},
+                   'vnfd-connection-point-ref': 'xe1',
+                   'name': 'xe1'}]}],
+               'description': 'Vpe approximation using DPDK',
+               'mgmt-interface':
+                   {'vdu-id': 'vpevnf-baremetal',
+                    'host': '1.1.1.1',
+                    'password': 'r00t',
+                    'user': 'root',
+                    'ip': '1.1.1.1'},
+               'benchmark':
+                   {'kpi': ['packets_in', 'packets_fwd', 'packets_dropped']},
+               'connection-point': [{'type': 'VPORT', 'name': 'xe0'},
+                                    {'type': 'VPORT', 'name': 'xe1'}],
+               'id': 'VpeApproxVnf', 'name': 'VPEVnfSsh'}]}}
+
+    TRAFFIC_PROFILE = {
+        "schema": "isb:traffic_profile:0.1",
+        "name": "fixed",
+        "description": "Fixed traffic profile to run UDP traffic",
+        "traffic_profile": {
+            "traffic_type": "FixedTraffic",
+            "frame_rate": 100,  # pps
+            "flow_number": 10,
+            "frame_size": 64}}
+
+    TC_YAML = {'scenarios': [{'tc_options':
+                              {'rfc2544': {'allowed_drop_rate': '0.8 - 1'}},
+                              'runner': {'duration': 400,
+                                         'interval': 35, 'type': 'Duration'},
+                              'traffic_options':
+                              {'flow': 'ipv4_1flow_Packets_vpe.yaml',
+                               'imix': 'imix_voice.yaml'},
+                              'vnf_options': {'vpe': {'cfg': 'vpe_config'}},
+                              'traffic_profile': 'ipv4_throughput_vpe.yaml',
+                              'type': 'NSPerf',
+                              'nodes': {'tg__1': 'trafficgen_1.yardstick',
+                                        'vnf__1': 'vnf.yardstick'},
+                              'topology': 'vpe_vnf_topology.yaml'}],
+               'context': {'nfvi_type': 'baremetal', 'type': 'Node',
+                           'name': 'yardstick',
+                           'file': '/etc/yardstick/nodes/pod.yaml'},
+               'schema': 'yardstick:task:0.1'}
+
+    def test___init__(self, mock_ixnextgen):
+        with mock.patch("yardstick.ssh.SSH") as ssh:
+            ssh_mock = mock.Mock(autospec=ssh.SSH)
+            ssh_mock.execute = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh.from_node.return_value = ssh_mock
+            vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+            ixnet_traffic_gen = IxiaTrafficGen(NAME, vnfd)
+
+    def test_listen_traffic(self, mock_ixnextgen):
+        with mock.patch("yardstick.ssh.SSH") as ssh:
+            ssh_mock = mock.Mock(autospec=ssh.SSH)
+            ssh_mock.execute = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh.from_node.return_value = ssh_mock
+            vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+            ixnet_traffic_gen = IxiaTrafficGen(NAME, vnfd)
+            self.assertEqual(None, ixnet_traffic_gen.listen_traffic({}))
+
+    def test_instantiate(self, mock_ixnextgen):
+        with mock.patch("yardstick.ssh.SSH") as ssh:
+            ssh_mock = mock.Mock(autospec=ssh.SSH)
+            ssh_mock.execute = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh_mock.run = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh.from_node.return_value = ssh_mock
+            vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+            ixnet_traffic_gen = IxiaTrafficGen(NAME, vnfd)
+            scenario_cfg = {'tc': "nsb_test_case", "topology": "",
+                            'ixia_profile': "ixload.cfg"}
+            scenario_cfg.update({'options': {'packetsize': 64,
+                                             'traffic_type': 4,
+                                             'rfc2544': {'allowed_drop_rate': '0.8 - 1'},
+                                             'vnf__1': {'rules': 'acl_1rule.yaml',
+                                                        'vnf_config': {'lb_config': 'SW',
+                                                                       'lb_count': 1,
+                                                                       'worker_config':
+                                                                       '1C/1T',
+                                                                       'worker_threads': 1}}
+                                             }})
+            ixnet_traffic_gen.topology = ""
+            ixnet_traffic_gen.get_ixobj = mock.MagicMock()
+            ixnet_traffic_gen._ixia_traffic_gen = mock.MagicMock()
+            ixnet_traffic_gen._ixia_traffic_gen._connect = mock.Mock()
+            self.assertRaises(
+                IOError,
+                ixnet_traffic_gen.instantiate(scenario_cfg, {}))
+
+    def test_collect_kpi(self, mock_ixnextgen):
+        with mock.patch("yardstick.ssh.SSH") as ssh:
+            ssh_mock = mock.Mock(autospec=ssh.SSH)
+            ssh_mock.execute = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh.from_node.return_value = ssh_mock
+            vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+            ixnet_traffic_gen = IxiaTrafficGen(NAME, vnfd)
+            ixnet_traffic_gen.data = {}
+            restult = ixnet_traffic_gen.collect_kpi()
+            self.assertEqual({}, restult)
+
+    def test_terminate(self, mock_ixnextgen):
+        with mock.patch("yardstick.ssh.SSH") as ssh:
+            vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+            ssh_mock = mock.Mock(autospec=ssh.SSH)
+            ssh_mock.execute = \
+                mock.Mock(return_value=(0, "", ""))
+            ssh.from_node.return_value = ssh_mock
+            ixnet_traffic_gen = IxiaTrafficGen(NAME, vnfd)
+            ixnet_traffic_gen._terminated = mock.MagicMock()
+            ixnet_traffic_gen._terminated.value = 0
+            ixnet_traffic_gen._ixia_traffic_gen = mock.MagicMock()
+            ixnet_traffic_gen._ixia_traffic_gen.ix_stop_traffic = mock.Mock()
+            ixnet_traffic_gen._traffic_process = mock.MagicMock()
+            ixnet_traffic_gen._traffic_process.terminate = mock.Mock()
+            self.assertEqual(None, ixnet_traffic_gen.terminate())
+
+    def _get_file_abspath(self, filename):
+        curr_path = os.path.dirname(os.path.abspath(__file__))
+        file_path = os.path.join(curr_path, filename)
+        return file_path
+
+    def test_scale(self, mock_ix_nextgen):
+        vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+        sut = IxiaTrafficGen('vnf1', vnfd)
+        sut.scale()
+
+    def test__check_status(self, mock_ix_nextgen):
+        vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+        sut = IxiaTrafficGen('vnf1', vnfd)
+        sut._check_status()
+
+    @mock.patch("yardstick.network_services.vnf_generic.vnf.tg_rfc2544_ixia.time")
+    @mock.patch("yardstick.ssh.SSH")
+    def test_traffic_runner(self, mock_ixnextgen, mock_ssh, mock_time):
+        mock_traffic_profile = mock.Mock(autospec=TrafficProfile)
+        mock_traffic_profile.get_traffic_definition.return_value = "64"
+        mock_traffic_profile.params = self.TRAFFIC_PROFILE
+
+        mock_ssh_instance = mock.Mock(autospec=mock_ssh.SSH)
+        mock_ssh_instance.execute.return_value = 0, "", ""
+        mock_ssh_instance.run.return_value = 0, "", ""
+
+        mock_ssh.from_node.return_value = mock_ssh_instance
+
+        vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+        vnfd["mgmt-interface"].update({
+            'tg-config': {
+                "ixchassis": "1.1.1.1",
+                "py_bin_path": "/root",
+            }
+        })
+
+        samples = {}
+        name = ''
+        for ifname in range(1):
+            name = "xe{}".format(ifname)
+            samples[name] = {
+                "Rx_Rate_Kbps": 20,
+                "Tx_Rate_Kbps": 20,
+                "Rx_Rate_Mbps": 10,
+                "Tx_Rate_Mbps": 10,
+                "RxThroughput": 10,
+                "TxThroughput": 10,
+                "Valid_Frames_Rx": 1000,
+                "Frames_Tx": 1000,
+                "in_packets": 1000,
+                "out_packets": 1000,
+            }
+
+        samples.update({"CurrentDropPercentage": 0.0})
+
+        last_res = [
+            0,
+            {
+                "Rx_Rate_Kbps": [20, 20],
+                "Tx_Rate_Kbps": [20, 20],
+                "Rx_Rate_Mbps": [10, 10],
+                "Tx_Rate_Mbps": [10, 10],
+                "CurrentDropPercentage": [0, 0],
+                "RxThroughput": [10, 10],
+                "TxThroughput": [10, 10],
+                "Frames_Tx": [1000, 1000],
+                "in_packets": [1000, 1000],
+                "Valid_Frames_Rx": [1000, 1000],
+                "out_packets": [1000, 1000],
+            },
+        ]
+
+        mock_traffic_profile.execute.return_value = ['Completed', samples]
+        mock_traffic_profile.get_drop_percentage.return_value = ['Completed', samples]
+
+        sut = IxiaTrafficGen(name, vnfd)
+        sut.tg_port_pairs = [[[0], [1]]]
+        sut.vnf_port_pairs = [[[0], [1]]]
+        sut.tc_file_name = self._get_file_abspath(TEST_FILE_YAML)
+        sut.topology = ""
+
+        sut.ssh_helper = mock.Mock()
+        sut._traffic_process = mock.MagicMock()
+        sut.generate_port_pairs = mock.Mock()
+
+        sut._ixia_traffic_gen = mock.MagicMock()
+        sut._ixia_traffic_gen.ix_get_statistics.return_value = last_res
+
+        sut.resource_helper.client = mock.MagicMock()
+        sut.resource_helper.client_started = mock.MagicMock()
+        sut.resource_helper.client_started.value = 1
+
+        sut.scenario_helper.scenario_cfg = {
+            'options': {
+                'packetsize': 64,
+                'traffic_type': 4,
+                'rfc2544': {
+                    'allowed_drop_rate': '0.8 - 1'
+                },
+                'vnf__1': {
+                    'rules': 'acl_1rule.yaml',
+                    'vnf_config': {
+                        'lb_config': 'SW',
+                        'lb_count': 1,
+                        'worker_config': '1C/1T',
+                        'worker_threads': 1,
+                    },
+                },
+            },
+            'ixia_profile': {}
+        }
+
+        result = sut._traffic_runner(mock_traffic_profile)
+        self.assertIsNone(result)