Added traffic update capability to Ixload TG

Allow yardstick to update IP/MAX of Net traffic

- Implemented functionality for updating traffics IP/MAC address
  in ixloadconfiguration.
- Extended http_ixload.py with functions for update IP/MAC
  address of netTraffic.
- In HTTP-vFW_IPv4_2Ports_Concurrency.rxf changed name's of
  networks from "client network" to "uplink_0 and the same for
  server

JIRA: YARDSTICK-1418

Change-Id: I28ef68b77466fff15af589954e3ef63e8099428f
Signed-off-by: Myron Sosyak <myronx.sosyak@intel.com>
Signed-off-by: Taras Chornyi <tarasx.chornyi@intel.com>

7 files changed, 289 insertions, 13 deletions

diff --git a/yardstick/common/exceptions.py b/yardstick/common/exceptions.py
index 10c1f3f27..c9d61e45c 100644
--- a/yardstick/common/exceptions.py
+++ b/yardstick/common/exceptions.py
@@ -83,6 +83,10 @@ class InvalidType(YardstickException):
     message = 'Type "%(type_to_convert)s" is not valid'
 
 
+class InvalidRxfFile(YardstickException):
+    message = 'Loaded rxf file has unexpected format'
+
+
 class InfluxDBConfigurationMissing(YardstickException):
     message = ('InfluxDB configuration is not available. Add "influxdb" as '
                'a dispatcher and the configuration section')
diff --git a/yardstick/network_services/traffic_profile/http.py b/yardstick/network_services/traffic_profile/http.py
index 2d00fb849..31ab17ef7 100644
--- a/yardstick/network_services/traffic_profile/http.py
+++ b/yardstick/network_services/traffic_profile/http.py
@@ -24,6 +24,10 @@ class TrafficProfileGenericHTTP(TrafficProfile):
     def __init__(self, TrafficProfile):
         super(TrafficProfileGenericHTTP, self).__init__(TrafficProfile)
 
+    def get_links_param(self):
+        return {k: v for k, v in self.params.items() if
+                "uplink" in k or "downlink" in k}
+
     def execute(self, traffic_generator):
         ''' send run traffic for a selected traffic generator'''
         pass
diff --git a/yardstick/network_services/traffic_profile/http_ixload.py b/yardstick/network_services/traffic_profile/http_ixload.py
index 6cbdb8ab2..3ccec637d 100644
--- a/yardstick/network_services/traffic_profile/http_ixload.py
+++ b/yardstick/network_services/traffic_profile/http_ixload.py
@@ -106,8 +106,10 @@ class IXLOADHttpTest(object):
         self.chassis = None
         self.card = None
         self.ports_to_reassign = None
+        self.links_param = None
         self.test_input = jsonutils.loads(test_input)
         self.parse_run_test()
+        self.test = None
 
     @staticmethod
     def format_ports_for_reassignment(ports):
@@ -171,6 +173,90 @@ class IXLOADHttpTest(object):
             LOG.error('Error: IxLoad config file not found: %s', config_file)
             raise
 
