yardstick/tests/unit/network_services/vnf_generic/vnf/test_tg_prox.py

   1 # Copyright (c) 2017 Intel Corporation
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #      http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14 #
  15
  16 import unittest
  17 import mock
  18
  19 from yardstick.tests.unit.network_services.vnf_generic.vnf.test_base import mock_ssh
  20 from yardstick.tests import STL_MOCKS
  21
  22
  23 SSH_HELPER = 'yardstick.network_services.vnf_generic.vnf.sample_vnf.VnfSshHelper'
  24 NAME = 'vnf__1'
  25
  26 STLClient = mock.MagicMock()
  27 stl_patch = mock.patch.dict("sys.modules", STL_MOCKS)
  28 stl_patch.start()
  29
  30 if stl_patch:
  31     from yardstick.network_services.vnf_generic.vnf.tg_prox import ProxTrafficGen
  32     from yardstick.network_services.traffic_profile.base import TrafficProfile
  33
  34
  35 @mock.patch('yardstick.network_services.vnf_generic.vnf.prox_helpers.time')
  36 class TestProxTrafficGen(unittest.TestCase):
  37     VNFD0 = {
  38         'short-name': 'ProxVnf',
  39         'vdu': [
  40             {
  41                 'routing_table': [
  42                     {
  43                         'network': '152.16.100.20',
  44                         'netmask': '255.255.255.0',
  45                         'gateway': '152.16.100.20',
  46                         'if': 'xe0',
  47                     },
  48                     {
  49                         'network': '152.16.40.20',
  50                         'netmask': '255.255.255.0',
  51                         'gateway': '152.16.40.20',
  52                         'if': 'xe1',
  53                     },
  54                 ],
  55                 'description': 'PROX approximation using DPDK',
  56                 'name': 'proxvnf-baremetal',
  57                 'nd_route_tbl': [
  58                     {
  59                         'network': '0064:ff9b:0:0:0:0:9810:6414',
  60                         'netmask': '112',
  61                         'gateway': '0064:ff9b:0:0:0:0:9810:6414',
  62                         'if': 'xe0',
  63                     },
  64                     {
  65                         'network': '0064:ff9b:0:0:0:0:9810:2814',
  66                         'netmask': '112',
  67                         'gateway': '0064:ff9b:0:0:0:0:9810:2814',
  68                         'if': 'xe1',
  69                     },
  70                 ],
  71                 'id': 'proxvnf-baremetal',
  72                 'external-interface': [
  73                     {
  74                         'virtual-interface': {
  75                             'dst_mac': '00:00:00:00:00:04',
  76                             'vpci': '0000:05:00.0',
  77                             'local_ip': '152.16.100.19',
  78                             'type': 'PCI-PASSTHROUGH',
  79                             'vld_id': '',
  80                             'netmask': '255.255.255.0',
  81                             'dpdk_port_num': 0,
  82                             'bandwidth': '10 Gbps',
  83                             'driver': "i40e",
  84                             'dst_ip': '152.16.100.20',
  85                             'local_iface_name': 'xe0',
  86                             'local_mac': '00:00:00:00:00:02',
  87                         },
  88                         'vnfd-connection-point-ref': 'xe0',
  89                         'name': 'xe0',
  90                     },
  91                     {
  92                         'virtual-interface': {
  93                             'dst_mac': '00:00:00:00:00:03',
  94                             'vpci': '0000:05:00.1',
  95                             'local_ip': '152.16.40.19',
  96                             'type': 'PCI-PASSTHROUGH',
  97                             'vld_id': '',
  98                             'driver': "i40e",
  99                             'netmask': '255.255.255.0',
 100                             'dpdk_port_num': 1,
 101                             'bandwidth': '10 Gbps',
 102                             'dst_ip': '152.16.40.20',
 103                             'local_iface_name': 'xe1',
 104                             'local_mac': '00:00:00:00:00:01',
 105                         },
 106                         'vnfd-connection-point-ref': 'xe1',
 107                         'name': 'xe1',
 108                     },
 109                 ],
 110             },
 111         ],
 112         'description': 'PROX approximation using DPDK',
 113         'mgmt-interface': {
 114             'vdu-id': 'proxvnf-baremetal',
 115             'host': '1.2.1.1',
 116             'password': 'r00t',
 117             'user': 'root',
 118             'ip': '1.2.1.1',
 119         },
 120         'benchmark': {
 121             'kpi': [
 122                 'packets_in',
 123                 'packets_fwd',
 124                 'packets_dropped',
 125             ],
 126         },
 127         'connection-point': [
 128             {
 129                 'type': 'VPORT',
 130                 'name': 'xe0',
 131             },
 132             {
 133                 'type': 'VPORT',
 134                 'name': 'xe1',
 135             },
 136         ],
 137         'id': 'ProxApproxVnf',
 138         'name': 'ProxVnf',
 139     }
 140
 141     VNFD = {
 142         'vnfd:vnfd-catalog': {
 143             'vnfd': [
 144                 VNFD0,
 145             ],
 146         },
 147     }
 148
 149     SCENARIO_CFG = {
 150         'task_path': "",
 151         'nodes': {
 152             'tg__1': 'trafficgen_1.yardstick',
 153             'vnf__1': 'vnf.yardstick'},
 154         'runner': {
 155             'duration': 600, 'type': 'Duration'},
 156         'topology': 'prox-tg-topology-2.yaml',
 157         'traffic_profile': '../../traffic_profiles/prox_binsearch.yaml',
 158         'type': 'NSPerf',
 159         'options': {
 160             'tg__1': {'prox_args': {'-e': '',
 161                                     '-t': ''},
 162                       'prox_config': 'configs/l3-gen-2.cfg',
 163                       'prox_path':
 164                           '/root/dppd-PROX-v035/build/prox'},
 165             'vnf__1': {
 166                 'prox_args': {'-t': ''},
 167                 'prox_config': 'configs/l3-swap-2.cfg',
 168                 'prox_path': '/root/dppd-PROX-v035/build/prox'}}}
 169
 170     CONTEXT_CFG = {
 171         'nodes': {
 172             'tg__2': {
 173                 'member-vnf-index': '3',
 174                 'role': 'TrafficGen',
 175                 'name': 'trafficgen_2.yardstick',
 176                 'vnfd-id-ref': 'tg__2',
 177                 'ip': '1.2.1.1',
 178                 'interfaces': {
 179                     'xe0': {
 180                         'local_iface_name': 'ens513f0',
 181                         'vld_id': ProxTrafficGen.DOWNLINK,
 182                         'netmask': '255.255.255.0',
 183                         'local_ip': '152.16.40.20',
 184                         'dst_mac': '00:00:00:00:00:01',
 185                         'local_mac': '00:00:00:00:00:03',
 186                         'dst_ip': '152.16.40.19',
 187                         'driver': 'ixgbe',
 188                         'vpci': '0000:02:00.0',
 189                         'dpdk_port_num': 0,
 190                     },
 191                     'xe1': {
 192                         'local_iface_name': 'ens513f1',
 193                         'netmask': '255.255.255.0',
 194                         'network': '202.16.100.0',
 195                         'local_ip': '202.16.100.20',
 196                         'local_mac': '00:1e:67:d0:60:5d',
 197                         'driver': 'ixgbe',
 198                         'vpci': '0000:02:00.1',
 199                         'dpdk_port_num': 1,
 200                     },
 201                 },
 202                 'password': 'r00t',
 203                 'VNF model': 'l3fwd_vnf.yaml',
 204                 'user': 'root',
 205             },
 206             'tg__1': {
 207                 'member-vnf-index': '1',
 208                 'role': 'TrafficGen',
 209                 'name': 'trafficgen_1.yardstick',
 210                 'vnfd-id-ref': 'tg__1',
 211                 'ip': '1.2.1.1',
 212                 'interfaces': {
 213                     'xe0': {
 214                         'local_iface_name': 'ens785f0',
 215                         'vld_id': ProxTrafficGen.UPLINK,
 216                         'netmask': '255.255.255.0',
 217                         'local_ip': '152.16.100.20',
 218                         'dst_mac': '00:00:00:00:00:02',
 219                         'local_mac': '00:00:00:00:00:04',
 220                         'dst_ip': '152.16.100.19',
 221                         'driver': 'i40e',
 222                         'vpci': '0000:05:00.0',
 223                         'dpdk_port_num': 0,
 224                     },
 225                     'xe1': {
 226                         'local_iface_name': 'ens785f1',
 227                         'netmask': '255.255.255.0',
 228                         'local_ip': '152.16.100.21',
 229                         'local_mac': '00:00:00:00:00:01',
 230                         'driver': 'i40e',
 231                         'vpci': '0000:05:00.1',
 232                         'dpdk_port_num': 1,
 233                     },
 234                 },
 235                 'password': 'r00t',
 236                 'VNF model': 'tg_rfc2544_tpl.yaml',
 237                 'user': 'root',
 238             },
 239             'vnf__1': {
 240                 'name': 'vnf.yardstick',
 241                 'vnfd-id-ref': 'vnf__1',
 242                 'ip': '1.2.1.1',
 243                 'interfaces': {
 244                     'xe0': {
 245                         'local_iface_name': 'ens786f0',
 246                         'vld_id': ProxTrafficGen.UPLINK,
 247                         'netmask': '255.255.255.0',
 248                         'local_ip': '152.16.100.19',
 249                         'dst_mac': '00:00:00:00:00:04',
 250                         'local_mac': '00:00:00:00:00:02',
 251                         'dst_ip': '152.16.100.20',
 252                         'driver': 'i40e',
 253                         'vpci': '0000:05:00.0',
 254                         'dpdk_port_num': 0,
 255                     },
 256                     'xe1': {
 257                         'local_iface_name': 'ens786f1',
 258                         'vld_id': ProxTrafficGen.DOWNLINK,
 259                         'netmask': '255.255.255.0',
 260                         'local_ip': '152.16.40.19',
 261                         'dst_mac': '00:00:00:00:00:03',
 262                         'local_mac': '00:00:00:00:00:01',
 263                         'dst_ip': '152.16.40.20',
 264                         'driver': 'i40e',
 265                         'vpci': '0000:05:00.1',
 266                         'dpdk_port_num': 1,
 267                     },
 268                 },
 269                 'routing_table': [
 270                     {
 271                         'netmask': '255.255.255.0',
 272                         'gateway': '152.16.100.20',
 273                         'network': '152.16.100.20',
 274                         'if': 'xe0',
 275                     },
 276                     {
 277                         'netmask': '255.255.255.0',
 278                         'gateway': '152.16.40.20',
 279                         'network': '152.16.40.20',
 280                         'if': 'xe1',
 281                     },
 282                 ],
 283                 'member-vnf-index': '2',
 284                 'host': '1.2.1.1',
 285                 'role': 'vnf',
 286                 'user': 'root',
 287                 'nd_route_tbl': [
 288                     {
 289                         'netmask': '112',
 290                         'gateway': '0064:ff9b:0:0:0:0:9810:6414',
 291                         'network': '0064:ff9b:0:0:0:0:9810:6414',
 292                         'if': 'xe0',
 293                     },
 294                     {
 295                         'netmask': '112',
 296                         'gateway': '0064:ff9b:0:0:0:0:9810:2814',
 297                         'network': '0064:ff9b:0:0:0:0:9810:2814',
 298                         'if': 'xe1',
 299                     },
 300                 ],
 301                 'password': 'r00t',
 302                 'VNF model': 'prox_vnf.yaml',
 303             },
 304         },
 305     }
 306
 307     TRAFFIC_PROFILE = {
 308         'description': 'Binary search for max no-drop throughput over given packet sizes',
 309         'name': 'prox_binsearch',
 310         'schema': 'nsb:traffic_profile:0.1',
 311         'traffic_profile': {
 312             'duration': 5,
 313             'lower_bound': 0.0,
 314             'packet_sizes': [64, 65],
 315             'test_precision': 1.0,
 316             'tolerated_loss': 0.0,
 317             'traffic_type': 'ProxBinSearchProfile',
 318             'upper_bound': 100.0}}
 319
 320     @mock.patch(SSH_HELPER)
 321     def test___init__(self, ssh, *args):
 322         mock_ssh(ssh)
 323         prox_traffic_gen = ProxTrafficGen(NAME, self.VNFD0)
 324         self.assertIsNone(prox_traffic_gen._tg_process)
 325         self.assertIsNone(prox_traffic_gen._traffic_process)
 326
 327     @mock.patch(SSH_HELPER)
 328     def test_collect_kpi(self, ssh, *args):
 329         mock_ssh(ssh)
 330
 331         prox_traffic_gen = ProxTrafficGen(NAME, self.VNFD0)
 332         prox_traffic_gen._vnf_wrapper.resource_helper.resource = mock.MagicMock(
 333             **{"self.check_if_system_agent_running.return_value": [False]})
 334         prox_traffic_gen._vnf_wrapper.vnf_execute = mock.Mock(return_value="")
 335         self.assertEqual({}, prox_traffic_gen.collect_kpi())
 336
 337
 338     @mock.patch('yardstick.network_services.vnf_generic.vnf.prox_helpers.find_relative_file')
 339     @mock.patch(
 340         'yardstick.network_services.vnf_generic.vnf.sample_vnf.CpuSysCores')
 341     @mock.patch(SSH_HELPER)
 342     def bad_test_instantiate(self, ssh, mock_cpu_sys_cores, *args):
 343         mock_ssh(ssh)
 344
 345         mock_cpu_sys_cores.get_core_socket.return_value = {'0': '01234'}
 346
 347         mock_traffic_profile = mock.Mock(autospec=TrafficProfile)
 348         mock_traffic_profile.get_traffic_definition.return_value = "64"
 349         mock_traffic_profile.params = self.TRAFFIC_PROFILE
 350
 351         vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
 352         prox_traffic_gen = ProxTrafficGen(NAME, vnfd)
 353         ssh_helper = mock.MagicMock(
 354             **{"execute.return_value": (0, "", ""), "bin_path": ""})
 355         prox_traffic_gen.ssh_helper = ssh_helper
 356         prox_traffic_gen.setup_helper.dpdk_bind_helper.ssh_helper = ssh_helper
 357         prox_traffic_gen.setup_helper._setup_resources = mock.MagicMock()
 358         prox_traffic_gen.setup_hugepages = mock.MagicMock()
 359         prox_traffic_gen.generate_prox_config_file = mock.MagicMock()
 360         prox_traffic_gen.upload_prox_config = mock.MagicMock()
 361         prox_traffic_gen.setup_helper._find_used_drivers = mock.MagicMock()
 362         prox_traffic_gen.setup_helper.used_drivers = {}
 363         prox_traffic_gen.setup_helper.bound_pci = []
 364         prox_traffic_gen._start_server = mock.Mock(return_value=0)
 365         prox_traffic_gen._tg_process = mock.MagicMock()
 366         prox_traffic_gen._tg_process.start = mock.Mock()
 367         prox_traffic_gen._tg_process.exitcode = 0
 368         prox_traffic_gen._tg_process._is_alive = mock.Mock(return_value=1)
 369         prox_traffic_gen.ssh_helper = mock.MagicMock()
 370         prox_traffic_gen.resource_helper.ssh_helper = mock.MagicMock()
 371         scenario_cfg = {
 372             'task_path': '',
 373             'options': {'tg__1': {'prox_args': {'-e': '',
 374                                                 '-t': ''},
 375                                   'prox_config': 'configs/l3-gen-2.cfg',
 376                                   'prox_path': '/root/dppd-PROX-v035/build/prox'},
 377                         'vnf__1': {'prox_args': {'-t': ''},
 378                                    'prox_config': 'configs/l3-swap-2.cfg',
 379                                    'prox_path': '/root/dppd-PROX-v035/build/prox'}
 380                         }
 381         }
 382         prox_traffic_gen.instantiate(scenario_cfg, {})
 383
 384     @mock.patch(SSH_HELPER)
 385     def test__traffic_runner(self, ssh, *args):
 386         mock_ssh(ssh)
 387
 388         mock_traffic_profile = mock.Mock(autospec=TrafficProfile)
 389         mock_traffic_profile.get_traffic_definition.return_value = "64"
 390         mock_traffic_profile.execute_traffic.return_value = "64"
 391         mock_traffic_profile.params = self.TRAFFIC_PROFILE
 392
 393         vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
 394         sut = ProxTrafficGen(NAME, vnfd)
 395         sut._get_socket = mock.MagicMock()
 396         sut.ssh_helper = mock.Mock()
 397         sut.ssh_helper.run = mock.Mock()
 398         sut.setup_helper.prox_config_dict = {}
 399         sut._connect_client = mock.Mock(autospec=STLClient)
 400         sut._connect_client.get_stats = mock.Mock(return_value="0")
 401         sut._traffic_runner(mock_traffic_profile)
 402
 403     @mock.patch('yardstick.network_services.vnf_generic.vnf.prox_helpers.socket')
 404     @mock.patch(SSH_HELPER)
 405     def test_listen_traffic(self, ssh, *args):
 406         mock_ssh(ssh)
 407         vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
 408         prox_traffic_gen = ProxTrafficGen(NAME, vnfd)
 409         self.assertIsNone(prox_traffic_gen.listen_traffic(mock.Mock()))
 410
 411     @mock.patch('yardstick.network_services.vnf_generic.vnf.prox_helpers.socket')
 412     @mock.patch(SSH_HELPER)
 413     def test_terminate(self, ssh, *args):
 414         mock_ssh(ssh)
 415         vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
 416         prox_traffic_gen = ProxTrafficGen(NAME, vnfd)
 417         prox_traffic_gen._terminated = mock.MagicMock()
 418         prox_traffic_gen._traffic_process = mock.MagicMock()
 419         prox_traffic_gen._traffic_process.terminate = mock.Mock()
 420         prox_traffic_gen.ssh_helper = mock.MagicMock()
 421         prox_traffic_gen.setup_helper = mock.MagicMock()
 422         prox_traffic_gen.resource_helper = mock.MagicMock()
 423         prox_traffic_gen._vnf_wrapper.setup_helper = mock.MagicMock()
 424         prox_traffic_gen._vnf_wrapper._vnf_process = mock.MagicMock()
 425         prox_traffic_gen._vnf_wrapper.resource_helper = mock.MagicMock()
 426         self.assertIsNone(prox_traffic_gen.terminate())