tests/unit/network_services/traffic_profile/test_traffic_profile.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 from __future__ import absolute_import
  19 import unittest
  20
  21 import mock
  22
  23 STL_MOCKS = {
  24     'stl': mock.MagicMock(),
  25     'stl.trex_stl_lib': mock.MagicMock(),
  26     'stl.trex_stl_lib.base64': mock.MagicMock(),
  27     'stl.trex_stl_lib.binascii': mock.MagicMock(),
  28     'stl.trex_stl_lib.collections': mock.MagicMock(),
  29     'stl.trex_stl_lib.copy': mock.MagicMock(),
  30     'stl.trex_stl_lib.datetime': mock.MagicMock(),
  31     'stl.trex_stl_lib.functools': mock.MagicMock(),
  32     'stl.trex_stl_lib.imp': mock.MagicMock(),
  33     'stl.trex_stl_lib.inspect': mock.MagicMock(),
  34     'stl.trex_stl_lib.json': mock.MagicMock(),
  35     'stl.trex_stl_lib.linecache': mock.MagicMock(),
  36     'stl.trex_stl_lib.math': mock.MagicMock(),
  37     'stl.trex_stl_lib.os': mock.MagicMock(),
  38     'stl.trex_stl_lib.platform': mock.MagicMock(),
  39     'stl.trex_stl_lib.pprint': mock.MagicMock(),
  40     'stl.trex_stl_lib.random': mock.MagicMock(),
  41     'stl.trex_stl_lib.re': mock.MagicMock(),
  42     'stl.trex_stl_lib.scapy': mock.MagicMock(),
  43     'stl.trex_stl_lib.socket': mock.MagicMock(),
  44     'stl.trex_stl_lib.string': mock.MagicMock(),
  45     'stl.trex_stl_lib.struct': mock.MagicMock(),
  46     'stl.trex_stl_lib.sys': mock.MagicMock(),
  47     'stl.trex_stl_lib.threading': mock.MagicMock(),
  48     'stl.trex_stl_lib.time': mock.MagicMock(),
  49     'stl.trex_stl_lib.traceback': mock.MagicMock(),
  50     'stl.trex_stl_lib.trex_stl_async_client': mock.MagicMock(),
  51     'stl.trex_stl_lib.trex_stl_client': mock.MagicMock(),
  52     'stl.trex_stl_lib.trex_stl_exceptions': mock.MagicMock(),
  53     'stl.trex_stl_lib.trex_stl_ext': mock.MagicMock(),
  54     'stl.trex_stl_lib.trex_stl_jsonrpc_client': mock.MagicMock(),
  55     'stl.trex_stl_lib.trex_stl_packet_builder_interface': mock.MagicMock(),
  56     'stl.trex_stl_lib.trex_stl_packet_builder_scapy': mock.MagicMock(),
  57     'stl.trex_stl_lib.trex_stl_port': mock.MagicMock(),
  58     'stl.trex_stl_lib.trex_stl_stats': mock.MagicMock(),
  59     'stl.trex_stl_lib.trex_stl_streams': mock.MagicMock(),
  60     'stl.trex_stl_lib.trex_stl_types': mock.MagicMock(),
  61     'stl.trex_stl_lib.types': mock.MagicMock(),
  62     'stl.trex_stl_lib.utils': mock.MagicMock(),
  63     'stl.trex_stl_lib.utils.argparse': mock.MagicMock(),
  64     'stl.trex_stl_lib.utils.collections': mock.MagicMock(),
  65     'stl.trex_stl_lib.utils.common': mock.MagicMock(),
  66     'stl.trex_stl_lib.utils.json': mock.MagicMock(),
  67     'stl.trex_stl_lib.utils.os': mock.MagicMock(),
  68     'stl.trex_stl_lib.utils.parsing_opts': mock.MagicMock(),
  69     'stl.trex_stl_lib.utils.pwd': mock.MagicMock(),
  70     'stl.trex_stl_lib.utils.random': mock.MagicMock(),
  71     'stl.trex_stl_lib.utils.re': mock.MagicMock(),
  72     'stl.trex_stl_lib.utils.string': mock.MagicMock(),
  73     'stl.trex_stl_lib.utils.sys': mock.MagicMock(),
  74     'stl.trex_stl_lib.utils.text_opts': mock.MagicMock(),
  75     'stl.trex_stl_lib.utils.text_tables': mock.MagicMock(),
  76     'stl.trex_stl_lib.utils.texttable': mock.MagicMock(),
  77     'stl.trex_stl_lib.warnings': mock.MagicMock(),
  78     'stl.trex_stl_lib.yaml': mock.MagicMock(),
  79     'stl.trex_stl_lib.zlib': mock.MagicMock(),
  80     'stl.trex_stl_lib.zmq': mock.MagicMock(),
  81 }
  82
  83 STLClient = mock.MagicMock()
  84 stl_patch = mock.patch.dict("sys.modules", STL_MOCKS)
  85 stl_patch.start()
  86
  87 if stl_patch:
  88     from yardstick.network_services.traffic_profile.base import TrafficProfile
  89     from yardstick.network_services.traffic_profile.traffic_profile import \
  90         TrexProfile
  91
  92
  93 class TestTrexProfile(unittest.TestCase):
  94     TRAFFIC_PROFILE = {
  95         "schema": "isb:traffic_profile:0.1",
  96         "name": "fixed",
  97         "description": "Fixed traffic profile to run UDP traffic",
  98         "traffic_profile": {
  99             "traffic_type": "FixedTraffic",
 100             "frame_rate": 100,  # pps
 101             "flow_number": 10,
 102             "frame_size": 64}}
 103
 104     PROFILE = {'description': 'Traffic profile to run RFC2544 latency',
 105                'name': 'rfc2544',
 106                'traffic_profile': {'traffic_type': 'RFC2544Profile',
 107                                    'frame_rate': 100},
 108                'public': {'ipv4': {'outer_l2': {'framesize': {'64B': '100',
 109                                                               '1518B': '0',
 110                                                               '128B': '0',
 111                                                               '1400B': '0',
 112                                                               '256B': '0',
 113                                                               '373b': '0',
 114                                                               '570B': '0'},
 115                                                 "srcmac": "00:00:00:00:00:02",
 116                                                 "dstmac": "00:00:00:00:00:01"},
 117                                    'outer_l3v4': {'dstip4': '1.1.1.1-1.1.2.2',
 118                                                   'proto': 'udp',
 119                                                   'srcip4': '9.9.1.1-90.1.2.2',
 120                                                   'dscp': 0, 'ttl': 32,
 121                                                   'count': 1},
 122                                    'outer_l4': {'srcport': '2001',
 123                                                 'dsrport': '1234',
 124                                                 'count': 1}}},
 125                'private': {'ipv4':
 126                            {'outer_l2': {'framesize':
 127                                          {'64B': '100', '1518B': '0',
 128                                           '128B': '0', '1400B': '0',
 129                                           '256B': '0', '373b': '0',
 130                                           '570B': '0'},
 131                                          "srcmac": "00:00:00:00:00:01",
 132                                          "dstmac": "00:00:00:00:00:02"},
 133                             'outer_l3v4': {'dstip4': '9.9.1.1-90.105.255.255',
 134                                            'proto': 'udp',
 135                                            'srcip4': '1.1.1.1-1.15.255.255',
 136                                            'dscp': 0, 'ttl': 32, 'count': 1},
 137                             'outer_l4': {'dstport': '2001',
 138                                          'srcport': '1234',
 139                                          'count': 1}}},
 140                'schema': 'isb:traffic_profile:0.1'}
 141     PROFILE_v6 = {'description': 'Traffic profile to run RFC2544 latency',
 142                   'name': 'rfc2544',
 143                   'traffic_profile': {'traffic_type': 'RFC2544Profile',
 144                                       'frame_rate': 100},
 145                   'public': {'ipv6': {'outer_l2': {'framesize':
 146                                                    {'64B': '100', '1518B': '0',
 147                                                     '128B': '0', '1400B': '0',
 148                                                     '256B': '0', '373b': '0',
 149                                                     '570B': '0'},
 150                                                    "srcmac": "00:00:00:00:00:02",
 151                                                    "dstmac": "00:00:00:00:00:01"},
 152                                       'outer_l3v4': {'dstip6': '0064:ff9b:0:0:0:0:9810:6414-0064:ff9b:0:0:0:0:9810:6420',
 153                                                      'proto': 'udp',
 154                                                      'srcip6': '0064:ff9b:0:0:0:0:9810:2814-0064:ff9b:0:0:0:0:9810:2820',
 155                                                      'dscp': 0, 'ttl': 32,
 156                                                      'count': 1},
 157                                       'outer_l4': {'srcport': '2001',
 158                                                    'dsrport': '1234',
 159                                                    'count': 1}}},
 160                   'private':
 161                   {'ipv6': {'outer_l2': {'framesize':
 162                                          {'64B': '100', '1518B': '0',
 163                                           '128B': '0', '1400B': '0',
 164                                           '256B': '0', '373b': '0',
 165                                           '570B': '0'},
 166                                          "srcmac": "00:00:00:00:00:01",
 167                                          "dstmac": "00:00:00:00:00:02"},
 168                             'outer_l3v4': {'dstip6': '0064:ff9b:0:0:0:0:9810:2814-0064:ff9b:0:0:0:0:9810:2820',
 169                                            'proto': 'udp',
 170                                            'srcip6': '0064:ff9b:0:0:0:0:9810:6414-0064:ff9b:0:0:0:0:9810:6420',
 171                                            'dscp': 0, 'ttl': 32,
 172                                            'count': 1},
 173                             'outer_l4': {'dstport': '2001',
 174                                          'srcport': '1234',
 175                                          'count': 1}}},
 176                   'schema': 'isb:traffic_profile:0.1'}
 177
 178     def test___init__(self):
 179         TrafficProfile.params = self.PROFILE
 180         trex_profile = \
 181             TrexProfile(TrafficProfile)
 182         self.assertEqual(trex_profile.pps, 100)
 183
 184     def test_execute(self):
 185         trex_profile = \
 186             TrexProfile(TrafficProfile)
 187         self.assertEqual(None, trex_profile.execute({}))
 188
 189     def test_set_src_mac(self):
 190         src_mac = "00:00:00:00:00:01"
 191         trex_profile = \
 192             TrexProfile(TrafficProfile)
 193         self.assertEqual(None, trex_profile.set_src_mac(src_mac))
 194
 195         src_mac = "00:00:00:00:00:01-00:00:00:00:00:02"
 196         self.assertEqual(None, trex_profile.set_src_mac(src_mac))
 197
 198     def test_set_dst_mac(self):
 199         dst_mac = "00:00:00:00:00:03"
 200         trex_profile = \
 201             TrexProfile(TrafficProfile)
 202         self.assertEqual(None, trex_profile.set_dst_mac(dst_mac))
 203
 204         dst_mac = "00:00:00:00:00:03-00:00:00:00:00:04"
 205         self.assertEqual(None, trex_profile.set_dst_mac(dst_mac))
 206
 207     def test_set_src_ip4(self):
 208         src_ipv4 = "152.16.100.20"
 209         trex_profile = \
 210             TrexProfile(TrafficProfile)
 211         self.assertEqual(None, trex_profile.set_src_ip4(src_ipv4))
 212
 213         src_ipv4 = "152.16.100.20-152.16.100.30"
 214         self.assertEqual(None, trex_profile.set_src_ip4(src_ipv4))
 215
 216     def test_set_dst_ip4(self):
 217         dst_ipv4 = "152.16.100.20"
 218         trex_profile = \
 219             TrexProfile(TrafficProfile)
 220         self.assertEqual(None, trex_profile.set_dst_ip4(dst_ipv4))
 221
 222         dst_ipv4 = "152.16.100.20-152.16.100.30"
 223         self.assertEqual(None, trex_profile.set_dst_ip4(dst_ipv4))
 224
 225     def test_set_src_ip6(self):
 226         src_ipv6 = "0064:ff9b:0:0:0:0:9810:6414"
 227         trex_profile = \
 228             TrexProfile(TrafficProfile)
 229         self.assertEqual(None, trex_profile.set_src_ip6(src_ipv6))
 230
 231         src_ipv6 = "0064:ff9b:0:0:0:0:9810:6414-0064:ff9b:0:0:0:0:9810:6420"
 232         self.assertEqual(None, trex_profile.set_src_ip6(src_ipv6))
 233
 234     def test_set_dst_ip6(self):
 235         dst_ipv6 = "0064:ff9b:0:0:0:0:9810:6414"
 236         trex_profile = \
 237             TrexProfile(TrafficProfile)
 238         self.assertEqual(None, trex_profile.set_dst_ip6(dst_ipv6))
 239
 240         dst_ipv6 = "0064:ff9b:0:0:0:0:9810:6414-0064:ff9b:0:0:0:0:9810:6420"
 241         self.assertEqual(None, trex_profile.set_dst_ip6(dst_ipv6))
 242
 243     def test_dscp(self):
 244         dscp = "0"
 245         trex_profile = \
 246             TrexProfile(TrafficProfile)
 247         self.assertEqual(None, trex_profile.set_dscp(dscp))
 248
 249         dscp = "0-1"
 250         self.assertEqual(None, trex_profile.set_dscp(dscp))
 251
 252     def test_src_port(self):
 253         port = "1234"
 254         trex_profile = \
 255             TrexProfile(TrafficProfile)
 256         self.assertEqual(None, trex_profile.set_src_port(port))
 257
 258         port = "1234-5678"
 259         self.assertEqual(None, trex_profile.set_src_port(port))
 260
 261     def test_dst_port(self):
 262         port = "1234"
 263         trex_profile = \
 264             TrexProfile(TrafficProfile)
 265         self.assertEqual(None, trex_profile.set_dst_port(port))
 266
 267         port = "1234-5678"
 268         self.assertEqual(None, trex_profile.set_dst_port(port))
 269
 270     def test_qinq(self):
 271         qinq = {"S-VLAN": {"id": 128, "priority": 0, "cfi": 0},
 272                 "C-VLAN": {"id": 512, "priority": 0, "cfi": 0}}
 273
 274         trex_profile = \
 275             TrexProfile(TrafficProfile)
 276         self.assertEqual(None, trex_profile.set_qinq(qinq))
 277
 278         qinq = {"S-VLAN": {"id": "128-130", "priority": 0, "cfi": 0},
 279                 "C-VLAN": {"id": "512-515", "priority": 0, "cfi": 0}}
 280         self.assertEqual(None, trex_profile.set_qinq(qinq))
 281
 282     def test_set_outer_l2_fields(self):
 283         trex_profile = \
 284             TrexProfile(TrafficProfile)
 285         qinq = {"S-VLAN": {"id": 128, "priority": 0, "cfi": 0},
 286                 "C-VLAN": {"id": 512, "priority": 0, "cfi": 0}}
 287         outer_l2 = self.PROFILE['private']['ipv4']['outer_l2']
 288         outer_l2['QinQ'] = qinq
 289         self.assertEqual(None, trex_profile.set_outer_l2_fields(outer_l2))
 290
 291     def test_set_outer_l3v4_fields(self):
 292         trex_profile = \
 293             TrexProfile(TrafficProfile)
 294         outer_l3v4 = self.PROFILE['private']['ipv4']['outer_l3v4']
 295         outer_l3v4['proto'] = 'tcp'
 296         self.assertEqual(None, trex_profile.set_outer_l3v4_fields(outer_l3v4))
 297
 298     def test_set_outer_l3v6_fields(self):
 299         trex_profile = \
 300             TrexProfile(TrafficProfile)
 301         outer_l3v6 = self.PROFILE_v6['private']['ipv6']['outer_l3v4']
 302         outer_l3v6['proto'] = 'tcp'
 303         outer_l3v6['tc'] = 1
 304         outer_l3v6['hlim'] = 10
 305         self.assertEqual(None, trex_profile.set_outer_l3v6_fields(outer_l3v6))
 306
 307     def test_set_outer_l4_fields(self):
 308         trex_profile = \
 309             TrexProfile(TrafficProfile)
 310         outer_l4 = self.PROFILE['private']['ipv4']['outer_l4']
 311         self.assertEqual(None, trex_profile.set_outer_l4_fields(outer_l4))
 312
 313     def test_get_streams(self):
 314         trex_profile = \
 315             TrexProfile(TrafficProfile)
 316         trex_profile.params = self.PROFILE
 317         profile_data = self.PROFILE["private"]
 318         self.assertIsNotNone(trex_profile.get_streams(profile_data))
 319         trex_profile.pg_id = 1
 320         self.assertIsNotNone(trex_profile.get_streams(profile_data))
 321         trex_profile.params = self.PROFILE_v6
 322         trex_profile.profile_data = self.PROFILE_v6["private"]
 323         self.assertIsNotNone(trex_profile.get_streams(profile_data))
 324         trex_profile.pg_id = 1
 325         self.assertIsNotNone(trex_profile.get_streams(profile_data))
 326
 327     def test_generate_packets(self):
 328         trex_profile = \
 329             TrexProfile(TrafficProfile)
 330         trex_profile.fsize = 10
 331         trex_profile.base_pkt = [10]
 332         self.assertIsNone(trex_profile.generate_packets())
 333
 334     def test_generate_imix_data_error(self):
 335         trex_profile = \
 336             TrexProfile(TrafficProfile)
 337         self.assertEqual({}, trex_profile.generate_imix_data(False))
 338
 339     def test__get_start_end_ipv6(self):
 340         trex_profile = \
 341             TrexProfile(TrafficProfile)
 342         self.assertRaises(SystemExit, trex_profile._get_start_end_ipv6,
 343                           "1.1.1.3", "1.1.1.1")