+    def update_network_address(self, net_traffic, address, gateway, prefix):
+        """Update ip address and gateway for net_traffic object
+
+        This function update field which configure source addresses for
+        traffic which is described by net_traffic object.
+        Do not return anything
+
+        :param net_traffic: (IxLoadObjectProxy) proxy obj to tcl net_traffic object
+        :param address: (str) Ipv4 range start address
+        :param gateway: (str) Ipv4 address of gateway
+        :param prefix: (int) subnet prefix
+        :return:
+        """
+        try:
+            ethernet = net_traffic.network.getL1Plugin()
+            ix_net_l2_ethernet_plugin = ethernet.childrenList[0]
+            ix_net_ip_v4_v6_plugin = ix_net_l2_ethernet_plugin.childrenList[0]
+            ix_net_ip_v4_v6_range = ix_net_ip_v4_v6_plugin.rangeList[0]
+
+            ix_net_ip_v4_v6_range.config(
+                prefix=prefix,
+                ipAddress=address,
+                gatewayAddress=gateway)
+        except Exception:
+            raise exceptions.InvalidRxfFile
+
+    def update_network_mac_address(self, net_traffic, mac):
+        """Update MACaddress for net_traffic object
+
+        This function update field which configure MACaddresses for
+        traffic which is described by net_traffic object.
+        If mac == "auto" then will be configured auto generated mac
+        Do not return anything.
+
+        :param net_traffic: (IxLoadObjectProxy) proxy obj to tcl net_traffic object
+        :param mac: (str) MAC
+        :return:
+        """
+        try:
+            ethernet = net_traffic.network.getL1Plugin()
+            ix_net_l2_ethernet_plugin = ethernet.childrenList[0]
+            ix_net_ip_v4_v6_plugin = ix_net_l2_ethernet_plugin.childrenList[0]
+            ix_net_ip_v4_v6_range = ix_net_ip_v4_v6_plugin.rangeList[0]
+
+            if str(mac).lower() == "auto":
+                ix_net_ip_v4_v6_range.config(autoMacGeneration=True)
+            else:
+                ix_net_ip_v4_v6_range.config(autoMacGeneration=False)
+                mac_range = ix_net_ip_v4_v6_range.getLowerRelatedRange(
+                    "MacRange")
+                mac_range.config(mac=mac)
+        except Exception:
+            raise exceptions.InvalidRxfFile
+
+    def update_network_param(self, net_traffic, param):
+        """Update net_traffic by parameters specified in param"""
+
+        self.update_network_address(net_traffic, param["address"],
+                                    param["gateway"], param["subnet_prefix"])
+
+        self.update_network_mac_address(net_traffic, param["mac"])
+
+    def update_config(self):
+        """Update some fields by parameters from traffic profile"""
+
+        net_traffics = {}
+        # self.test.communityList is a IxLoadObjectProxy to some tcl object
+        # which contain all net_traffic objects in scenario.
+        # net_traffic item has a name in format "activity_name@item_name"
+        try:
+            for item in self.test.communityList:
+                net_traffics[item.name.split('@')[1]] = item
+        except Exception:  # pylint: disable=broad-except
+            pass
+
+        for name, net_traffic in net_traffics.items():
+            try:
+                param = self.links_param[name]
+            except KeyError:
+                LOG.debug('There is no param for net_traffic %s', name)
+                continue
+
+            self.update_network_param(net_traffic, param["ip"])
+
     def start_http_test(self):
         self.ix_load = IxLoad()
 
@@ -197,16 +283,18 @@ class IXLOADHttpTest(object):
 
         # Get the first test on the testList
         test_name = repository.testList[0].cget("name")
-        test = repository.testList.getItem(test_name)
+        self.test = repository.testList.getItem(test_name)
 
         self.set_results_dir(test_controller, self.results_on_windows)
 
-        test.config(statsRequired=1, enableResetPorts=1, csvInterval=2,
-                    enableForceOwnership=True)
+        self.test.config(statsRequired=1, enableResetPorts=1, csvInterval=2,
+                         enableForceOwnership=True)
+
+        self.update_config()
 
         #  ---- Remap ports ----
         try:
-            self.reassign_ports(test, repository, self.ports_to_reassign)
+            self.reassign_ports(self.test, repository, self.ports_to_reassign)
         except Exception:  # pylint: disable=broad-except
             LOG.exception("Exception occurred during reassign_ports")
 
@@ -246,7 +334,7 @@ class IXLOADHttpTest(object):
 
         self.stat_utils.StartCollector(self.IxL_StatCollectorCommand)
 
-        test_controller.run(test)
+        test_controller.run(self.test)
         self.ix_load.waitForTestFinish()
 
         test_controller.releaseConfigWaitFinish()
@@ -258,7 +346,7 @@ class IXLOADHttpTest(object):
         test_controller.generateReport(detailedReport=1, format="PDF;HTML")
         test_controller.releaseConfigWaitFinish()
 
-        self.ix_load.delete(test)
+        self.ix_load.delete(self.test)
         self.ix_load.delete(test_controller)
         self.ix_load.delete(logger)
         self.ix_load.delete(log_engine)
@@ -296,6 +384,9 @@ class IXLOADHttpTest(object):
 
         LOG.debug("Ports to be reassigned: %s", self.ports_to_reassign)
 
+        self.links_param = self.test_input["links_param"]
+        LOG.debug("Links param to be applied: %s", self.links_param)
+
 
 def main(args):
     # Get the args from cmdline and parse and run the test
