3 # Copyright (c) 2016-2017 Intel Corporation
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
9 # http://www.apache.org/licenses/LICENSE-2.0
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.
18 from __future__ import absolute_import
23 SSH_HELPER = 'yardstick.network_services.vnf_generic.vnf.sample_vnf.VnfSshHelper'
25 from tests.unit.network_services.vnf_generic.vnf.test_base import mock_ssh
26 from tests.unit import STL_MOCKS
30 STLClient = mock.MagicMock()
31 stl_patch = mock.patch.dict("sys.modules", STL_MOCKS)
35 from yardstick.network_services.vnf_generic.vnf.tg_trex import \
36 TrexTrafficGen, TrexResourceHelper
37 from yardstick.network_services.traffic_profile.base import TrafficProfile
40 class TestTrexTrafficGen(unittest.TestCase):
41 VNFD = {'vnfd:vnfd-catalog':
43 [{'short-name': 'VpeVnf',
46 [{'network': '152.16.100.20',
47 'netmask': '255.255.255.0',
48 'gateway': '152.16.100.20',
50 {'network': '152.16.40.20',
51 'netmask': '255.255.255.0',
52 'gateway': '152.16.40.20',
54 'description': 'VPE approximation using DPDK',
55 'name': 'vpevnf-baremetal',
57 [{'network': '0064:ff9b:0:0:0:0:9810:6414',
59 'gateway': '0064:ff9b:0:0:0:0:9810:6414',
61 {'network': '0064:ff9b:0:0:0:0:9810:2814',
63 'gateway': '0064:ff9b:0:0:0:0:9810:2814',
65 'id': 'vpevnf-baremetal',
67 [{'virtual-interface':
68 {'dst_mac': '00:00:00:00:00:04',
69 'vpci': '0000:05:00.0',
70 'local_ip': '152.16.100.19',
71 'type': 'PCI-PASSTHROUGH',
72 'netmask': '255.255.255.0',
74 'bandwidth': '10 Gbps',
76 'dst_ip': '152.16.100.20',
77 'local_iface_name': 'xe0',
78 'local_mac': '00:00:00:00:00:02'},
79 'vnfd-connection-point-ref': 'xe0',
82 {'dst_mac': '00:00:00:00:00:03',
83 'vpci': '0000:05:00.1',
84 'local_ip': '152.16.40.19',
85 'type': 'PCI-PASSTHROUGH',
87 'netmask': '255.255.255.0',
89 'bandwidth': '10 Gbps',
90 'dst_ip': '152.16.40.20',
91 'local_iface_name': 'xe1',
92 'local_mac': '00:00:00:00:00:01'},
93 'vnfd-connection-point-ref': 'xe1',
95 'description': 'Vpe approximation using DPDK',
97 {'vdu-id': 'vpevnf-baremetal',
103 {'kpi': ['packets_in', 'packets_fwd', 'packets_dropped']},
104 'connection-point': [{'type': 'VPORT', 'name': 'xe0'},
105 {'type': 'VPORT', 'name': 'xe1'}],
106 'id': 'VpeApproxVnf', 'name': 'VPEVnfSsh'}]}}
109 "schema": "isb:traffic_profile:0.1",
111 "description": "Fixed traffic profile to run UDP traffic",
113 "traffic_type": "FixedTraffic",
114 "frame_rate": 100, # pps
125 "allowed_drop_rate": "0.8 - 1",
128 "rules": "acl_1rule.yaml",
132 "worker_config": "1C/1T",
137 "task_id": "a70bdf4a-8e67-47a3-9dc1-273c14506eb7",
138 "tc": "tc_ipv4_1Mflow_64B_packetsize",
140 "object": "NetworkServiceTestCase",
142 "output_filename": "/tmp/yardstick.out",
143 "runner_id": 74476, "duration": 400,
146 "traffic_profile": "ipv4_throughput_acl.yaml",
148 "flow": "ipv4_Packets_acl.yaml",
149 "imix": "imix_voice.yaml"
153 "tg__2": "trafficgen_2.yardstick",
154 "tg__1": "trafficgen_1.yardstick",
155 "vnf__1": "vnf.yardstick"
157 "topology": "udpreplay-tg-topology-baremetal.yaml"
163 "vnfd-id-ref": "vnf__1",
167 "local_iface_name": "ens786f0",
168 "vld_id": TrafficProfile.UPLINK,
169 "netmask": "255.255.255.0",
170 "vpci": "0000:05:00.0",
171 "local_ip": "152.16.100.19",
173 "dst_ip": "152.16.100.20",
174 "local_mac": "00:00:00:00:00:02",
175 "dst_mac": "00:00:00:00:00:04",
179 "local_iface_name": "ens786f1",
180 "vld_id": TrafficProfile.DOWNLINK,
181 "netmask": "255.255.255.0",
182 "vpci": "0000:05:00.1",
183 "local_ip": "152.16.40.19",
185 "dst_ip": "152.16.40.20",
186 "local_mac": "00:00:00:00:00:01",
187 "dst_mac": "00:00:00:00:00:03",
197 "gateway": "0064:ff9b:0:0:0:0:9810:6414",
198 "network": "0064:ff9b:0:0:0:0:9810:6414"
203 "gateway": "0064:ff9b:0:0:0:0:9810:2814",
204 "network": "0064:ff9b:0:0:0:0:9810:2814"
208 "VNF model": "udp_replay.yaml",
209 "name": "vnf.yardstick",
210 "member-vnf-index": "2",
213 "netmask": "255.255.255.0",
215 "gateway": "152.16.100.20",
216 "network": "152.16.100.20"
219 "netmask": "255.255.255.0",
221 "gateway": "152.16.40.20",
222 "network": "152.16.40.20"
227 "trafficgen_2.yardstick": {
228 "member-vnf-index": "3",
229 "role": "TrafficGen",
230 "name": "trafficgen_2.yardstick",
231 "vnfd-id-ref": "tg__2",
235 "local_iface_name": "ens513f0",
236 "vld_id": TrafficProfile.DOWNLINK,
237 "netmask": "255.255.255.0",
238 "vpci": "0000:02:00.0",
239 "local_ip": "152.16.40.20",
241 "dst_ip": "152.16.40.19",
242 "local_mac": "00:00:00:00:00:03",
243 "dst_mac": "00:00:00:00:00:01",
247 "local_iface_name": "ens513f1",
248 "netmask": "255.255.255.0",
249 "network": "202.16.100.0",
250 "local_ip": "202.16.100.20",
252 "local_mac": "00:1e:67:d0:60:5d",
253 "vpci": "0000:02:00.1",
258 "VNF model": "l3fwd_vnf.yaml",
261 "trafficgen_1.yardstick": {
262 "member-vnf-index": "1",
263 "role": "TrafficGen",
264 "name": "trafficgen_1.yardstick",
265 "vnfd-id-ref": "tg__1",
269 "local_iface_name": "ens785f0",
270 "vld_id": TrafficProfile.UPLINK,
271 "netmask": "255.255.255.0",
272 "vpci": "0000:05:00.0",
273 "local_ip": "152.16.100.20",
275 "dst_ip": "152.16.100.19",
276 "local_mac": "00:00:00:00:00:04",
277 "dst_mac": "00:00:00:00:00:02",
281 "local_ip": "152.16.100.21",
283 "vpci": "0000:05:00.1",
285 "local_iface_name": "ens785f1",
286 "netmask": "255.255.255.0",
287 "local_mac": "00:00:00:00:00:01"
291 "VNF model": "tg_rfc2544_tpl.yaml",
297 @mock.patch(SSH_HELPER)
298 def test___init__(self, ssh):
300 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
301 trex_traffic_gen = TrexTrafficGen(NAME, vnfd)
302 self.assertIsInstance(trex_traffic_gen.resource_helper, TrexResourceHelper)
304 @mock.patch(SSH_HELPER)
305 def test_collect_kpi(self, ssh):
307 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
308 trex_traffic_gen = TrexTrafficGen(NAME, vnfd)
309 trex_traffic_gen.resource_helper._queue.put({})
310 result = trex_traffic_gen.collect_kpi()
311 self.assertEqual({}, result)
313 @mock.patch(SSH_HELPER)
314 def test_listen_traffic(self, ssh):
316 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
317 trex_traffic_gen = TrexTrafficGen(NAME, vnfd)
318 self.assertIsNone(trex_traffic_gen.listen_traffic({}))
320 @mock.patch(SSH_HELPER)
321 def test_instantiate(self, ssh):
324 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
325 trex_traffic_gen = TrexTrafficGen(NAME, vnfd)
326 trex_traffic_gen._start_server = mock.Mock(return_value=0)
327 trex_traffic_gen._tg_process = mock.MagicMock()
328 trex_traffic_gen._tg_process.start = mock.Mock()
329 trex_traffic_gen._tg_process.exitcode = 0
330 trex_traffic_gen._tg_process._is_alive = mock.Mock(return_value=1)
331 trex_traffic_gen.ssh_helper = mock.MagicMock()
332 trex_traffic_gen.resource_helper.ssh_helper = mock.MagicMock()
333 trex_traffic_gen.setup_helper.setup_vnf_environment = mock.MagicMock()
335 self.assertIsNone(trex_traffic_gen.instantiate(self.SCENARIO_CFG, self.CONTEXT_CFG))
337 @mock.patch(SSH_HELPER)
338 def test_instantiate_error(self, ssh):
339 mock_ssh(ssh, exec_result=(1, "", ""))
341 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
342 trex_traffic_gen = TrexTrafficGen(NAME, vnfd)
343 trex_traffic_gen._start_server = mock.Mock(return_value=0)
344 trex_traffic_gen._tg_process = mock.MagicMock()
345 trex_traffic_gen._tg_process.start = mock.Mock()
346 trex_traffic_gen._tg_process._is_alive = mock.Mock(return_value=0)
347 trex_traffic_gen.ssh_helper = mock.MagicMock()
348 trex_traffic_gen.resource_helper.ssh_helper = mock.MagicMock()
349 trex_traffic_gen.setup_helper.setup_vnf_environment = mock.MagicMock()
350 self.assertIsNone(trex_traffic_gen.instantiate(self.SCENARIO_CFG, self.CONTEXT_CFG))
352 @mock.patch(SSH_HELPER)
353 def test__start_server(self, ssh):
355 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
356 trex_traffic_gen = TrexTrafficGen(NAME, vnfd)
357 trex_traffic_gen.ssh_helper = mock.MagicMock()
358 trex_traffic_gen.resource_helper.ssh_helper = mock.MagicMock()
359 self.assertIsNone(trex_traffic_gen._start_server())
361 @mock.patch(SSH_HELPER)
362 def test__traffic_runner(self, ssh):
365 mock_traffic_profile = mock.Mock(autospec=TrafficProfile)
366 mock_traffic_profile.get_traffic_definition.return_value = "64"
367 mock_traffic_profile.execute_traffic.return_value = "64"
368 mock_traffic_profile.params = self.TRAFFIC_PROFILE
370 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
371 self.sut = TrexTrafficGen(NAME, vnfd)
372 self.sut.ssh_helper = mock.Mock()
373 self.sut.ssh_helper.run = mock.Mock()
374 self.sut._vpci_ascending = ["0000:05:00.0", "0000:05:00.1"]
375 self.sut._connect_client = mock.Mock(autospec=STLClient)
376 self.sut._connect_client.get_stats = mock.Mock(return_value="0")
377 self.sut.resource_helper.RUN_DURATION = 0
378 self.sut.resource_helper.QUEUE_WAIT_TIME = 0
379 self.sut._traffic_runner(mock_traffic_profile)
381 @mock.patch(SSH_HELPER)
382 def test__generate_trex_cfg(self, ssh):
384 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
385 trex_traffic_gen = TrexTrafficGen(NAME, vnfd)
386 trex_traffic_gen.resource_helper.ssh_helper = mock.MagicMock()
387 self.assertIsNone(trex_traffic_gen.resource_helper.generate_cfg())
389 @mock.patch(SSH_HELPER)
390 def test_run_traffic(self, ssh):
393 mock_traffic_profile = mock.Mock(autospec=TrafficProfile)
394 mock_traffic_profile.get_traffic_definition.return_value = "64"
395 mock_traffic_profile.params = self.TRAFFIC_PROFILE
397 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
398 self.sut = TrexTrafficGen(NAME, vnfd)
399 self.sut.ssh_helper = mock.Mock()
400 self.sut.ssh_helper.run = mock.Mock()
401 self.sut._traffic_runner = mock.Mock(return_value=0)
402 self.sut.resource_helper.client_started.value = 1
403 result = self.sut.run_traffic(mock_traffic_profile)
404 self.sut._traffic_process.terminate()
405 self.assertIsNotNone(result)
407 @mock.patch(SSH_HELPER)
408 def test_scale(self, ssh):
409 mock_ssh(ssh, exec_result=(1, "", ""))
410 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
411 trex_traffic_gen = TrexTrafficGen(NAME, vnfd)
412 trex_traffic_gen.scale('')
414 @mock.patch(SSH_HELPER)
415 def test_terminate(self, ssh):
417 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
418 trex_traffic_gen = TrexTrafficGen(NAME, vnfd)
419 trex_traffic_gen.ssh_helper = mock.MagicMock()
420 trex_traffic_gen.resource_helper.ssh_helper = mock.MagicMock()
421 self.assertIsNone(trex_traffic_gen.terminate())
423 @mock.patch(SSH_HELPER)
424 def test__connect_client(self, ssh):
426 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
427 trex_traffic_gen = TrexTrafficGen(NAME, vnfd)
428 client = mock.Mock(autospec=STLClient)
429 client.connect = mock.Mock(return_value=0)
430 self.assertIsNotNone(trex_traffic_gen.resource_helper._connect(client))
433 if __name__ == '__main__':