Merge "Sample CGNAPT VNF"
[yardstick.git] / tests / unit / network_services / vnf_generic / vnf / test_cgnapt_vnf.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
20 import os
21 import unittest
22
23 import mock
24
25 STL_MOCKS = {
26     'stl': mock.MagicMock(),
27     'stl.trex_stl_lib': mock.MagicMock(),
28     'stl.trex_stl_lib.base64': mock.MagicMock(),
29     'stl.trex_stl_lib.binascii': mock.MagicMock(),
30     'stl.trex_stl_lib.collections': mock.MagicMock(),
31     'stl.trex_stl_lib.copy': mock.MagicMock(),
32     'stl.trex_stl_lib.datetime': mock.MagicMock(),
33     'stl.trex_stl_lib.functools': mock.MagicMock(),
34     'stl.trex_stl_lib.imp': mock.MagicMock(),
35     'stl.trex_stl_lib.inspect': mock.MagicMock(),
36     'stl.trex_stl_lib.json': mock.MagicMock(),
37     'stl.trex_stl_lib.linecache': mock.MagicMock(),
38     'stl.trex_stl_lib.math': mock.MagicMock(),
39     'stl.trex_stl_lib.os': mock.MagicMock(),
40     'stl.trex_stl_lib.platform': mock.MagicMock(),
41     'stl.trex_stl_lib.pprint': mock.MagicMock(),
42     'stl.trex_stl_lib.random': mock.MagicMock(),
43     'stl.trex_stl_lib.re': mock.MagicMock(),
44     'stl.trex_stl_lib.scapy': mock.MagicMock(),
45     'stl.trex_stl_lib.socket': mock.MagicMock(),
46     'stl.trex_stl_lib.string': mock.MagicMock(),
47     'stl.trex_stl_lib.struct': mock.MagicMock(),
48     'stl.trex_stl_lib.sys': mock.MagicMock(),
49     'stl.trex_stl_lib.threading': mock.MagicMock(),
50     'stl.trex_stl_lib.time': mock.MagicMock(),
51     'stl.trex_stl_lib.traceback': mock.MagicMock(),
52     'stl.trex_stl_lib.trex_stl_async_client': mock.MagicMock(),
53     'stl.trex_stl_lib.trex_stl_client': mock.MagicMock(),
54     'stl.trex_stl_lib.trex_stl_exceptions': mock.MagicMock(),
55     'stl.trex_stl_lib.trex_stl_ext': mock.MagicMock(),
56     'stl.trex_stl_lib.trex_stl_jsonrpc_client': mock.MagicMock(),
57     'stl.trex_stl_lib.trex_stl_packet_builder_interface': mock.MagicMock(),
58     'stl.trex_stl_lib.trex_stl_packet_builder_scapy': mock.MagicMock(),
59     'stl.trex_stl_lib.trex_stl_port': mock.MagicMock(),
60     'stl.trex_stl_lib.trex_stl_stats': mock.MagicMock(),
61     'stl.trex_stl_lib.trex_stl_streams': mock.MagicMock(),
62     'stl.trex_stl_lib.trex_stl_types': mock.MagicMock(),
63     'stl.trex_stl_lib.types': mock.MagicMock(),
64     'stl.trex_stl_lib.utils': mock.MagicMock(),
65     'stl.trex_stl_lib.utils.argparse': mock.MagicMock(),
66     'stl.trex_stl_lib.utils.collections': mock.MagicMock(),
67     'stl.trex_stl_lib.utils.common': mock.MagicMock(),
68     'stl.trex_stl_lib.utils.json': mock.MagicMock(),
69     'stl.trex_stl_lib.utils.os': mock.MagicMock(),
70     'stl.trex_stl_lib.utils.parsing_opts': mock.MagicMock(),
71     'stl.trex_stl_lib.utils.pwd': mock.MagicMock(),
72     'stl.trex_stl_lib.utils.random': mock.MagicMock(),
73     'stl.trex_stl_lib.utils.re': mock.MagicMock(),
74     'stl.trex_stl_lib.utils.string': mock.MagicMock(),
75     'stl.trex_stl_lib.utils.sys': mock.MagicMock(),
76     'stl.trex_stl_lib.utils.text_opts': mock.MagicMock(),
77     'stl.trex_stl_lib.utils.text_tables': mock.MagicMock(),
78     'stl.trex_stl_lib.utils.texttable': mock.MagicMock(),
79     'stl.trex_stl_lib.warnings': mock.MagicMock(),
80     'stl.trex_stl_lib.yaml': mock.MagicMock(),
81     'stl.trex_stl_lib.zlib': mock.MagicMock(),
82     'stl.trex_stl_lib.zmq': mock.MagicMock(),
83 }
84
85 STLClient = mock.MagicMock()
86 stl_patch = mock.patch.dict("sys.modules", STL_MOCKS)
87 stl_patch.start()
88
89 if stl_patch:
90     from yardstick.network_services.vnf_generic.vnf.cgnapt_vnf import CgnaptApproxVnf, \
91         CgnaptApproxSetupEnvHelper
92     from yardstick.network_services.vnf_generic.vnf import cgnapt_vnf
93     from yardstick.network_services.nfvi.resource import ResourceProfile
94
95 TEST_FILE_YAML = 'nsb_test_case.yaml'
96
97
98 name = 'vnf__1'
99
100
101 class TestCgnaptApproxSetupEnvHelper(unittest.TestCase):
102
103     def test__generate_ip_from_pool(self):
104
105         ip = CgnaptApproxSetupEnvHelper._generate_ip_from_pool("1.2.3.4")
106         self.assertEqual(next(ip), '1.2.3.4')
107         self.assertEqual(next(ip), '1.2.4.4')
108         self.assertEqual(next(ip), '1.2.5.4')
109
110     def test__update_cgnat_script_file(self):
111
112         sample = """\
113 # See the License for the specific language governing permissions and
114 # limitations under the License.
115
116 link 0 down
117 link 0 config {port0_local_ip} {port0_prefixlen}
118 link 0 up
119 link 1 down
120 link 1 config {port1_local_ip} {port1_prefixlen}
121 link 1 up
122 """
123         header = "This is a header"
124
125         out = CgnaptApproxSetupEnvHelper._update_cgnat_script_file(header, sample.splitlines(), "")
126         self.assertNotIn("This is a header", out)
127
128     def test__get_cgnapt_confgi(self):
129
130         c = CgnaptApproxSetupEnvHelper(mock.MagicMock(), mock.MagicMock(), mock.MagicMock())
131         c._get_ports_gateway = mock.Mock(return_value=3)
132         ret = c._get_cgnapt_config([{"name": 'a'}, {}, {"name": "b"}, {}, {"name": "c"}])
133         self.assertEqual(ret, [3, 3, 3])
134
135
136 @mock.patch("yardstick.network_services.vnf_generic.vnf.sample_vnf.Process")
137 class TestCgnaptApproxVnf(unittest.TestCase):
138     VNFD = {'vnfd:vnfd-catalog':
139             {'vnfd':
140              [{'short-name': 'VpeVnf',
141                'vdu':
142                [{'routing_table':
143                  [{'network': '152.16.100.20',
144                    'netmask': '255.255.255.0',
145                    'gateway': '152.16.100.20',
146                    'if': 'xe0'},
147                   {'network': '152.16.40.20',
148                    'netmask': '255.255.255.0',
149                    'gateway': '152.16.40.20',
150                    'if': 'xe1'}],
151                  'description': 'VPE approximation using DPDK',
152                  'name': 'vpevnf-baremetal',
153                  'nd_route_tbl':
154                  [{'network': '0064:ff9b:0:0:0:0:9810:6414',
155                    'netmask': '112',
156                    'gateway': '0064:ff9b:0:0:0:0:9810:6414',
157                    'if': 'xe0'},
158                   {'network': '0064:ff9b:0:0:0:0:9810:2814',
159                    'netmask': '112',
160                    'gateway': '0064:ff9b:0:0:0:0:9810:2814',
161                    'if': 'xe1'}],
162                  'id': 'vpevnf-baremetal',
163                  'external-interface':
164                  [{'virtual-interface':
165                    {'dst_mac': '00:00:00:00:00:04',
166                     'vpci': '0000:05:00.0',
167                     'local_ip': '152.16.100.19',
168                     'type': 'PCI-PASSTHROUGH',
169                     'netmask': '255.255.255.0',
170                     'dpdk_port_num': '0',
171                     'bandwidth': '10 Gbps',
172                     'driver': "i40e",
173                     'dst_ip': '152.16.100.20',
174                     'local_iface_name': 'xe0',
175                     'local_mac': '00:00:00:00:00:02'},
176                    'vnfd-connection-point-ref': 'xe0',
177                    'name': 'xe0'},
178                   {'virtual-interface':
179                    {'dst_mac': '00:00:00:00:00:03',
180                     'vpci': '0000:05:00.1',
181                     'local_ip': '152.16.40.19',
182                     'type': 'PCI-PASSTHROUGH',
183                     'driver': "i40e",
184                     'netmask': '255.255.255.0',
185                     'dpdk_port_num': '1',
186                     'bandwidth': '10 Gbps',
187                     'dst_ip': '152.16.40.20',
188                     'local_iface_name': 'xe1',
189                     'local_mac': '00:00:00:00:00:01'},
190                    'vnfd-connection-point-ref': 'xe1',
191                    'name': 'xe1'}]}],
192                'description': 'Vpe approximation using DPDK',
193                'mgmt-interface':
194                    {'vdu-id': 'vpevnf-baremetal',
195                     'host': '1.2.1.1',
196                     'password': 'r00t',
197                     'user': 'root',
198                     'ip': '1.2.1.1'},
199                'benchmark':
200                    {'kpi': ['packets_in', 'packets_fwd', 'packets_dropped']},
201                'connection-point': [{'type': 'VPORT', 'name': 'xe0'},
202                                     {'type': 'VPORT', 'name': 'xe1'}],
203                'id': 'CgnaptApproxVnf', 'name': 'VPEVnfSsh'}]}}
204
205     scenario_cfg = {'options': {'packetsize': 64, 'traffic_type': 4,
206                                 'rfc2544': {'allowed_drop_rate': '0.8 - 1'},
207                                 'vnf__1': {'rules': 'acl_1rule.yaml',
208                                            'vnf_config': {'lb_config': 'SW',
209                                                           'lb_count': 1,
210                                                           'worker_config':
211                                                           '1C/1T',
212                                                           'worker_threads': 1}}
213                                 },
214                     'task_id': 'a70bdf4a-8e67-47a3-9dc1-273c14506eb7',
215                     'task_path': '/tmp',
216                     'tc': 'tc_ipv4_1Mflow_64B_packetsize',
217                     'runner': {'object': 'NetworkServiceTestCase',
218                                'interval': 35,
219                                'output_filename': '/tmp/yardstick.out',
220                                'runner_id': 74476, 'duration': 400,
221                                'type': 'Duration'},
222                     'traffic_profile': 'ipv4_throughput_acl.yaml',
223                     'traffic_options': {'flow': 'ipv4_Packets_acl.yaml',
224                                         'imix': 'imix_voice.yaml'},
225                     'type': 'ISB',
226                     'nodes': {'tg__2': 'trafficgen_2.yardstick',
227                               'tg__1': 'trafficgen_1.yardstick',
228                               'vnf__1': 'vnf.yardstick'},
229                     'topology': 'vpe-tg-topology-baremetal.yaml'}
230
231     context_cfg = {'nodes': {'tg__2':
232                              {'member-vnf-index': '3',
233                               'role': 'TrafficGen',
234                               'name': 'trafficgen_2.yardstick',
235                               'vnfd-id-ref': 'tg__2',
236                               'ip': '1.2.1.1',
237                               'interfaces':
238                               {'xe0': {'local_iface_name': 'ens513f0',
239                                        'vld_id': 'public',
240                                        'netmask': '255.255.255.0',
241                                        'local_ip': '152.16.40.20',
242                                        'dst_mac': '00:00:00:00:00:01',
243                                        'local_mac': '00:00:00:00:00:03',
244                                        'dst_ip': '152.16.40.19',
245                                        'driver': 'ixgbe',
246                                        'vpci': '0000:02:00.0',
247                                        'dpdk_port_num': 0},
248                                'xe1': {'local_iface_name': 'ens513f1',
249                                        'netmask': '255.255.255.0',
250                                        'network': '202.16.100.0',
251                                        'local_ip': '202.16.100.20',
252                                        'local_mac': '00:1e:67:d0:60:5d',
253                                        'driver': 'ixgbe',
254                                        'vpci': '0000:02:00.1',
255                                        'dpdk_port_num': 1}},
256                               'password': 'r00t',
257                               'VNF model': 'l3fwd_vnf.yaml',
258                               'user': 'root'},
259                              'tg__1':
260                              {'member-vnf-index': '1',
261                               'role': 'TrafficGen',
262                               'name': 'trafficgen_1.yardstick',
263                               'vnfd-id-ref': 'tg__1',
264                               'ip': '1.2.1.1',
265                               'interfaces':
266                               {'xe0': {'local_iface_name': 'ens785f0',
267                                        'vld_id': 'private',
268                                        'netmask': '255.255.255.0',
269                                        'local_ip': '152.16.100.20',
270                                        'dst_mac': '00:00:00:00:00:02',
271                                        'local_mac': '00:00:00:00:00:04',
272                                        'dst_ip': '152.16.100.19',
273                                        'driver': 'i40e',
274                                        'vpci': '0000:05:00.0',
275                                        'dpdk_port_num': 0},
276                                'xe1': {'local_iface_name': 'ens785f1',
277                                        'netmask': '255.255.255.0',
278                                        'local_ip': '152.16.100.21',
279                                        'local_mac': '00:00:00:00:00:01',
280                                        'driver': 'i40e',
281                                        'vpci': '0000:05:00.1',
282                                        'dpdk_port_num': 1}},
283                               'password': 'r00t',
284                               'VNF model': 'tg_rfc2544_tpl.yaml',
285                               'user': 'root'},
286                              'vnf__1':
287                              {'name': 'vnf.yardstick',
288                               'vnfd-id-ref': 'vnf__1',
289                               'ip': '1.2.1.1',
290                               'interfaces':
291                               {'xe0': {'local_iface_name': 'ens786f0',
292                                        'vld_id': 'private',
293                                        'netmask': '255.255.255.0',
294                                        'local_ip': '152.16.100.19',
295                                        'dst_mac': '00:00:00:00:00:04',
296                                        'local_mac': '00:00:00:00:00:02',
297                                        'dst_ip': '152.16.100.20',
298                                        'driver': 'i40e',
299                                        'vpci': '0000:05:00.0',
300                                        'dpdk_port_num': 0},
301                                'xe1': {'local_iface_name': 'ens786f1',
302                                        'vld_id': 'public',
303                                        'netmask': '255.255.255.0',
304                                        'local_ip': '152.16.40.19',
305                                        'dst_mac': '00:00:00:00:00:03',
306                                        'local_mac': '00:00:00:00:00:01',
307                                        'dst_ip': '152.16.40.20',
308                                        'driver': 'i40e',
309                                        'vpci': '0000:05:00.1',
310                                        'dpdk_port_num': 1}},
311                               'routing_table':
312                               [{'netmask': '255.255.255.0',
313                                 'gateway': '152.16.100.20',
314                                 'network': '152.16.100.20',
315                                 'if': 'xe0'},
316                                {'netmask': '255.255.255.0',
317                                 'gateway': '152.16.40.20',
318                                 'network': '152.16.40.20',
319                                 'if': 'xe1'}],
320                               'member-vnf-index': '2',
321                               'host': '1.2.1.1',
322                               'role': 'vnf',
323                               'user': 'root',
324                               'nd_route_tbl':
325                               [{'netmask': '112',
326                                 'gateway': '0064:ff9b:0:0:0:0:9810:6414',
327                                 'network': '0064:ff9b:0:0:0:0:9810:6414',
328                                 'if': 'xe0'},
329                                {'netmask': '112',
330                                 'gateway': '0064:ff9b:0:0:0:0:9810:2814',
331                                 'network': '0064:ff9b:0:0:0:0:9810:2814',
332                                 'if': 'xe1'}],
333                               'password': 'r00t',
334                               'VNF model': 'cgnapt_vnf.yaml'}}}
335
336     def test___init__(self, mock_process):
337         vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
338         cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
339         self.assertIsNone(cgnapt_approx_vnf._vnf_process)
340
341     @mock.patch("yardstick.network_services.vnf_generic.vnf.sample_vnf.time")
342     def test_collect_kpi(self, mock_time, mock_process):
343         with mock.patch("yardstick.ssh.SSH") as ssh:
344             ssh_mock = mock.Mock(autospec=ssh.SSH)
345             ssh_mock.execute = \
346                 mock.Mock(return_value=(0, "", ""))
347             ssh.from_node.return_value = ssh_mock
348             vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
349             cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
350             cgnapt_approx_vnf.q_in = mock.MagicMock()
351             cgnapt_approx_vnf.q_out = mock.MagicMock()
352             cgnapt_approx_vnf.q_out.qsize = mock.Mock(return_value=0)
353             cgnapt_approx_vnf.resource = mock.Mock(autospec=ResourceProfile)
354             result = {'packets_dropped': 0, 'packets_fwd': 0, 'packets_in': 0}
355             self.assertEqual(result, cgnapt_approx_vnf.collect_kpi())
356
357     @mock.patch("yardstick.network_services.vnf_generic.vnf.sample_vnf.time")
358     def test_vnf_execute_command(self, mock_time, mock_process):
359         with mock.patch("yardstick.ssh.SSH") as ssh:
360             ssh_mock = mock.Mock(autospec=ssh.SSH)
361             ssh_mock.execute = \
362                 mock.Mock(return_value=(0, "", ""))
363             ssh.from_node.return_value = ssh_mock
364             vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
365             cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
366             cgnapt_approx_vnf.q_in = mock.MagicMock()
367             cgnapt_approx_vnf.q_out = mock.MagicMock()
368             cgnapt_approx_vnf.q_out.qsize = mock.Mock(return_value=0)
369             cmd = "quit"
370             self.assertEqual("", cgnapt_approx_vnf.vnf_execute(cmd))
371
372     def test_get_stats(self, mock_process):
373         with mock.patch("yardstick.ssh.SSH") as ssh:
374             ssh_mock = mock.Mock(autospec=ssh.SSH)
375             ssh_mock.execute = \
376                 mock.Mock(return_value=(0, "", ""))
377             ssh.from_node.return_value = ssh_mock
378             vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
379             cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
380             cgnapt_approx_vnf.q_in = mock.MagicMock()
381             cgnapt_approx_vnf.q_out = mock.MagicMock()
382             cgnapt_approx_vnf.q_out.qsize = mock.Mock(return_value=0)
383             mock_result = \
384                 "CG-NAPT(.*\n)*Received 100, Missed 0, Dropped 0,Translated 100,ingress"
385             cgnapt_approx_vnf.vnf_execute = mock.Mock(return_value=mock_result)
386             self.assertListEqual(list(mock_result), list(cgnapt_approx_vnf.get_stats()))
387
388     def _get_file_abspath(self, filename):
389         curr_path = os.path.dirname(os.path.abspath(__file__))
390         file_path = os.path.join(curr_path, filename)
391         return file_path
392
393     @mock.patch("yardstick.network_services.vnf_generic.vnf.cgnapt_vnf.hex")
394     @mock.patch("yardstick.network_services.vnf_generic.vnf.cgnapt_vnf.eval")
395     @mock.patch('yardstick.network_services.vnf_generic.vnf.cgnapt_vnf.open')
396     def test_run_vcgnapt(self, hex, eval, mock_open, mock_process):
397         with mock.patch("yardstick.ssh.SSH") as ssh:
398             ssh_mock = mock.Mock(autospec=ssh.SSH)
399             ssh_mock.execute = \
400                 mock.Mock(return_value=(0, "", ""))
401             ssh_mock.run = \
402                 mock.Mock(return_value=(0, "", ""))
403             ssh.from_node.return_value = ssh_mock
404             vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
405             cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
406             cgnapt_approx_vnf._build_config = mock.MagicMock()
407             cgnapt_approx_vnf.queue_wrapper = mock.MagicMock()
408             cgnapt_approx_vnf.ssh_helper = mock.MagicMock()
409             cgnapt_approx_vnf.ssh_helper.run = mock.MagicMock()
410             cgnapt_approx_vnf.scenario_helper.scenario_cfg = self.scenario_cfg
411             cgnapt_approx_vnf._run()
412             cgnapt_approx_vnf.ssh_helper.run.assert_called_once()
413
414     @mock.patch("yardstick.network_services.vnf_generic.vnf.sample_vnf.Context")
415     def test_instantiate(self, mock_context, mock_process):
416         with mock.patch("yardstick.ssh.SSH") as ssh:
417             ssh_mock = mock.Mock(autospec=ssh.SSH)
418             ssh_mock.execute = \
419                 mock.Mock(return_value=(0, "", ""))
420             ssh.from_node.return_value = ssh_mock
421             vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
422             cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
423             cgnapt_approx_vnf.ssh_helper = ssh
424             cgnapt_approx_vnf.deploy_helper = mock.MagicMock()
425             cgnapt_approx_vnf.resource_helper = mock.MagicMock()
426             cgnapt_approx_vnf._build_config = mock.MagicMock()
427             self.scenario_cfg['vnf_options'] = {'acl': {'cfg': "",
428                                                         'rules': ""}}
429             cgnapt_approx_vnf.q_out.put("pipeline>")
430             cgnapt_vnf.WAIT_TIME = 3
431             self.scenario_cfg.update({"nodes": {"vnf__1": ""}})
432             self.assertIsNone(cgnapt_approx_vnf.instantiate(self.scenario_cfg,
433                                                             self.context_cfg))
434
435     def test_scale(self, mock_process):
436         vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
437         cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
438         flavor = ""
439         self.assertRaises(NotImplementedError, cgnapt_approx_vnf.scale, flavor)
440
441     @mock.patch("yardstick.network_services.vnf_generic.vnf.sample_vnf.time")
442     def test_terminate(self, mock_time, mock_process):
443         with mock.patch("yardstick.ssh.SSH") as ssh:
444             ssh_mock = mock.Mock(autospec=ssh.SSH)
445             ssh_mock.execute = \
446                 mock.Mock(return_value=(0, "", ""))
447             ssh.from_node.return_value = ssh_mock
448             vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
449             cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
450             cgnapt_approx_vnf._vnf_process = mock.MagicMock()
451             cgnapt_approx_vnf._vnf_process.terminate = mock.Mock()
452             cgnapt_approx_vnf.used_drivers = {"01:01.0": "i40e",
453                                               "01:01.1": "i40e"}
454             cgnapt_approx_vnf.vnf_execute = mock.MagicMock()
455             cgnapt_approx_vnf.ssh_helper = ssh_mock
456             cgnapt_approx_vnf.dpdk_nic_bind = "dpdk_nic_bind.py"
457             cgnapt_approx_vnf._resource_collect_stop = mock.Mock()
458             self.assertEqual(None, cgnapt_approx_vnf.terminate())
459
460     @mock.patch("yardstick.network_services.vnf_generic.vnf.sample_vnf.time")
461     @mock.patch("yardstick.network_services.vnf_generic.vnf.cgnapt_vnf.time")
462     def test__vnf_up_post(self, mock_time, mock_cgnapt_time, mock_process):
463         with mock.patch("yardstick.ssh.SSH") as ssh:
464             ssh_mock = mock.Mock(autospec=ssh.SSH)
465             ssh_mock.execute = \
466                 mock.Mock(return_value=(0, "", ""))
467             ssh.from_node.return_value = ssh_mock
468             vnfd = self.VNFD['vnfd:vnfd-catalog']['vnfd'][0]
469             cgnapt_approx_vnf = CgnaptApproxVnf(name, vnfd)
470             cgnapt_approx_vnf._vnf_process = mock.MagicMock()
471             cgnapt_approx_vnf._vnf_process.terminate = mock.Mock()
472             cgnapt_approx_vnf.vnf_execute = mock.MagicMock()
473             cgnapt_approx_vnf.ssh_helper = ssh_mock
474             cgnapt_approx_vnf.scenario_helper.scenario_cfg = self.scenario_cfg
475             cgnapt_approx_vnf._resource_collect_stop = mock.Mock()
476             cgnapt_approx_vnf._vnf_up_post()
477             cgnapt_approx_vnf.vnf_execute.assert_called_once()
478
479
480 if __name__ == '__main__':
481     unittest.main()