diff --git a/yardstick/network_services/vnf_generic/vnf/tg_ixload.py b/yardstick/network_services/vnf_generic/vnf/tg_ixload.py
index e0fc47dbf..d25402740 100644
--- a/yardstick/network_services/vnf_generic/vnf/tg_ixload.py
+++ b/yardstick/network_services/vnf_generic/vnf/tg_ixload.py
@@ -20,9 +20,11 @@ import os
 import shutil
 import subprocess
 
+from oslo_serialization import jsonutils
+
 from yardstick.common import utils
-from yardstick.network_services.vnf_generic.vnf.sample_vnf import SampleVNFTrafficGen
-from yardstick.network_services.vnf_generic.vnf.sample_vnf import ClientResourceHelper
+from yardstick.network_services.vnf_generic.vnf import sample_vnf
+
 
 
 LOG = logging.getLogger(__name__)
@@ -43,7 +45,8 @@ IXLOAD_CONFIG_TEMPLATE = '''\
     },
     "remote_server": "%s",
     "result_dir": "%s",
-    "ixload_cfg": "C:/Results/%s"
+    "ixload_cfg": "C:/Results/%s",
+    "links_param": %s
 }'''
 
 IXLOAD_CMD = "{ixloadpy} {http_ixload} {args}"
@@ -59,7 +62,7 @@ class ResourceDataHelper(list):
         }
 
 
-class IxLoadResourceHelper(ClientResourceHelper):
+class IxLoadResourceHelper(sample_vnf.ClientResourceHelper):
 
     RESULTS_MOUNT = "/mnt/Results"
 
@@ -121,7 +124,7 @@ class IxLoadResourceHelper(ClientResourceHelper):
             LOG.debug(self.result[key])
 
 
-class IxLoadTrafficGen(SampleVNFTrafficGen):
+class IxLoadTrafficGen(sample_vnf.SampleVNFTrafficGen):
 
     def __init__(self, name, vnfd, task_id, setup_env_helper_type=None,
                  resource_helper_type=None):
@@ -132,6 +135,21 @@ class IxLoadTrafficGen(SampleVNFTrafficGen):
             name, vnfd, task_id, setup_env_helper_type, resource_helper_type)
         self._result = {}
 
+    def update_gateways(self, links):
+        for name in links:
+            try:
+                gateway = next(intf["virtual-interface"]["dst_ip"] for intf in
+                               self.setup_helper.vnfd_helper["vdu"][0][
+                                   "external-interface"] if
+                               intf["virtual-interface"]["vld_id"] == name)
+
+                links[name]["ip"]["gateway"] = gateway
+            except StopIteration:
+                LOG.debug("Cant find gateway for link %s", name)
+                links[name]["ip"]["gateway"] = "0.0.0.0"
+
+        return links
+
     def run_traffic(self, traffic_profile):
         ports = []
         card = None
@@ -143,11 +161,16 @@ class IxLoadTrafficGen(SampleVNFTrafficGen):
         for csv_file in glob.iglob(self.ssh_helper.join_bin_path('*.csv')):
             os.unlink(csv_file)
 
+        links_param = self.update_gateways(
+            traffic_profile.get_links_param())
+
         ixia_config = self.vnfd_helper.mgmt_interface["tg-config"]
         ixload_config = IXLOAD_CONFIG_TEMPLATE % (
             ixia_config["ixchassis"], ports, card,
             self.vnfd_helper.mgmt_interface["ip"], self.ssh_helper.bin_path,
-            os.path.basename(self.resource_helper.resource_file_name))
+            os.path.basename(self.resource_helper.resource_file_name),
+            jsonutils.dumps(links_param)
+        )
 
         http_ixload_path = os.path.join(VNF_PATH, "../../traffic_profile")
 
diff --git a/yardstick/tests/unit/network_services/traffic_profile/test_http.py b/yardstick/tests/unit/network_services/traffic_profile/test_http.py
index d44fab2b5..874ec37d4 100644
--- a/yardstick/tests/unit/network_services/traffic_profile/test_http.py
+++ b/yardstick/tests/unit/network_services/traffic_profile/test_http.py
@@ -19,13 +19,21 @@ from yardstick.network_services.traffic_profile import http
 
 class TestTrafficProfileGenericHTTP(unittest.TestCase):
 
-    TP_CONFIG = {'traffic_profile': {'duration': 10}}
+    TP_CONFIG = {'traffic_profile': {'duration': 10},
+                 "uplink_0": {}, "downlink_0": {}}
 
     def test___init__(self):
         tp_generic_http = http.TrafficProfileGenericHTTP(
             self.TP_CONFIG)
         self.assertIsNotNone(tp_generic_http)
 
