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                                    'outer_l4': {'srcport': '2001',
 122                                                 'dsrport': '1234'}}},
 123                'private': {'ipv4':
 124                            {'outer_l2': {'framesize':
 125                                          {'64B': '100', '1518B': '0',
 126                                           '128B': '0', '1400B': '0',
 127                                           '256B': '0', '373b': '0',
 128                                           '570B': '0'},
 129                                          "srcmac": "00:00:00:00:00:01",
 130                                          "dstmac": "00:00:00:00:00:02"},
 131                             'outer_l3v4': {'dstip4': '9.9.1.1-90.105.255.255',
 132                                            'proto': 'udp',
 133                                            'srcip4': '1.1.1.1-1.15.255.255',
 134                                            'dscp': 0, 'ttl': 32},
 135                             'outer_l4': {'dstport': '2001',
 136                                          'srcport': '1234'}}},
 137                'schema': 'isb:traffic_profile:0.1'}
 138     PROFILE_v6 = {'description': 'Traffic profile to run RFC2544 latency',
 139                   'name': 'rfc2544',
 140                   'traffic_profile': {'traffic_type': 'RFC2544Profile',
 141                                       'frame_rate': 100},
 142                   'public': {'ipv6': {'outer_l2': {'framesize':
 143                                                    {'64B': '100', '1518B': '0',
 144                                                     '128B': '0', '1400B': '0',
 145                                                     '256B': '0', '373b': '0',
 146                                                     '570B': '0'},
 147                                                    "srcmac": "00:00:00:00:00:02",
 148                                                    "dstmac": "00:00:00:00:00:01"},
 149                                       'outer_l3v4': {'dstip6': '0064:ff9b:0:0:0:0:9810:6414-0064:ff9b:0:0:0:0:9810:6420',
 150                                                      'proto': 'udp',
 151                                                      'srcip6': '0064:ff9b:0:0:0:0:9810:2814-0064:ff9b:0:0:0:0:9810:2820',
 152                                                      'dscp': 0, 'ttl': 32},
 153                                       'outer_l4': {'srcport': '2001',
 154                                                    'dsrport': '1234'}}},
 155                   'private':
 156                   {'ipv6': {'outer_l2': {'framesize':
 157                                          {'64B': '100', '1518B': '0',
 158                                           '128B': '0', '1400B': '0',
 159                                           '256B': '0', '373b': '0',
 160                                           '570B': '0'},
 161                                          "srcmac": "00:00:00:00:00:01",
 162                                          "dstmac": "00:00:00:00:00:02"},
 163                             'outer_l3v4': {'dstip6': '0064:ff9b:0:0:0:0:9810:2814-0064:ff9b:0:0:0:0:9810:2820',
 164                                            'proto': 'udp',
 165                                            'srcip6': '0064:ff9b:0:0:0:0:9810:6414-0064:ff9b:0:0:0:0:9810:6420',
 166                                            'dscp': 0, 'ttl': 32},
 167                             'outer_l4': {'dstport': '2001',
 168                                          'srcport': '1234'}}},
 169                   'schema': 'isb:traffic_profile:0.1'}
 170
 171     def test___init__(self):
 172         TrafficProfile.params = self.PROFILE
 173         trex_profile = \
 174             TrexProfile(TrafficProfile)
 175         self.assertEqual(trex_profile.pps, 100)
 176
 177     def test_execute(self):
 178         trex_profile = \
 179             TrexProfile(TrafficProfile)
 180         self.assertEqual(None, trex_profile.execute({}))
 181
 182     def test_set_src_mac(self):
 183         src_mac = "00:00:00:00:00:01"
 184         trex_profile = \
 185             TrexProfile(TrafficProfile)
 186         self.assertEqual(None, trex_profile.set_src_mac(src_mac))
 187
 188         src_mac = "00:00:00:00:00:01-00:00:00:00:00:02"
 189         self.assertEqual(None, trex_profile.set_src_mac(src_mac))
 190
 191     def test_set_dst_mac(self):
 192         dst_mac = "00:00:00:00:00:03"
 193         trex_profile = \
 194             TrexProfile(TrafficProfile)
 195         self.assertEqual(None, trex_profile.set_dst_mac(dst_mac))
 196
 197         dst_mac = "00:00:00:00:00:03-00:00:00:00:00:04"
 198         self.assertEqual(None, trex_profile.set_dst_mac(dst_mac))
 199
 200     def test_set_src_ip4(self):
 201         src_ipv4 = "152.16.100.20"
 202         trex_profile = \
 203             TrexProfile(TrafficProfile)
 204         self.assertEqual(None, trex_profile.set_src_ip4(src_ipv4))
 205
 206         src_ipv4 = "152.16.100.20-152.16.100.30"
 207         self.assertEqual(None, trex_profile.set_src_ip4(src_ipv4))
 208
 209     def test_set_dst_ip4(self):
 210         dst_ipv4 = "152.16.100.20"
 211         trex_profile = \
 212             TrexProfile(TrafficProfile)
 213         self.assertEqual(None, trex_profile.set_dst_ip4(dst_ipv4))
 214
 215         dst_ipv4 = "152.16.100.20-152.16.100.30"
 216         self.assertEqual(None, trex_profile.set_dst_ip4(dst_ipv4))
 217
 218     def test_set_src_ip6(self):
 219         src_ipv6 = "0064:ff9b:0:0:0:0:9810:6414"
 220         trex_profile = \
 221             TrexProfile(TrafficProfile)
 222         self.assertEqual(None, trex_profile.set_src_ip6(src_ipv6))
 223
 224         src_ipv6 = "0064:ff9b:0:0:0:0:9810:6414-0064:ff9b:0:0:0:0:9810:6420"
 225         self.assertEqual(None, trex_profile.set_src_ip6(src_ipv6))
 226
 227     def test_set_dst_ip6(self):
 228         dst_ipv6 = "0064:ff9b:0:0:0:0:9810:6414"
 229         trex_profile = \
 230             TrexProfile(TrafficProfile)
 231         self.assertEqual(None, trex_profile.set_dst_ip6(dst_ipv6))
 232
 233         dst_ipv6 = "0064:ff9b:0:0:0:0:9810:6414-0064:ff9b:0:0:0:0:9810:6420"
 234         self.assertEqual(None, trex_profile.set_dst_ip6(dst_ipv6))
 235
 236     def test_dscp(self):
 237         dscp = "0"
 238         trex_profile = \
 239             TrexProfile(TrafficProfile)
 240         self.assertEqual(None, trex_profile.set_dscp(dscp))
 241
 242         dscp = "0-1"
 243         self.assertEqual(None, trex_profile.set_dscp(dscp))
 244
 245     def test_src_port(self):
 246         port = "1234"
 247         trex_profile = \
 248             TrexProfile(TrafficProfile)
 249         self.assertEqual(None, trex_profile.set_src_port(port))
 250
 251         port = "1234-5678"
 252         self.assertEqual(None, trex_profile.set_src_port(port))
 253
 254     def test_dst_port(self):
 255         port = "1234"
 256         trex_profile = \
 257             TrexProfile(TrafficProfile)
 258         self.assertEqual(None, trex_profile.set_dst_port(port))
 259
 260         port = "1234-5678"
 261         self.assertEqual(None, trex_profile.set_dst_port(port))
 262
 263     def test_qinq(self):
 264         qinq = {"S-VLAN": {"id": 128, "priority": 0, "cfi": 0},
 265                 "C-VLAN": {"id": 512, "priority": 0, "cfi": 0}}
 266
 267         trex_profile = \
 268             TrexProfile(TrafficProfile)
 269         self.assertEqual(None, trex_profile.set_qinq(qinq))
 270
 271         qinq = {"S-VLAN": {"id": "128-130", "priority": 0, "cfi": 0},
 272                 "C-VLAN": {"id": "512-515", "priority": 0, "cfi": 0}}
 273         self.assertEqual(None, trex_profile.set_qinq(qinq))
 274
 275     def test_set_outer_l2_fields(self):
 276         trex_profile = \
 277             TrexProfile(TrafficProfile)
 278         qinq = {"S-VLAN": {"id": 128, "priority": 0, "cfi": 0},
 279                 "C-VLAN": {"id": 512, "priority": 0, "cfi": 0}}
 280         outer_l2 = self.PROFILE['private']['ipv4']['outer_l2']
 281         outer_l2['QinQ'] = qinq
 282         self.assertEqual(None, trex_profile.set_outer_l2_fields(outer_l2))
 283
 284     def test_set_outer_l3v4_fields(self):
 285         trex_profile = \
 286             TrexProfile(TrafficProfile)
 287         outer_l3v4 = self.PROFILE['private']['ipv4']['outer_l3v4']
 288         outer_l3v4['proto'] = 'tcp'
 289         self.assertEqual(None, trex_profile.set_outer_l3v4_fields(outer_l3v4))
 290
 291     def test_set_outer_l3v6_fields(self):
 292         trex_profile = \
 293             TrexProfile(TrafficProfile)
 294         outer_l3v6 = self.PROFILE_v6['private']['ipv6']['outer_l3v4']
 295         outer_l3v6['proto'] = 'tcp'
 296         outer_l3v6['tc'] = 1
 297         outer_l3v6['hlim'] = 10
 298         self.assertEqual(None, trex_profile.set_outer_l3v6_fields(outer_l3v6))
 299
 300     def test_set_outer_l4_fields(self):
 301         trex_profile = \
 302             TrexProfile(TrafficProfile)
 303         outer_l4 = self.PROFILE['private']['ipv4']['outer_l4']
 304         self.assertEqual(None, trex_profile.set_outer_l4_fields(outer_l4))
 305
 306     def test_get_streams(self):
 307         trex_profile = \
 308             TrexProfile(TrafficProfile)
 309         trex_profile.params = self.PROFILE
 310         trex_profile.profile_data = self.PROFILE["private"]
 311         self.assertIsNotNone(trex_profile.get_streams())
 312         trex_profile.pg_id = 1
 313         self.assertIsNotNone(trex_profile.get_streams())
 314         trex_profile.params = self.PROFILE_v6
 315         trex_profile.profile_data = self.PROFILE_v6["private"]
 316         self.assertIsNotNone(trex_profile.get_streams())
 317         trex_profile.pg_id = 1
 318         self.assertIsNotNone(trex_profile.get_streams())
 319
 320     def test_generate_packets(self):
 321         trex_profile = \
 322             TrexProfile(TrafficProfile)
 323         trex_profile.fsize = 10
 324         trex_profile.base_pkt = [10]
 325         self.assertIsNone(trex_profile.generate_packets())
 326
 327     def test_generate_imix_data_error(self):
 328         trex_profile = \
 329             TrexProfile(TrafficProfile)
 330         self.assertEqual({}, trex_profile.generate_imix_data(False))
 331
 332     def test__get_start_end_ipv6(self):
 333         trex_profile = \
 334             TrexProfile(TrafficProfile)
 335         self.assertRaises(SystemExit, trex_profile._get_start_end_ipv6,
 336                           "1.1.1.3", "1.1.1.1")