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
24 from tests.unit import STL_MOCKS
27 STLClient = mock.MagicMock()
28 stl_patch = mock.patch.dict("sys.modules", STL_MOCKS)
32 from yardstick.network_services.vnf_generic.vnf.cgnapt_vnf import CgnaptApproxVnf, \
33 CgnaptApproxSetupEnvHelper
34 from yardstick.network_services.vnf_generic.vnf import cgnapt_vnf
35 from yardstick.network_services.nfvi.resource import ResourceProfile
37 TEST_FILE_YAML = 'nsb_test_case.yaml'
43 class TestCgnaptApproxSetupEnvHelper(unittest.TestCase):
45 def test__generate_ip_from_pool(self):
47 ip = CgnaptApproxSetupEnvHelper._generate_ip_from_pool("1.2.3.4")
48 self.assertEqual(next(ip), '1.2.3.4')
49 self.assertEqual(next(ip), '1.2.4.4')
50 self.assertEqual(next(ip), '1.2.5.4')
52 def test__update_cgnat_script_file(self):
55 # See the License for the specific language governing permissions and
56 # limitations under the License.
59 link 0 config {port0_local_ip} {port0_prefixlen}
62 link 1 config {port1_local_ip} {port1_prefixlen}
65 header = "This is a header"
67 out = CgnaptApproxSetupEnvHelper._update_cgnat_script_file(header, sample.splitlines(), "")
68 self.assertNotIn("This is a header", out)
70 def test__get_cgnapt_confgi(self):
72 c = CgnaptApproxSetupEnvHelper(mock.MagicMock(), mock.MagicMock(), mock.MagicMock())
73 c._get_ports_gateway = mock.Mock(return_value=3)
74 ret = c._get_cgnapt_config([{"name": 'a'}, {}, {"name": "b"}, {}, {"name": "c"}])
75 self.assertEqual(ret, [3, 3, 3])
78 @mock.patch("yardstick.network_services.vnf_generic.vnf.sample_vnf.Process")
79 class TestCgnaptApproxVnf(unittest.TestCase):
80 VNFD = {'vnfd:vnfd-catalog':
82 [{'short-name': 'VpeVnf',
85 [{'network': '152.16.100.20',
86 'netmask': '255.255.255.0',
87 'gateway': '152.16.100.20',
89 {'network': '152.16.40.20',
90 'netmask': '255.255.255.0',
91 'gateway': '152.16.40.20',
93 'description': 'VPE approximation using DPDK',
94 'name': 'vpevnf-baremetal',
96 [{'network': '0064:ff9b:0:0:0:0:9810:6414',
98 'gateway': '0064:ff9b:0:0:0:0:9810:6414',
100 {'network': '0064:ff9b:0:0:0:0:9810:2814',
102 'gateway': '0064:ff9b:0:0:0:0:9810:2814',
104 'id': 'vpevnf-baremetal',
105 'external-interface':
106 [{'virtual-interface':
107 {'dst_mac': '00:00:00:00:00:04',
108 'vpci': '0000:05:00.0',
109 'local_ip': '152.16.100.19',
110 'type': 'PCI-PASSTHROUGH',
111 'netmask': '255.255.255.0',
112 'dpdk_port_num': '0',
113 'bandwidth': '10 Gbps',
115 'dst_ip': '152.16.100.20',
116 'local_iface_name': 'xe0',
117 'local_mac': '00:00:00:00:00:02'},
118 'vnfd-connection-point-ref': 'xe0',
120 {'virtual-interface':
121 {'dst_mac': '00:00:00:00:00:03',
122 'vpci': '0000:05:00.1',
123 'local_ip': '152.16.40.19',
124 'type': 'PCI-PASSTHROUGH',
126 'netmask': '255.255.255.0',
127 'dpdk_port_num': '1',
128 'bandwidth': '10 Gbps',
129 'dst_ip': '152.16.40.20',
130 'local_iface_name': 'xe1',
131 'local_mac': '00:00:00:00:00:01'},
132 'vnfd-connection-point-ref': 'xe1',
134 'description': 'Vpe approximation using DPDK',
136 {'vdu-id': 'vpevnf-baremetal',
142 {'kpi': ['packets_in', 'packets_fwd', 'packets_dropped']},
143 'connection-point': [{'type': 'VPORT', 'name': 'xe0'},
144 {'type': 'VPORT', 'name': 'xe1'}],
145 'id': 'CgnaptApproxVnf', 'name': 'VPEVnfSsh'}]}}
147 scenario_cfg = {'options': {'packetsize': 64, 'traffic_type': 4,
148 'rfc2544': {'allowed_drop_rate': '0.8 - 1'},
149 'vnf__1': {'rules': 'acl_1rule.yaml',
150 'vnf_config': {'lb_config': 'SW',
154 'worker_threads': 1}}
156 'task_id': 'a70bdf4a-8e67-47a3-9dc1-273c14506eb7',
158 'tc': 'tc_ipv4_1Mflow_64B_packetsize',
159 'runner': {'object': 'NetworkServiceTestCase',
161 'output_filename': '/tmp/yardstick.out',
162 'runner_id': 74476, 'duration': 400,
164 'traffic_profile': 'ipv4_throughput_acl.yaml',
165 'traffic_options': {'flow': 'ipv4_Packets_acl.yaml',
166 'imix': 'imix_voice.yaml'},
168 'nodes': {'tg__2': 'trafficgen_2.yardstick',
169 'tg__1': 'trafficgen_1.yardstick',
170 'vnf__1': 'vnf.yardstick'},
171 'topology': 'vpe-tg-topology-baremetal.yaml'}
173 context_cfg = {'nodes': {'tg__2':
174 {'member-vnf-index': '3',
175 'role': 'TrafficGen',
176 'name': 'trafficgen_2.yardstick',
177 'vnfd-id-ref': 'tg__2',
180 {'xe0': {'local_iface_name': 'ens513f0',
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',
188 'vpci': '0000:02:00.0',
190 'xe1': {'local_iface_name': 'ens513f1',
191 'netmask': '255.255.255.0',
192 'network': '202.16.100.0',
193 'local_ip': '202.16.100.20',
194 'local_mac': '00:1e:67:d0:60:5d',
196 'vpci': '0000:02:00.1',
197 'dpdk_port_num': 1}},
199 'VNF model': 'l3fwd_vnf.yaml',
202 {'member-vnf-index': '1',
203 'role': 'TrafficGen',
204 'name': 'trafficgen_1.yardstick',
205 'vnfd-id-ref': 'tg__1',
208 {'xe0': {'local_iface_name': 'ens785f0',
210 'netmask': '255.255.255.0',
211 'local_ip': '152.16.100.20',
212 'dst_mac': '00:00:00:00:00:02',
213 'local_mac': '00:00:00:00:00:04',
214 'dst_ip': '152.16.100.19',
216 'vpci': '0000:05:00.0',
218 'xe1': {'local_iface_name': 'ens785f1',
219 'netmask': '255.255.255.0',
220 'local_ip': '152.16.100.21',
221 'local_mac': '00:00:00:00:00:01',
223 'vpci': '0000:05:00.1',
224 'dpdk_port_num': 1}},
226 'VNF model': 'tg_rfc2544_tpl.yaml',
229 {'name': 'vnf.yardstick',
230 'vnfd-id-ref': 'vnf__1',
233 {'xe0': {'local_iface_name': 'ens786f0',
235 'netmask': '255.255.255.0',
236 'local_ip': '152.16.100.19',
237 'dst_mac': '00:00:00:00:00:04',
238 'local_mac': '00:00:00:00:00:02',
239 'dst_ip': '152.16.100.20',
241 'vpci': '0000:05:00.0',
243 'xe1': {'local_iface_name': 'ens786f1',
245 'netmask': '255.255.255.0',
246 'local_ip': '152.16.40.19',
247 'dst_mac': '00:00:00:00:00:03',
248 'local_mac': '00:00:00:00:00:01',
249 'dst_ip': '152.16.40.20',
251 'vpci': '0000:05:00.1',
252 'dpdk_port_num': 1}},
254 [{'netmask': '255.255.255.0',
255 'gateway': '152.16.100.20',
256 'network': '152.16.100.20',
258 {'netmask': '255.255.255.0',
259 'gateway': '152.16.40.20',
260 'network': '152.16.40.20',
262 'member-vnf-index': '2',
268 'gateway': '0064:ff9b:0:0:0:0:9810:6414',
269 'network': '0064:ff9b:0:0:0:0:9810:6414',
272 'gateway': '0064:ff9b:0:0:0:0:9810:2814',
273 'network': '0064:ff9b:0:0:0:0:9810:2814',
276 'VNF model': 'cgnapt_vnf.yaml'}}}
278 def test___init__(self, mock_process):
279 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
280 cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
281 self.assertIsNone(cgnapt_approx_vnf._vnf_process)
283 @mock.patch("yardstick.network_services.vnf_generic.vnf.sample_vnf.time")
284 def test_collect_kpi(self, mock_time, mock_process):
285 with mock.patch("yardstick.ssh.SSH") as ssh:
286 ssh_mock = mock.Mock(autospec=ssh.SSH)
288 mock.Mock(return_value=(0, "", ""))
289 ssh.from_node.return_value = ssh_mock
290 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
291 cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
292 cgnapt_approx_vnf.q_in = mock.MagicMock()
293 cgnapt_approx_vnf.q_out = mock.MagicMock()
294 cgnapt_approx_vnf.q_out.qsize = mock.Mock(return_value=0)
295 cgnapt_approx_vnf.resource = mock.Mock(autospec=ResourceProfile)
296 result = {'packets_dropped': 0, 'packets_fwd': 0, 'packets_in': 0}
297 self.assertEqual(result, cgnapt_approx_vnf.collect_kpi())
299 @mock.patch("yardstick.network_services.vnf_generic.vnf.sample_vnf.time")
300 def test_vnf_execute_command(self, mock_time, mock_process):
301 with mock.patch("yardstick.ssh.SSH") as ssh:
302 ssh_mock = mock.Mock(autospec=ssh.SSH)
304 mock.Mock(return_value=(0, "", ""))
305 ssh.from_node.return_value = ssh_mock
306 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
307 cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
308 cgnapt_approx_vnf.q_in = mock.MagicMock()
309 cgnapt_approx_vnf.q_out = mock.MagicMock()
310 cgnapt_approx_vnf.q_out.qsize = mock.Mock(return_value=0)
312 self.assertEqual("", cgnapt_approx_vnf.vnf_execute(cmd))
314 def test_get_stats(self, mock_process):
315 with mock.patch("yardstick.ssh.SSH") as ssh:
316 ssh_mock = mock.Mock(autospec=ssh.SSH)
318 mock.Mock(return_value=(0, "", ""))
319 ssh.from_node.return_value = ssh_mock
320 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
321 cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
322 cgnapt_approx_vnf.q_in = mock.MagicMock()
323 cgnapt_approx_vnf.q_out = mock.MagicMock()
324 cgnapt_approx_vnf.q_out.qsize = mock.Mock(return_value=0)
326 "CG-NAPT(.*\n)*Received 100, Missed 0, Dropped 0,Translated 100,ingress"
327 cgnapt_approx_vnf.vnf_execute = mock.Mock(return_value=mock_result)
328 self.assertListEqual(list(mock_result), list(cgnapt_approx_vnf.get_stats()))
330 def _get_file_abspath(self, filename):
331 curr_path = os.path.dirname(os.path.abspath(__file__))
332 file_path = os.path.join(curr_path, filename)
335 @mock.patch("yardstick.network_services.vnf_generic.vnf.cgnapt_vnf.hex")
336 @mock.patch("yardstick.network_services.vnf_generic.vnf.cgnapt_vnf.eval")
337 @mock.patch('yardstick.network_services.vnf_generic.vnf.cgnapt_vnf.open')
338 def test_run_vcgnapt(self, hex, eval, mock_open, mock_process):
339 with mock.patch("yardstick.ssh.SSH") as ssh:
340 ssh_mock = mock.Mock(autospec=ssh.SSH)
342 mock.Mock(return_value=(0, "", ""))
344 mock.Mock(return_value=(0, "", ""))
345 ssh.from_node.return_value = ssh_mock
346 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
347 cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
348 cgnapt_approx_vnf._build_config = mock.MagicMock()
349 cgnapt_approx_vnf.queue_wrapper = mock.MagicMock()
350 cgnapt_approx_vnf.ssh_helper = mock.MagicMock()
351 cgnapt_approx_vnf.ssh_helper.run = mock.MagicMock()
352 cgnapt_approx_vnf.scenario_helper.scenario_cfg = self.scenario_cfg
353 cgnapt_approx_vnf._run()
354 cgnapt_approx_vnf.ssh_helper.run.assert_called_once()
356 @mock.patch("yardstick.network_services.vnf_generic.vnf.sample_vnf.Context")
357 def test_instantiate(self, mock_context, mock_process):
358 with mock.patch("yardstick.ssh.SSH") as ssh:
359 ssh_mock = mock.Mock(autospec=ssh.SSH)
361 mock.Mock(return_value=(0, "", ""))
362 ssh.from_node.return_value = ssh_mock
363 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
364 cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
365 cgnapt_approx_vnf.ssh_helper = ssh
366 cgnapt_approx_vnf.deploy_helper = mock.MagicMock()
367 cgnapt_approx_vnf.resource_helper = mock.MagicMock()
368 cgnapt_approx_vnf._build_config = mock.MagicMock()
369 self.scenario_cfg['vnf_options'] = {'acl': {'cfg': "",
371 cgnapt_approx_vnf.q_out.put("pipeline>")
372 cgnapt_vnf.WAIT_TIME = 3
373 self.scenario_cfg.update({"nodes": {"vnf__1": ""}})
374 self.assertIsNone(cgnapt_approx_vnf.instantiate(self.scenario_cfg,
377 def test_scale(self, mock_process):
378 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
379 cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
381 self.assertRaises(NotImplementedError, cgnapt_approx_vnf.scale, flavor)
383 @mock.patch("yardstick.network_services.vnf_generic.vnf.sample_vnf.time")
384 def test_terminate(self, mock_time, mock_process):
385 with mock.patch("yardstick.ssh.SSH") as ssh:
386 ssh_mock = mock.Mock(autospec=ssh.SSH)
388 mock.Mock(return_value=(0, "", ""))
389 ssh.from_node.return_value = ssh_mock
390 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
391 cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
392 cgnapt_approx_vnf._vnf_process = mock.MagicMock()
393 cgnapt_approx_vnf._vnf_process.terminate = mock.Mock()
394 cgnapt_approx_vnf.used_drivers = {"01:01.0": "i40e",
396 cgnapt_approx_vnf.vnf_execute = mock.MagicMock()
397 cgnapt_approx_vnf.ssh_helper = ssh_mock
398 cgnapt_approx_vnf.dpdk_nic_bind = "dpdk_nic_bind.py"
399 cgnapt_approx_vnf._resource_collect_stop = mock.Mock()
400 self.assertEqual(None, cgnapt_approx_vnf.terminate())
402 @mock.patch("yardstick.network_services.vnf_generic.vnf.sample_vnf.time")
403 @mock.patch("yardstick.network_services.vnf_generic.vnf.cgnapt_vnf.time")
404 def test__vnf_up_post(self, mock_time, mock_cgnapt_time, mock_process):
405 with mock.patch("yardstick.ssh.SSH") as ssh:
406 ssh_mock = mock.Mock(autospec=ssh.SSH)
408 mock.Mock(return_value=(0, "", ""))
409 ssh.from_node.return_value = ssh_mock
410 vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
411 cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
412 cgnapt_approx_vnf._vnf_process = mock.MagicMock()
413 cgnapt_approx_vnf._vnf_process.terminate = mock.Mock()
414 cgnapt_approx_vnf.vnf_execute = mock.MagicMock()
415 cgnapt_approx_vnf.ssh_helper = ssh_mock
416 cgnapt_approx_vnf.scenario_helper.scenario_cfg = self.scenario_cfg
417 cgnapt_approx_vnf._resource_collect_stop = mock.Mock()
418 cgnapt_approx_vnf._vnf_up_post()
419 cgnapt_approx_vnf.vnf_execute.assert_called_once()
422 if __name__ == '__main__':