+    def test_get_links_param(self):
+        tp_generic_http = http.TrafficProfileGenericHTTP(
+            self.TP_CONFIG)
+
+        links = tp_generic_http.get_links_param()
+        self.assertEqual({"uplink_0": {}, "downlink_0": {}}, links)
+
     def test_execute(self):
         tp_generic_http = http.TrafficProfileGenericHTTP(
             self.TP_CONFIG)
diff --git a/yardstick/tests/unit/network_services/traffic_profile/test_http_ixload.py b/yardstick/tests/unit/network_services/traffic_profile/test_http_ixload.py
index 57de6602d..1adab48bc 100644
--- a/yardstick/tests/unit/network_services/traffic_profile/test_http_ixload.py
+++ b/yardstick/tests/unit/network_services/traffic_profile/test_http_ixload.py
@@ -17,6 +17,7 @@ import mock
 
 from oslo_serialization import jsonutils
 
+from yardstick.common import exceptions
 from yardstick.network_services.traffic_profile import http_ixload
 from yardstick.network_services.traffic_profile.http_ixload import \
     join_non_strings, validate_non_string_sequence
@@ -45,6 +46,26 @@ class TestJoinNonStrings(unittest.TestCase):
 
 class TestIxLoadTrafficGen(unittest.TestCase):
 
+    def setUp(self):
+        ports = [1, 2, 3]
+        self.test_input = {
+            "remote_server": "REMOTE_SERVER",
+            "ixload_cfg": "IXLOAD_CFG",
+            "result_dir": "RESULT_DIR",
+            "ixia_chassis": "IXIA_CHASSIS",
+            "IXIA": {
+                "card": "CARD",
+                "ports": ports,
+            },
+            'links_param': {
+                "uplink_0": {
+                    "ip": {"address": "address",
+                           "gateway": "gateway",
+                           "subnet_prefix": "subnet_prefix",
+                           "mac": "mac"
+                           }}}
+        }
+
     def test_parse_run_test(self):
         ports = [1, 2, 3]
         test_input = {
@@ -56,6 +77,7 @@ class TestIxLoadTrafficGen(unittest.TestCase):
                 "card": "CARD",
                 "ports": ports,
             },
+            'links_param': {}
         }
         j = jsonutils.dump_as_bytes(test_input)
         ixload = http_ixload.IXLOADHttpTest(j)
@@ -66,6 +88,7 @@ class TestIxLoadTrafficGen(unittest.TestCase):
             ["IXIA_CHASSIS", "CARD", 2],
             ["IXIA_CHASSIS", "CARD", 3],
         ])
+        self.assertEqual({}, ixload.links_param)
 
     def test_format_ports_for_reassignment(self):
         ports = [
@@ -91,6 +114,7 @@ class TestIxLoadTrafficGen(unittest.TestCase):
                 "card": "CARD",
                 "ports": ports,
             },
+            'links_param': {}
         }
         j = jsonutils.dump_as_bytes(test_input)
         ixload = http_ixload.IXLOADHttpTest(j)
@@ -112,6 +136,7 @@ class TestIxLoadTrafficGen(unittest.TestCase):
                 "card": "CARD",
                 "ports": ports,
             },
+            'links_param': {}
         }
         j = jsonutils.dump_as_bytes(test_input)
         ixload = http_ixload.IXLOADHttpTest(j)
@@ -160,6 +185,7 @@ class TestIxLoadTrafficGen(unittest.TestCase):
                 "card": "CARD",
                 "ports": ports,
             },
+            'links_param': {}
         }
         j = jsonutils.dump_as_bytes(test_input)
         ixload = http_ixload.IXLOADHttpTest(j)
@@ -178,6 +204,7 @@ class TestIxLoadTrafficGen(unittest.TestCase):
                 "card": "CARD",
                 "ports": ports,
             },
+            'links_param': {}
         }
         j = jsonutils.dump_as_bytes(test_input)
         ixload = http_ixload.IXLOADHttpTest(j)
@@ -198,6 +225,7 @@ class TestIxLoadTrafficGen(unittest.TestCase):
                 "card": "CARD",
                 "ports": ports,
             },
