tests/unit/benchmark/scenarios/networking/test_vnf_generic.py

   1 #!/usr/bin/env python
   2
   3 # Copyright (c) 2016-2017 Intel Corporation
   4 #
   5 # Licensed under the Apache License, Version 2.0 (the "License");
   6 # you may not use this file except in compliance with the License.
   7 # You may obtain a copy of the License at
   8 #
   9 #      http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 # Unless required by applicable law or agreed to in writing, software
  12 # distributed under the License is distributed on an "AS IS" BASIS,
  13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 # See the License for the specific language governing permissions and
  15 # limitations under the License.
  16 #
  17
  18 # Unittest for yardstick.benchmark.scenarios.networking.test_vnf_generic
  19
  20 from __future__ import absolute_import
  21
  22 import os
  23 import errno
  24 import unittest
  25 import mock
  26
  27 from yardstick.benchmark.scenarios.networking.vnf_generic import \
  28     SshManager, NetworkServiceTestCase, IncorrectConfig, \
  29     IncorrectSetup, open_relative_file
  30 from yardstick.network_services.collector.subscriber import Collector
  31 from yardstick.network_services.vnf_generic.vnf.base import \
  32     GenericTrafficGen, GenericVNF
  33
  34 STL_MOCKS = {
  35     'stl': mock.MagicMock(),
  36     'stl.trex_stl_lib': mock.MagicMock(),
  37     'stl.trex_stl_lib.base64': mock.MagicMock(),
  38     'stl.trex_stl_lib.binascii': mock.MagicMock(),
  39     'stl.trex_stl_lib.collections': mock.MagicMock(),
  40     'stl.trex_stl_lib.copy': mock.MagicMock(),
  41     'stl.trex_stl_lib.datetime': mock.MagicMock(),
  42     'stl.trex_stl_lib.functools': mock.MagicMock(),
  43     'stl.trex_stl_lib.imp': mock.MagicMock(),
  44     'stl.trex_stl_lib.inspect': mock.MagicMock(),
  45     'stl.trex_stl_lib.json': mock.MagicMock(),
  46     'stl.trex_stl_lib.linecache': mock.MagicMock(),
  47     'stl.trex_stl_lib.math': mock.MagicMock(),
  48     'stl.trex_stl_lib.os': mock.MagicMock(),
  49     'stl.trex_stl_lib.platform': mock.MagicMock(),
  50     'stl.trex_stl_lib.pprint': mock.MagicMock(),
  51     'stl.trex_stl_lib.random': mock.MagicMock(),
  52     'stl.trex_stl_lib.re': mock.MagicMock(),
  53     'stl.trex_stl_lib.scapy': mock.MagicMock(),
  54     'stl.trex_stl_lib.socket': mock.MagicMock(),
  55     'stl.trex_stl_lib.string': mock.MagicMock(),
  56     'stl.trex_stl_lib.struct': mock.MagicMock(),
  57     'stl.trex_stl_lib.sys': mock.MagicMock(),
  58     'stl.trex_stl_lib.threading': mock.MagicMock(),
  59     'stl.trex_stl_lib.time': mock.MagicMock(),
  60     'stl.trex_stl_lib.traceback': mock.MagicMock(),
  61     'stl.trex_stl_lib.trex_stl_async_client': mock.MagicMock(),
  62     'stl.trex_stl_lib.trex_stl_client': mock.MagicMock(),
  63     'stl.trex_stl_lib.trex_stl_exceptions': mock.MagicMock(),
  64     'stl.trex_stl_lib.trex_stl_ext': mock.MagicMock(),
  65     'stl.trex_stl_lib.trex_stl_jsonrpc_client': mock.MagicMock(),
  66     'stl.trex_stl_lib.trex_stl_packet_builder_interface': mock.MagicMock(),
  67     'stl.trex_stl_lib.trex_stl_packet_builder_scapy': mock.MagicMock(),
  68     'stl.trex_stl_lib.trex_stl_port': mock.MagicMock(),
  69     'stl.trex_stl_lib.trex_stl_stats': mock.MagicMock(),
  70     'stl.trex_stl_lib.trex_stl_streams': mock.MagicMock(),
  71     'stl.trex_stl_lib.trex_stl_types': mock.MagicMock(),
  72     'stl.trex_stl_lib.types': mock.MagicMock(),
  73     'stl.trex_stl_lib.utils': mock.MagicMock(),
  74     'stl.trex_stl_lib.utils.argparse': mock.MagicMock(),
  75     'stl.trex_stl_lib.utils.collections': mock.MagicMock(),
  76     'stl.trex_stl_lib.utils.common': mock.MagicMock(),
  77     'stl.trex_stl_lib.utils.json': mock.MagicMock(),
  78     'stl.trex_stl_lib.utils.os': mock.MagicMock(),
  79     'stl.trex_stl_lib.utils.parsing_opts': mock.MagicMock(),
  80     'stl.trex_stl_lib.utils.pwd': mock.MagicMock(),
  81     'stl.trex_stl_lib.utils.random': mock.MagicMock(),
  82     'stl.trex_stl_lib.utils.re': mock.MagicMock(),
  83     'stl.trex_stl_lib.utils.string': mock.MagicMock(),
  84     'stl.trex_stl_lib.utils.sys': mock.MagicMock(),
  85     'stl.trex_stl_lib.utils.text_opts': mock.MagicMock(),
  86     'stl.trex_stl_lib.utils.text_tables': mock.MagicMock(),
  87     'stl.trex_stl_lib.utils.texttable': mock.MagicMock(),
  88     'stl.trex_stl_lib.warnings': mock.MagicMock(),
  89     'stl.trex_stl_lib.yaml': mock.MagicMock(),
  90     'stl.trex_stl_lib.zlib': mock.MagicMock(),
  91     'stl.trex_stl_lib.zmq': mock.MagicMock(),
  92 }
  93
  94 COMPLETE_TREX_VNFD = {
  95     'vnfd:vnfd-catalog': {
  96         'vnfd': [
  97             {
  98                 'benchmark': {
  99                     'kpi': [
 100                         'rx_throughput_fps',
 101                         'tx_throughput_fps',
 102                         'tx_throughput_mbps',
 103                         'rx_throughput_mbps',
 104                         'tx_throughput_pc_linerate',
 105                         'rx_throughput_pc_linerate',
 106                         'min_latency',
 107                         'max_latency',
 108                         'avg_latency',
 109                     ],
 110                 },
 111                 'connection-point': [
 112                     {
 113                         'name': 'xe0',
 114                         'type': 'VPORT',
 115                     },
 116                     {
 117                         'name': 'xe1',
 118                         'type': 'VPORT',
 119                     },
 120                 ],
 121                 'description': 'TRex stateless traffic generator for RFC2544',
 122                 'id': 'TrexTrafficGen',
 123                 'mgmt-interface': {
 124                     'ip': '1.1.1.1',
 125                     'password': 'berta',
 126                     'user': 'berta',
 127                     'vdu-id': 'trexgen-baremetal',
 128                 },
 129                 'name': 'trexgen',
 130                 'short-name': 'trexgen',
 131                 'class-name': 'TrexTrafficGen',
 132                 'vdu': [
 133                     {
 134                         'description': 'TRex stateless traffic generator for RFC2544',
 135                         'external-interface': [
 136                             {
 137                                 'name': 'xe0',
 138                                 'virtual-interface': {
 139                                     'bandwidth': '10 Gbps',
 140                                     'dst_ip': '1.1.1.1',
 141                                     'dst_mac': '00:01:02:03:04:05',
 142                                     'local_ip': '1.1.1.2',
 143                                     'local_mac': '00:01:02:03:05:05',
 144                                     'type': 'PCI-PASSTHROUGH',
 145                                     'netmask': "255.255.255.0",
 146                                     'driver': 'i40',
 147                                     'vpci': '0000:00:10.2',
 148                                 },
 149                                 'vnfd-connection-point-ref': 'xe0',
 150                             },
 151                             {
 152                                 'name': 'xe1',
 153                                 'virtual-interface': {
 154                                     'bandwidth': '10 Gbps',
 155                                     'dst_ip': '2.1.1.1',
 156                                     'dst_mac': '00:01:02:03:04:06',
 157                                     'local_ip': '2.1.1.2',
 158                                     'local_mac': '00:01:02:03:05:06',
 159                                     'type': 'PCI-PASSTHROUGH',
 160                                     'netmask': "255.255.255.0",
 161                                     'driver': 'i40',
 162                                     'vpci': '0000:00:10.1',
 163                                 },
 164                                 'vnfd-connection-point-ref': 'xe1',
 165                             },
 166                         ],
 167                         'id': 'trexgen-baremetal',
 168                         'name': 'trexgen-baremetal',
 169                     },
 170                 ],
 171             },
 172         ],
 173     },
 174 }
 175
 176 IP_ADDR_SHOW = """
 177 28: eth1: <BROADCAST,MULTICAST,UP,LOWER_UP> mtu 1500 qdisc mq state UP \
 178 group default qlen 1000
 179     link/ether 90:e2:ba:a7:6a:c8 brd ff:ff:ff:ff:ff:ff
 180     inet 1.1.1.1/8 brd 1.255.255.255 scope global eth1
 181     inet6 fe80::92e2:baff:fea7:6ac8/64 scope link
 182        valid_lft forever preferred_lft forever
 183 29: eth5: <BROADCAST,MULTICAST,UP,LOWER_UP> mtu 1500 qdisc mq state UP \
 184 group default qlen 1000
 185     link/ether 90:e2:ba:a7:6a:c9 brd ff:ff:ff:ff:ff:ff
 186     inet 2.1.1.1/8 brd 2.255.255.255 scope global eth5
 187     inet6 fe80::92e2:baff:fea7:6ac9/64 scope link tentative
 188        valid_lft forever preferred_lft forever
 189 """
 190
 191 SYS_CLASS_NET = """
 192 lrwxrwxrwx 1 root root 0 sie 10 14:16 eth1 -> \
 193 ../../devices/pci0000:80/0000:80:02.2/0000:84:00.1/net/eth1
 194 lrwxrwxrwx 1 root root 0 sie  3 10:37 eth2 -> \
 195 ../../devices/pci0000:00/0000:00:01.1/0000:84:00.2/net/eth5
 196 """
 197
 198 TRAFFIC_PROFILE = {
 199     "schema": "isb:traffic_profile:0.1",
 200     "name": "fixed",
 201     "description": "Fixed traffic profile to run UDP traffic",
 202     "traffic_profile": {
 203         "traffic_type": "FixedTraffic",
 204         "frame_rate": 100,  # pps
 205         "flow_number": 10,
 206         "frame_size": 64,
 207     },
 208 }
 209
 210
 211 class TestNetworkServiceTestCase(unittest.TestCase):
 212     def setUp(self):
 213         self.trexgen__1 = {
 214             'name': 'trafficgen_1.yardstick',
 215             'ip': '10.10.10.11',
 216             'role': 'TrafficGen',
 217             'user': 'root',
 218             'password': 'r00t',
 219             'interfaces': {
 220                 'xe0': {
 221                     'netmask': '255.255.255.0',
 222                     'local_ip': '152.16.100.20',
 223                     'local_mac': '00:00:00:00:00:01',
 224                     'driver': 'i40e',
 225                     'vpci': '0000:07:00.0',
 226                     'dpdk_port_num': 0,
 227                 },
 228                 'xe1': {
 229                     'netmask': '255.255.255.0',
 230                     'local_ip': '152.16.40.20',
 231                     'local_mac': '00:00:00:00:00:02',
 232                     'driver': 'i40e',
 233                     'vpci': '0000:07:00.1',
 234                     'dpdk_port_num': 1,
 235                 },
 236             },
 237         }
 238
 239         self.trexvnf__1 = {
 240             'name': 'vnf.yardstick',
 241             'ip': '10.10.10.12',
 242             'host': '10.223.197.164',
 243             'role': 'vnf',
 244             'user': 'root',
 245             'password': 'r00t',
 246             'interfaces': {
 247                 'xe0': {
 248                     'netmask': '255.255.255.0',
 249                     'local_ip': '152.16.100.19',
 250                     'local_mac': '00:00:00:00:00:03',
 251                     'driver': 'i40e',
 252                     'vpci': '0000:07:00.0',
 253                     'dpdk_port_num': 0,
 254                 },
 255                 'xe1': {
 256                     'netmask': '255.255.255.0',
 257                     'local_ip': '152.16.40.19',
 258                     'local_mac': '00:00:00:00:00:04',
 259                     'driver': 'i40e',
 260                     'vpci': '0000:07:00.1',
 261                     'dpdk_port_num': 1,
 262                 },
 263             },
 264             'routing_table': [
 265                 {
 266                     'netmask': '255.255.255.0',
 267                     'gateway': '152.16.100.20',
 268                     'network': '152.16.100.20',
 269                     'if': 'xe0',
 270                 },
 271                 {
 272                     'netmask': '255.255.255.0',
 273                     'gateway': '152.16.40.20',
 274                     'network': '152.16.40.20',
 275                     'if': 'xe1',
 276                 },
 277             ],
 278             'nd_route_tbl': [
 279                 {
 280                     'netmask': '112',
 281                     'gateway': '0064:ff9b:0:0:0:0:9810:6414',
 282                     'network': '0064:ff9b:0:0:0:0:9810:6414',
 283                     'if': 'xe0',
 284                 },
 285                 {
 286                     'netmask': '112',
 287                     'gateway': '0064:ff9b:0:0:0:0:9810:2814',
 288                     'network': '0064:ff9b:0:0:0:0:9810:2814',
 289                     'if': 'xe1',
 290                 },
 291             ],
 292         }
 293
 294         self.context_cfg = {
 295             'nodes': {
 296                 'trexgen__1': self.trexgen__1,
 297                 'trexvnf__1': self.trexvnf__1,
 298             },
 299             'networks': {
 300                 'private': {
 301                     'vld_id': 'private',
 302                 },
 303                 'public': {
 304                     'vld_id': 'public',
 305                 },
 306             },
 307         }
 308
 309         self.vld0 = {
 310             'vnfd-connection-point-ref': [
 311                 {
 312                     'vnfd-connection-point-ref': 'xe0',
 313                     'member-vnf-index-ref': '1',
 314                     'vnfd-id-ref': 'trexgen'
 315                 },
 316                 {
 317                     'vnfd-connection-point-ref': 'xe0',
 318                     'member-vnf-index-ref': '2',
 319                     'vnfd-id-ref': 'trexgen'
 320                 }
 321             ],
 322             'type': 'ELAN',
 323             'id': 'private',
 324             'name': 'trexgen__1 to trexvnf__1 link 1'
 325         }
 326
 327         self.vld1 = {
 328             'vnfd-connection-point-ref': [
 329                 {
 330                     'vnfd-connection-point-ref': 'xe1',
 331                     'member-vnf-index-ref': '1',
 332                     'vnfd-id-ref': 'trexgen'
 333                 },
 334                 {
 335                     'vnfd-connection-point-ref': 'xe1',
 336                     'member-vnf-index-ref': '2',
 337                     'vnfd-id-ref': 'trexgen'
 338                 }
 339             ],
 340             'type': 'ELAN',
 341             'id': 'public',
 342             'name': 'trexvnf__1 to trexgen__1 link 2'
 343         }
 344
 345         self.topology = {
 346             'id': 'trex-tg-topology',
 347             'short-name': 'trex-tg-topology',
 348             'name': 'trex-tg-topology',
 349             'description': 'trex-tg-topology',
 350             'constituent-vnfd': [
 351                 {
 352                     'member-vnf-index': '1',
 353                     'VNF model': 'tg_trex_tpl.yaml',
 354                     'vnfd-id-ref': 'trexgen__1',
 355                 },
 356                 {
 357                     'member-vnf-index': '2',
 358                     'VNF model': 'tg_trex_tpl.yaml',
 359                     'vnfd-id-ref': 'trexvnf__1',
 360                 },
 361             ],
 362             'vld': [self.vld0, self.vld1],
 363         }
 364
 365         self.scenario_cfg = {
 366             'task_path': "",
 367             "topology": self._get_file_abspath("vpe_vnf_topology.yaml"),
 368             'task_id': 'a70bdf4a-8e67-47a3-9dc1-273c14506eb7',
 369             'tc': 'tc_ipv4_1Mflow_64B_packetsize',
 370             'traffic_profile': 'ipv4_throughput_vpe.yaml',
 371             'type': 'ISB',
 372             'tc_options': {
 373                 'rfc2544': {
 374                     'allowed_drop_rate': '0.8 - 1',
 375                 },
 376             },
 377             'runner': {
 378                 'object': 'NetworkServiceTestCase',
 379                 'interval': 35,
 380                 'output_filename': 'yardstick.out',
 381                 'runner_id': 74476,
 382                 'duration': 400,
 383                 'type': 'Duration',
 384             },
 385             'traffic_options': {
 386                 'flow': 'ipv4_1flow_Packets_vpe.yaml',
 387                 'imix': 'imix_voice.yaml'
 388             },
 389             'nodes': {
 390                 'tg__2': 'trafficgen_2.yardstick',
 391                 'tg__1': 'trafficgen_1.yardstick',
 392                 'vnf__1': 'vnf.yardstick',
 393             },
 394         }
 395
 396         self.s = NetworkServiceTestCase(self.scenario_cfg, self.context_cfg)
 397
 398     def _get_file_abspath(self, filename):
 399         curr_path = os.path.dirname(os.path.abspath(__file__))
 400         file_path = os.path.join(curr_path, filename)
 401         return file_path
 402
 403     def test_ssh_manager(self):
 404         with mock.patch("yardstick.ssh.SSH") as ssh:
 405             ssh_mock = mock.Mock(autospec=ssh.SSH)
 406             ssh_mock.execute = \
 407                 mock.Mock(return_value=(0, SYS_CLASS_NET + IP_ADDR_SHOW, ""))
 408             ssh.from_node.return_value = ssh_mock
 409             for node, node_dict in self.context_cfg["nodes"].items():
 410                 with SshManager(node_dict) as conn:
 411                     self.assertIsNotNone(conn)
 412
 413     def test___init__(self):
 414         assert self.topology
 415
 416     def test___get_traffic_flow(self):
 417         self.scenario_cfg["traffic_options"]["flow"] = \
 418             self._get_file_abspath("ipv4_1flow_Packets_vpe.yaml")
 419         result = {'flow': {'dstip4_range': '152.40.0.20',
 420                            'srcip4_range': '152.16.0.20', 'count': 1}}
 421         self.assertEqual(result, self.s._get_traffic_flow(self.scenario_cfg))
 422
 423     def test___get_traffic_flow_error(self):
 424         self.scenario_cfg["traffic_options"]["flow"] = \
 425             "ipv4_1flow_Packets_vpe.yaml1"
 426         self.assertEqual({}, self.s._get_traffic_flow(self.scenario_cfg))
 427
 428     def test_get_vnf_imp(self):
 429         vnfd = COMPLETE_TREX_VNFD['vnfd:vnfd-catalog']['vnfd'][0]['class-name']
 430         with mock.patch.dict("sys.modules", STL_MOCKS):
 431             self.assertIsNotNone(self.s.get_vnf_impl(vnfd))
 432
 433             with self.assertRaises(IncorrectConfig) as raised:
 434                 self.s.get_vnf_impl('NonExistentClass')
 435
 436             exc_str = str(raised.exception)
 437             print(exc_str)
 438             self.assertIn('No implementation', exc_str)
 439             self.assertIn('found in', exc_str)
 440
 441     def test_load_vnf_models_invalid(self):
 442         self.context_cfg["nodes"]['trexgen__1']['VNF model'] = \
 443             self._get_file_abspath("tg_trex_tpl.yaml")
 444         self.context_cfg["nodes"]['trexvnf__1']['VNF model'] = \
 445             self._get_file_abspath("tg_trex_tpl.yaml")
 446
 447         vnf = mock.Mock(autospec=GenericVNF)
 448         self.s.get_vnf_impl = mock.Mock(return_value=vnf)
 449
 450         self.assertIsNotNone(
 451             self.s.load_vnf_models(self.scenario_cfg, self.context_cfg))
 452
 453     def test_map_topology_to_infrastructure(self):
 454         with mock.patch("yardstick.ssh.SSH") as ssh:
 455             ssh_mock = mock.Mock(autospec=ssh.SSH)
 456             ssh_mock.execute = \
 457                 mock.Mock(return_value=(0, SYS_CLASS_NET + IP_ADDR_SHOW, ""))
 458             ssh.from_node.return_value = ssh_mock
 459             self.s.map_topology_to_infrastructure(self.context_cfg,
 460                                                   self.topology)
 461
 462         nodes = self.context_cfg["nodes"]
 463         self.assertEqual("tg_trex_tpl.yaml", nodes['trexgen__1']['VNF model'])
 464         self.assertEqual("tg_trex_tpl.yaml", nodes['trexvnf__1']['VNF model'])
 465
 466     def test_map_topology_to_infrastructure_insufficient_nodes(self):
 467         del self.context_cfg['nodes']['trexvnf__1']
 468         with mock.patch("yardstick.ssh.SSH") as ssh:
 469             ssh_mock = mock.Mock(autospec=ssh.SSH)
 470             ssh_mock.execute = \
 471                 mock.Mock(return_value=(1, SYS_CLASS_NET + IP_ADDR_SHOW, ""))
 472             ssh.from_node.return_value = ssh_mock
 473
 474             with self.assertRaises(IncorrectSetup):
 475                 self.s.map_topology_to_infrastructure(self.context_cfg, self.topology)
 476
 477     def test_map_topology_to_infrastructure_config_invalid(self):
 478         cfg = dict(self.context_cfg)
 479         del cfg['nodes']['trexvnf__1']['interfaces']['xe0']['local_mac']
 480         with mock.patch("yardstick.ssh.SSH") as ssh:
 481             ssh_mock = mock.Mock(autospec=ssh.SSH)
 482             ssh_mock.execute = \
 483                 mock.Mock(return_value=(0, SYS_CLASS_NET + IP_ADDR_SHOW, ""))
 484             ssh.from_node.return_value = ssh_mock
 485
 486             with self.assertRaises(IncorrectConfig):
 487                 self.s.map_topology_to_infrastructure(self.context_cfg, self.topology)
 488
 489     def test__resolve_topology_invalid_config(self):
 490         with mock.patch("yardstick.ssh.SSH") as ssh:
 491             ssh_mock = mock.Mock(autospec=ssh.SSH)
 492             ssh_mock.execute = \
 493                 mock.Mock(return_value=(0, SYS_CLASS_NET + IP_ADDR_SHOW, ""))
 494             ssh.from_node.return_value = ssh_mock
 495
 496             # purge an important key from the data structure
 497             for interface in self.trexgen__1['interfaces'].values():
 498                 del interface['local_mac']
 499
 500             with self.assertRaises(IncorrectConfig) as raised:
 501                 self.s._resolve_topology(self.context_cfg, self.topology)
 502
 503             self.assertIn('not found', str(raised.exception))
 504
 505             # make a connection point ref with 3 points
 506             self.vld0['vnfd-connection-point-ref'].append(
 507                 self.vld0['vnfd-connection-point-ref'][0])
 508
 509             with self.assertRaises(IncorrectConfig) as raised:
 510                 self.s._resolve_topology(self.context_cfg, self.topology)
 511
 512             self.assertIn('wrong number of endpoints', str(raised.exception))
 513
 514             # make a connection point ref with 1 point
 515             self.vld0['vnfd-connection-point-ref'] = \
 516                 self.vld0['vnfd-connection-point-ref'][:1]
 517
 518             with self.assertRaises(IncorrectConfig) as raised:
 519                 self.s._resolve_topology(self.context_cfg, self.topology)
 520
 521             self.assertIn('wrong number of endpoints', str(raised.exception))
 522
 523     def test_run(self):
 524         tgen = mock.Mock(autospec=GenericTrafficGen)
 525         tgen.traffic_finished = True
 526         verified_dict = {"verified": True}
 527         tgen.verify_traffic = lambda x: verified_dict
 528         tgen.name = "tgen__1"
 529         vnf = mock.Mock(autospec=GenericVNF)
 530         vnf.runs_traffic = False
 531         self.s.vnfs = [tgen, vnf]
 532         self.s.traffic_profile = mock.Mock()
 533         self.s.collector = mock.Mock(autospec=Collector)
 534         self.s.collector.get_kpi = \
 535             mock.Mock(return_value={tgen.name: verified_dict})
 536         result = {}
 537         self.s.run(result)
 538         self.assertDictEqual(result, {tgen.name: verified_dict})
 539
 540     def test_setup(self):
 541         with mock.patch("yardstick.ssh.SSH") as ssh:
 542             ssh_mock = mock.Mock(autospec=ssh.SSH)
 543             ssh_mock.execute = \
 544                 mock.Mock(return_value=(0, SYS_CLASS_NET + IP_ADDR_SHOW, ""))
 545             ssh.from_node.return_value = ssh_mock
 546
 547             tgen = mock.Mock(autospec=GenericTrafficGen)
 548             tgen.traffic_finished = True
 549             verified_dict = {"verified": True}
 550             tgen.verify_traffic = lambda x: verified_dict
 551             tgen.terminate = mock.Mock(return_value=True)
 552             tgen.name = "tgen__1"
 553             vnf = mock.Mock(autospec=GenericVNF)
 554             vnf.runs_traffic = False
 555             vnf.terminate = mock.Mock(return_value=True)
 556             self.s.vnfs = [tgen, vnf]
 557             self.s.traffic_profile = mock.Mock()
 558             self.s.collector = mock.Mock(autospec=Collector)
 559             self.s.collector.get_kpi = \
 560                 mock.Mock(return_value={tgen.name: verified_dict})
 561             self.s.map_topology_to_infrastructure = mock.Mock(return_value=0)
 562             self.s.load_vnf_models = mock.Mock(return_value=self.s.vnfs)
 563             self.s._fill_traffic_profile = \
 564                 mock.Mock(return_value=TRAFFIC_PROFILE)
 565             self.assertEqual(None, self.s.setup())
 566
 567     def test__get_traffic_profile(self):
 568         self.scenario_cfg["traffic_profile"] = \
 569             self._get_file_abspath("ipv4_throughput_vpe.yaml")
 570         self.assertIsNotNone(self.s._get_traffic_profile(self.scenario_cfg,
 571                                                          self.context_cfg))
 572
 573     def test__get_traffic_profile_exception(self):
 574         cfg = dict(self.scenario_cfg)
 575         cfg["traffic_profile"] = ""
 576         with self.assertRaises(IOError):
 577             self.s._get_traffic_profile(cfg, self.context_cfg)
 578
 579     def test___get_traffic_imix_exception(self):
 580         cfg = dict(self.scenario_cfg)
 581         cfg["traffic_options"]["imix"] = ""
 582         self.assertEqual({}, self.s._get_traffic_imix(cfg))
 583
 584     def test__fill_traffic_profile(self):
 585         with mock.patch.dict("sys.modules", STL_MOCKS):
 586             self.scenario_cfg["traffic_profile"] = \
 587                 self._get_file_abspath("ipv4_throughput_vpe.yaml")
 588             self.scenario_cfg["traffic_options"]["flow"] = \
 589                 self._get_file_abspath("ipv4_1flow_Packets_vpe.yaml")
 590             self.scenario_cfg["traffic_options"]["imix"] = \
 591                 self._get_file_abspath("imix_voice.yaml")
 592             self.assertIsNotNone(self.s._fill_traffic_profile(self.scenario_cfg,
 593                                                               self.context_cfg))
 594
 595     def test_teardown(self):
 596         vnf = mock.Mock(autospec=GenericVNF)
 597         vnf.terminate = \
 598             mock.Mock(return_value=True)
 599         self.s.vnfs = [vnf]
 600         self.s.traffic_profile = mock.Mock()
 601         self.s.collector = mock.Mock(autospec=Collector)
 602         self.s.collector.stop = \
 603             mock.Mock(return_value=True)
 604         self.assertIsNone(self.s.teardown())
 605
 606     SAMPLE_NETDEVS = {
 607             'enp11s0': {
 608                 'address': '0a:de:ad:be:ef:f5',
 609                 'device': '0x1533',
 610                 'driver': 'igb',
 611                 'ifindex': '2',
 612                 'interface_name': 'enp11s0',
 613                 'operstate': 'down',
 614                 'pci_bus_id': '0000:0b:00.0',
 615                 'subsystem_device': '0x1533',
 616                 'subsystem_vendor': '0x15d9',
 617                 'vendor': '0x8086'
 618                 },
 619             'lan': {
 620                 'address': '0a:de:ad:be:ef:f4',
 621                 'device': '0x153a',
 622                 'driver': 'e1000e',
 623                 'ifindex': '3',
 624                 'interface_name': 'lan',
 625                 'operstate': 'up',
 626                 'pci_bus_id': '0000:00:19.0',
 627                 'subsystem_device': '0x153a',
 628                 'subsystem_vendor': '0x15d9',
 629                 'vendor': '0x8086'
 630                 }
 631         }
 632     SAMPLE_VM_NETDEVS = {
 633         'eth1': {
 634             'address': 'fa:de:ad:be:ef:5b',
 635             'device': '0x0001',
 636             'driver': 'virtio_net',
 637             'ifindex': '3',
 638             'interface_name': 'eth1',
 639             'operstate': 'down',
 640             'pci_bus_id': '0000:00:04.0',
 641             'vendor': '0x1af4'
 642         }
 643     }
 644
 645     def test_parse_netdev_info(self):
 646         output = """\
 647 /sys/devices/pci0000:00/0000:00:1c.3/0000:0b:00.0/net/enp11s0/ifindex:2
 648 /sys/devices/pci0000:00/0000:00:1c.3/0000:0b:00.0/net/enp11s0/address:0a:de:ad:be:ef:f5
 649 /sys/devices/pci0000:00/0000:00:1c.3/0000:0b:00.0/net/enp11s0/operstate:down
 650 /sys/devices/pci0000:00/0000:00:1c.3/0000:0b:00.0/net/enp11s0/device/vendor:0x8086
 651 /sys/devices/pci0000:00/0000:00:1c.3/0000:0b:00.0/net/enp11s0/device/device:0x1533
 652 /sys/devices/pci0000:00/0000:00:1c.3/0000:0b:00.0/net/enp11s0/device/subsystem_vendor:0x15d9
 653 /sys/devices/pci0000:00/0000:00:1c.3/0000:0b:00.0/net/enp11s0/device/subsystem_device:0x1533
 654 /sys/devices/pci0000:00/0000:00:1c.3/0000:0b:00.0/net/enp11s0/driver:igb
 655 /sys/devices/pci0000:00/0000:00:1c.3/0000:0b:00.0/net/enp11s0/pci_bus_id:0000:0b:00.0
 656 /sys/devices/pci0000:00/0000:00:19.0/net/lan/ifindex:3
 657 /sys/devices/pci0000:00/0000:00:19.0/net/lan/address:0a:de:ad:be:ef:f4
 658 /sys/devices/pci0000:00/0000:00:19.0/net/lan/operstate:up
 659 /sys/devices/pci0000:00/0000:00:19.0/net/lan/device/vendor:0x8086
 660 /sys/devices/pci0000:00/0000:00:19.0/net/lan/device/device:0x153a
 661 /sys/devices/pci0000:00/0000:00:19.0/net/lan/device/subsystem_vendor:0x15d9
 662 /sys/devices/pci0000:00/0000:00:19.0/net/lan/device/subsystem_device:0x153a
 663 /sys/devices/pci0000:00/0000:00:19.0/net/lan/driver:e1000e
 664 /sys/devices/pci0000:00/0000:00:19.0/net/lan/pci_bus_id:0000:00:19.0
 665 """
 666         res = NetworkServiceTestCase.parse_netdev_info(output)
 667         assert res == self.SAMPLE_NETDEVS
 668
 669     def test_parse_netdev_info_virtio(self):
 670         output = """\
 671 /sys/devices/pci0000:00/0000:00:04.0/virtio1/net/eth1/ifindex:3
 672 /sys/devices/pci0000:00/0000:00:04.0/virtio1/net/eth1/address:fa:de:ad:be:ef:5b
 673 /sys/devices/pci0000:00/0000:00:04.0/virtio1/net/eth1/operstate:down
 674 /sys/devices/pci0000:00/0000:00:04.0/virtio1/net/eth1/device/vendor:0x1af4
 675 /sys/devices/pci0000:00/0000:00:04.0/virtio1/net/eth1/device/device:0x0001
 676 /sys/devices/pci0000:00/0000:00:04.0/virtio1/net/eth1/driver:virtio_net
 677 """
 678         res = NetworkServiceTestCase.parse_netdev_info(output)
 679         assert res == self.SAMPLE_VM_NETDEVS
 680
 681     def test_sort_dpdk_port_num(self):
 682         netdevs = self.SAMPLE_NETDEVS.copy()
 683         NetworkServiceTestCase._sort_dpdk_port_num(netdevs)
 684         assert netdevs['lan']['dpdk_port_num'] == 1
 685         assert netdevs['enp11s0']['dpdk_port_num'] == 2
 686
 687     def test_probe_missing_values(self):
 688         netdevs = self.SAMPLE_NETDEVS.copy()
 689         NetworkServiceTestCase._sort_dpdk_port_num(netdevs)
 690         network = {'local_mac': '0a:de:ad:be:ef:f5'}
 691         NetworkServiceTestCase._probe_missing_values(netdevs, network, set())
 692         assert network['dpdk_port_num'] == 2
 693
 694         network = {'local_mac': '0a:de:ad:be:ef:f4'}
 695         NetworkServiceTestCase._probe_missing_values(netdevs, network, set())
 696         assert network['dpdk_port_num'] == 1
 697
 698     def test_open_relative_path(self):
 699         mock_open = mock.mock_open()
 700         mock_open_result = mock_open()
 701         mock_open_call_count = 1  # initial call to get result
 702
 703         module_name = \
 704             'yardstick.benchmark.scenarios.networking.vnf_generic.open'
 705
 706         # test
 707         with mock.patch(module_name, mock_open, create=True):
 708             self.assertEqual(open_relative_file('foo', 'bar'), mock_open_result)
 709
 710             mock_open_call_count += 1  # one more call expected
 711             self.assertEqual(mock_open.call_count, mock_open_call_count)
 712             self.assertIn('foo', mock_open.call_args_list[-1][0][0])
 713             self.assertNotIn('bar', mock_open.call_args_list[-1][0][0])
 714
 715             def open_effect(*args, **kwargs):
 716                 if kwargs.get('name', args[0]) == os.path.join('bar', 'foo'):
 717                     return mock_open_result
 718                 raise IOError(errno.ENOENT, 'not found')
 719
 720             mock_open.side_effect = open_effect
 721             self.assertEqual(open_relative_file('foo', 'bar'), mock_open_result)
 722
 723             mock_open_call_count += 2  # two more calls expected
 724             self.assertEqual(mock_open.call_count, mock_open_call_count)
 725             self.assertIn('foo', mock_open.call_args_list[-1][0][0])
 726             self.assertIn('bar', mock_open.call_args_list[-1][0][0])
 727
 728             # test an IOError of type ENOENT
 729             mock_open.side_effect = IOError(errno.ENOENT, 'not found')
 730             with self.assertRaises(IOError):
 731                 # the second call still raises
 732                 open_relative_file('foo', 'bar')
 733
 734             mock_open_call_count += 2  # two more calls expected
 735             self.assertEqual(mock_open.call_count, mock_open_call_count)
 736             self.assertIn('foo', mock_open.call_args_list[-1][0][0])
 737             self.assertIn('bar', mock_open.call_args_list[-1][0][0])
 738
 739             # test an IOError other than ENOENT
 740             mock_open.side_effect = IOError(errno.EBUSY, 'busy')
 741             with self.assertRaises(IOError):
 742                 open_relative_file('foo', 'bar')
 743
 744             mock_open_call_count += 1  # one more call expected
 745             self.assertEqual(mock_open.call_count, mock_open_call_count)