+            'links_param': {}
         }
 
         j = jsonutils.dump_as_bytes(test_input)
@@ -211,6 +239,105 @@ class TestIxLoadTrafficGen(unittest.TestCase):
         with self.assertRaises(RuntimeError):
             ixload.start_http_test()
 
+    def test_update_config(self):
+        net_taraffic_0 = mock.Mock()
+        net_taraffic_0.name = "HTTP client@uplink_0"
+        net_taraffic_1 = mock.Mock()
+        net_taraffic_1.name = "HTTP client@uplink_1"
+
+        community_list = [net_taraffic_0, net_taraffic_1, Exception]
+        ixload = http_ixload.IXLOADHttpTest(
+            jsonutils.dump_as_bytes(self.test_input))
+
+        ixload.links_param = {"uplink_0": {"ip": {}}}
+
+        ixload.test = mock.Mock()
+        ixload.test.communityList = community_list
+
+        ixload.update_network_param = mock.Mock()
+
+        ixload.update_config()
+
+        ixload.update_network_param.assert_called_once_with(net_taraffic_0, {})
+
+    def test_update_network_mac_address(self):
+        ethernet = mock.MagicMock()
+        net_traffic = mock.Mock()
+        net_traffic.network.getL1Plugin.return_value = ethernet
+
+        ixload = http_ixload.IXLOADHttpTest(
+            jsonutils.dump_as_bytes(self.test_input))
+
+        ix_net_l2_ethernet_plugin = ethernet.childrenList[0]
+        ix_net_ip_v4_v6_plugin = ix_net_l2_ethernet_plugin.childrenList[0]
+        ix_net_ip_v4_v6_range = ix_net_ip_v4_v6_plugin.rangeList[0]
+
+        ixload.update_network_mac_address(net_traffic, "auto")
+        ix_net_ip_v4_v6_range.config.assert_called_once_with(
+            autoMacGeneration=True)
+
+        ixload.update_network_mac_address(net_traffic, "00:00:00:00:00:01")
+        ix_net_ip_v4_v6_range.config.assert_called_with(
+            autoMacGeneration=False)
+        mac_range = ix_net_ip_v4_v6_range.getLowerRelatedRange("MacRange")
+        mac_range.config.assert_called_once_with(mac="00:00:00:00:00:01")
+
+        net_traffic.network.getL1Plugin.return_value = Exception
+
+        with self.assertRaises(exceptions.InvalidRxfFile):
+            ixload.update_network_mac_address(net_traffic, "auto")
+
+    def test_update_network_address(self):
+        ethernet = mock.MagicMock()
+        net_traffic = mock.Mock()
+        net_traffic.network.getL1Plugin.return_value = ethernet
+
+        ixload = http_ixload.IXLOADHttpTest(
+            jsonutils.dump_as_bytes(self.test_input))
+
+        ix_net_l2_ethernet_plugin = ethernet.childrenList[0]
+        ix_net_ip_v4_v6_plugin = ix_net_l2_ethernet_plugin.childrenList[0]
+        ix_net_ip_v4_v6_range = ix_net_ip_v4_v6_plugin.rangeList[0]
+
+        ixload.update_network_address(net_traffic, "address", "gateway",
+                                      "prefix")
+        ix_net_ip_v4_v6_range.config.assert_called_once_with(
+            prefix="prefix",
+            ipAddress="address",
+            gatewayAddress="gateway")
+
+        net_traffic.network.getL1Plugin.return_value = Exception
+
+        with self.assertRaises(exceptions.InvalidRxfFile):
+            ixload.update_network_address(net_traffic, "address", "gateway",
+                                          "prefix")
+
+    def test_update_network_param(self):
+        net_traffic = mock.Mock()
+
+        ixload = http_ixload.IXLOADHttpTest(
+            jsonutils.dump_as_bytes(self.test_input))
+
+        ixload.update_network_address = mock.Mock()
+        ixload.update_network_mac_address = mock.Mock()
+
+        param = {"address": "address",
+                 "gateway": "gateway",
+                 "subnet_prefix": "subnet_prefix",
+                 "mac": "mac"
+                 }
+
+        ixload.update_network_param(net_traffic, param)
+
+        ixload.update_network_address.assert_called_once_with(net_traffic,
+                                                              "address",
+                                                              "gateway",
+                                                              "subnet_prefix")
+
+        ixload.update_network_mac_address.assert_called_once_with(
+            net_traffic,
+            "mac")
+
     @mock.patch('yardstick.network_services.traffic_profile.http_ixload.IxLoad')
     @mock.patch('yardstick.network_services.traffic_profile.http_ixload.StatCollectorUtils')
     def test_start_http_test(self, *args):
@@ -224,6 +351,7 @@ class TestIxLoadTrafficGen(unittest.TestCase):
                 "card": "CARD",
                 "ports": ports,
             },
+            'links_param': {}
         }
 
         j = jsonutils.dump_as_bytes(test_input)
@@ -248,6 +376,7 @@ class TestIxLoadTrafficGen(unittest.TestCase):
                 "card": "CARD",
                 "ports": ports,
             },
+            'links_param': {}
         }
 
         j = jsonutils.dump_as_bytes(test_input)
diff --git a/yardstick/tests/unit/network_services/vnf_generic/vnf/test_tg_ixload.py b/yardstick/tests/unit/network_services/vnf_generic/vnf/test_tg_ixload.py
index 53474b96e..e7f6206eb 100644
--- a/yardstick/tests/unit/network_services/vnf_generic/vnf/test_tg_ixload.py
+++ b/yardstick/tests/unit/network_services/vnf_generic/vnf/test_tg_ixload.py
@@ -57,6 +57,7 @@ class TestIxLoadTrafficGen(ut_base.BaseUnitTestCase):
                  'external-interface':
                  [{'virtual-interface':
                    {'dst_mac': '00:00:00:00:00:04',
+                    'vld_id': 'uplink_0',
                     'vpci': '0000:05:00.0',
                     'local_ip': '152.16.100.19',
                     'type': 'PCI-PASSTHROUGH',
@@ -71,6 +72,7 @@ class TestIxLoadTrafficGen(ut_base.BaseUnitTestCase):
                    'name': 'xe0'},
                   {'virtual-interface':
                    {'dst_mac': '00:00:00:00:00:03',
+                    'vld_id': 'downlink_0',
                     'vpci': '0000:05:00.1',
                     'local_ip': '152.16.40.19',
                     'type': 'PCI-PASSTHROUGH',
@@ -129,6 +131,17 @@ class TestIxLoadTrafficGen(ut_base.BaseUnitTestCase):
         ixload_traffic_gen = tg_ixload.IxLoadTrafficGen(NAME, vnfd, 'task_id')
         self.assertIsNone(ixload_traffic_gen.resource_helper.data)
 
+    def test_update_gateways(self):
+        vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
+        ixload_traffic_gen = tg_ixload.IxLoadTrafficGen(NAME, vnfd, 'task_id')
+        links = {'uplink_0': {'ip': {}},
+                 'downlink_1': {'ip': {}}}
+
+        ixload_traffic_gen.update_gateways(links)
+
+        self.assertEqual("152.16.100.20", links["uplink_0"]["ip"]["gateway"])
+        self.assertEqual("0.0.0.0", links["downlink_1"]["ip"]["gateway"])
+
     @mock.patch.object(ctx_base.Context, 'get_physical_node_from_server',
                        return_value='mock_node')
     def test_collect_kpi(self, *args):
@@ -189,6 +202,8 @@ class TestIxLoadTrafficGen(ut_base.BaseUnitTestCase):
     def test_run_traffic(self, *args):
         mock_traffic_profile = mock.Mock(autospec=tp_base.TrafficProfile)
         mock_traffic_profile.get_traffic_definition.return_value = '64'
+        mock_traffic_profile.get_links_param.return_value = {
+            'uplink_0': {'ip': {}}}
         mock_traffic_profile.params = self.TRAFFIC_PROFILE
         vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
         vnfd['mgmt-interface'].update({'tg-config': {}})
@@ -208,6 +223,8 @@ class TestIxLoadTrafficGen(ut_base.BaseUnitTestCase):
     def test_run_traffic_csv(self, *args):
         mock_traffic_profile = mock.Mock(autospec=tp_base.TrafficProfile)
         mock_traffic_profile.get_traffic_definition.return_value = '64'
+        mock_traffic_profile.get_links_param.return_value = {
+            'uplink_0': {'ip': {}}}
         mock_traffic_profile.params = self.TRAFFIC_PROFILE
         vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
         vnfd['mgmt-interface'].update({'tg-config': {}})