Coverage for n2vc/tests/unit/test_n2vc_juju_conn.py: 98%

1# Copyright 2020 Canonical Ltd. 

2# 

3# Licensed under the Apache License, Version 2.0 (the "License"); 

4# you may not use this file except in compliance with the License. 

5# You may obtain a copy of the License at 

6# 

7# http://www.apache.org/licenses/LICENSE-2.0 

8# 

9# Unless required by applicable law or agreed to in writing, software 

10# distributed under the License is distributed on an "AS IS" BASIS, 

11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

12# See the License for the specific language governing permissions and 

13# limitations under the License. 

14 

15 

16import asyncio 

17import logging 

18from unittest.mock import Mock, MagicMock 

19from unittest.mock import patch 

20 

21 

22import asynctest 

23from n2vc.definitions import Offer, RelationEndpoint 

24from n2vc.n2vc_juju_conn import N2VCJujuConnector 

25from osm_common import fslocal 

26from osm_common.dbmemory import DbMemory 

27from n2vc.exceptions import ( 

28 N2VCBadArgumentsException, 

29 N2VCException, 

30 JujuApplicationNotFound, 

31) 

32from n2vc.tests.unit.utils import AsyncMock 

33from n2vc.vca.connection_data import ConnectionData 

34from n2vc.tests.unit.testdata import test_db_descriptors as descriptors 

35import yaml 

36 

37 

38class N2VCJujuConnTestCase(asynctest.TestCase): 

39 @asynctest.mock.patch("n2vc.n2vc_juju_conn.MotorStore") 

40 @asynctest.mock.patch("n2vc.n2vc_juju_conn.get_connection") 

41 @asynctest.mock.patch("n2vc.vca.connection_data.base64_to_cacert") 

42 def setUp( 

43 self, mock_base64_to_cacert=None, mock_get_connection=None, mock_store=None 

44 ): 

45 self.loop = asyncio.get_event_loop() 

46 self.db = Mock() 

47 mock_base64_to_cacert.return_value = """ 

48 -----BEGIN CERTIFICATE----- 

49 SOMECERT 

50 -----END CERTIFICATE-----""" 

51 mock_store.return_value = AsyncMock() 

52 mock_vca_connection = Mock() 

53 mock_get_connection.return_value = mock_vca_connection 

54 mock_vca_connection.data.return_value = ConnectionData( 

55 **{ 

56 "endpoints": ["1.2.3.4:17070"], 

57 "user": "user", 

58 "secret": "secret", 

59 "cacert": "cacert", 

60 "pubkey": "pubkey", 

61 "lxd-cloud": "cloud", 

62 "lxd-credentials": "credentials", 

63 "k8s-cloud": "k8s_cloud", 

64 "k8s-credentials": "k8s_credentials", 

65 "model-config": {}, 

66 "api-proxy": "api_proxy", 

67 } 

68 ) 

69 logging.disable(logging.CRITICAL) 

70 

71 N2VCJujuConnector.get_public_key = Mock() 

72 self.n2vc = N2VCJujuConnector( 

73 db=self.db, 

74 fs=fslocal.FsLocal(), 

75 log=None, 

76 on_update_db=None, 

77 ) 

78 N2VCJujuConnector.get_public_key.assert_not_called() 

79 self.n2vc.libjuju = Mock() 

80 

81 

82class GetMetricssTest(N2VCJujuConnTestCase): 

83 def setUp(self): 

84 super(GetMetricssTest, self).setUp() 

85 self.n2vc.libjuju.get_metrics = AsyncMock() 

86 

87 def test_success(self): 

88 _ = self.loop.run_until_complete(self.n2vc.get_metrics("model", "application")) 

89 self.n2vc.libjuju.get_metrics.assert_called_once() 

90 

91 def test_except(self): 

92 self.n2vc.libjuju.get_metrics.side_effect = Exception() 

93 with self.assertRaises(Exception): 

94 _ = self.loop.run_until_complete( 

95 self.n2vc.get_metrics("model", "application") 

96 ) 

97 self.n2vc.libjuju.get_metrics.assert_called_once() 

98 

99 

100class UpdateVcaStatusTest(N2VCJujuConnTestCase): 

101 def setUp(self): 

102 super(UpdateVcaStatusTest, self).setUp() 

103 self.n2vc.libjuju.get_controller = AsyncMock() 

104 self.n2vc.libjuju.get_model = AsyncMock() 

105 self.n2vc.libjuju.get_executed_actions = AsyncMock() 

106 self.n2vc.libjuju.get_actions = AsyncMock() 

107 self.n2vc.libjuju.get_application_configs = AsyncMock() 

108 self.n2vc.libjuju._get_application = AsyncMock() 

109 

110 def test_success( 

111 self, 

112 ): 

113 self.loop.run_until_complete( 

114 self.n2vc.update_vca_status( 

115 {"model": {"applications": {"app": {"actions": {}}}}} 

116 ) 

117 ) 

118 self.n2vc.libjuju.get_executed_actions.assert_called_once() 

119 self.n2vc.libjuju.get_actions.assert_called_once() 

120 self.n2vc.libjuju.get_application_configs.assert_called_once() 

121 

122 def test_exception(self): 

123 self.n2vc.libjuju.get_model.return_value = None 

124 self.n2vc.libjuju.get_executed_actions.side_effect = Exception() 

125 with self.assertRaises(Exception): 

126 self.loop.run_until_complete( 

127 self.n2vc.update_vca_status( 

128 {"model": {"applications": {"app": {"actions": {}}}}} 

129 ) 

130 ) 

131 self.n2vc.libjuju.get_executed_actions.assert_not_called() 

132 self.n2vc.libjuju.get_actions.assert_not_called_once() 

133 self.n2vc.libjuju.get_application_configs.assert_not_called_once() 

134 

135 

136class K8sProxyCharmsTest(N2VCJujuConnTestCase): 

137 def setUp(self): 

138 super(K8sProxyCharmsTest, self).setUp() 

139 self.n2vc.libjuju.model_exists = AsyncMock() 

140 self.n2vc.libjuju.add_model = AsyncMock() 

141 self.n2vc.libjuju.deploy_charm = AsyncMock() 

142 self.n2vc.libjuju.model_exists.return_value = False 

143 self.db = DbMemory() 

144 self.fs = fslocal.FsLocal() 

145 self.fs.path = "/" 

146 self.n2vc.fs = self.fs 

147 self.n2vc.db = self.db 

148 self.db.create_list("nsrs", yaml.safe_load(descriptors.db_nsrs_text)) 

149 self.db.create_list("vnfrs", yaml.safe_load(descriptors.db_vnfrs_text)) 

150 

151 @patch( 

152 "n2vc.n2vc_juju_conn.generate_random_alfanum_string", 

153 **{"return_value": "random"} 

154 ) 

155 def test_success(self, mock_generate_random_alfanum_string): 

156 self.n2vc.fs.file_exists = MagicMock(create_autospec=True) 

157 self.n2vc.fs.file_exists.return_value = True 

158 ee_id = self.loop.run_until_complete( 

159 self.n2vc.install_k8s_proxy_charm( 

160 "simple", 

161 ".dbfbd751-3de4-4e68-bd40-ec5ae0a53898.1b6a4eb3-4fbf-415e-985c-4aeb3161eec0-0", 

162 "path", 

163 {}, 

164 ) 

165 ) 

166 

167 self.n2vc.libjuju.add_model.assert_called_once() 

168 self.n2vc.libjuju.deploy_charm.assert_called_once_with( 

169 model_name="dbfbd751-3de4-4e68-bd40-ec5ae0a53898-k8s", 

170 application_name="simple-ee-z0-vnf1-vnf", 

171 path="//path", 

172 machine_id=None, 

173 db_dict={}, 

174 progress_timeout=None, 

175 total_timeout=None, 

176 config=None, 

177 ) 

178 self.assertEqual( 

179 ee_id, "dbfbd751-3de4-4e68-bd40-ec5ae0a53898-k8s.simple-ee-z0-vnf1-vnf.k8s" 

180 ) 

181 

182 def test_no_artifact_path( 

183 self, 

184 ): 

185 with self.assertRaises(N2VCBadArgumentsException): 

186 ee_id = self.loop.run_until_complete( 

187 self.n2vc.install_k8s_proxy_charm( 

188 "simple", 

189 ".dbfbd751-3de4-4e68-bd40-ec5ae0a53898.1b6a4eb3-4fbf-415e-985c-4aeb3161eec0-0", 

190 "", 

191 {}, 

192 ) 

193 ) 

194 self.assertIsNone(ee_id) 

195 

196 def test_no_db( 

197 self, 

198 ): 

199 with self.assertRaises(N2VCBadArgumentsException): 

200 ee_id = self.loop.run_until_complete( 

201 self.n2vc.install_k8s_proxy_charm( 

202 "simple", 

203 ".dbfbd751-3de4-4e68-bd40-ec5ae0a53898.1b6a4eb3-4fbf-415e-985c-4aeb3161eec0-0", 

204 "path", 

205 None, 

206 ) 

207 ) 

208 self.assertIsNone(ee_id) 

209 

210 def test_file_not_exists( 

211 self, 

212 ): 

213 self.n2vc.fs.file_exists = MagicMock(create_autospec=True) 

214 self.n2vc.fs.file_exists.return_value = False 

215 with self.assertRaises(N2VCBadArgumentsException): 

216 ee_id = self.loop.run_until_complete( 

217 self.n2vc.install_k8s_proxy_charm( 

218 "simple", 

219 ".dbfbd751-3de4-4e68-bd40-ec5ae0a53898.1b6a4eb3-4fbf-415e-985c-4aeb3161eec0-0", 

220 "path", 

221 {}, 

222 ) 

223 ) 

224 self.assertIsNone(ee_id) 

225 

226 def test_exception( 

227 self, 

228 ): 

229 self.n2vc.fs.file_exists = MagicMock(create_autospec=True) 

230 self.n2vc.fs.file_exists.return_value = True 

231 self.n2vc.fs.path = MagicMock(create_autospec=True) 

232 self.n2vc.fs.path.return_value = "path" 

233 self.n2vc.libjuju.deploy_charm.side_effect = Exception() 

234 with self.assertRaises(N2VCException): 

235 ee_id = self.loop.run_until_complete( 

236 self.n2vc.install_k8s_proxy_charm( 

237 "simple", 

238 ".dbfbd751-3de4-4e68-bd40-ec5ae0a53898.1b6a4eb3-4fbf-415e-985c-4aeb3161eec0-0", 

239 "path", 

240 {}, 

241 ) 

242 ) 

243 self.assertIsNone(ee_id) 

244 

245 

246class AddRelationTest(N2VCJujuConnTestCase): 

247 def setUp(self): 

248 super(AddRelationTest, self).setUp() 

249 self.n2vc.libjuju.add_relation = AsyncMock() 

250 self.n2vc.libjuju.offer = AsyncMock() 

251 self.n2vc.libjuju.get_controller = AsyncMock() 

252 self.n2vc.libjuju.consume = AsyncMock() 

253 

254 def test_standard_relation_same_model_and_controller(self): 

255 relation_endpoint_1 = RelationEndpoint("model-1.app1.0", None, "endpoint1") 

256 relation_endpoint_2 = RelationEndpoint("model-1.app2.1", None, "endpoint2") 

257 self.loop.run_until_complete( 

258 self.n2vc.add_relation(relation_endpoint_1, relation_endpoint_2) 

259 ) 

260 self.n2vc.libjuju.add_relation.assert_called_once_with( 

261 model_name="model-1", 

262 endpoint_1="app1:endpoint1", 

263 endpoint_2="app2:endpoint2", 

264 ) 

265 self.n2vc.libjuju.offer.assert_not_called() 

266 self.n2vc.libjuju.consume.assert_not_called() 

267 

268 def test_cmr_relation_same_controller(self): 

269 relation_endpoint_1 = RelationEndpoint("model-1.app1.0", None, "endpoint") 

270 relation_endpoint_2 = RelationEndpoint("model-2.app2.1", None, "endpoint") 

271 offer = Offer("admin/model-1.app1") 

272 self.n2vc.libjuju.offer.return_value = offer 

273 self.n2vc.libjuju.consume.return_value = "saas" 

274 self.loop.run_until_complete( 

275 self.n2vc.add_relation(relation_endpoint_1, relation_endpoint_2) 

276 ) 

277 self.n2vc.libjuju.offer.assert_called_once_with(relation_endpoint_1) 

278 self.n2vc.libjuju.consume.assert_called_once() 

279 self.n2vc.libjuju.add_relation.assert_called_once_with( 

280 "model-2", "app2:endpoint", "saas" 

281 ) 

282 

283 def test_cmr_relation_different_controller(self): 

284 self.n2vc._get_libjuju = AsyncMock(return_value=self.n2vc.libjuju) 

285 relation_endpoint_1 = RelationEndpoint( 

286 "model-1.app1.0", "vca-id-1", "endpoint1" 

287 ) 

288 relation_endpoint_2 = RelationEndpoint( 

289 "model-1.app2.1", "vca-id-2", "endpoint2" 

290 ) 

291 offer = Offer("admin/model-1.app1") 

292 self.n2vc.libjuju.offer.return_value = offer 

293 self.n2vc.libjuju.consume.return_value = "saas" 

294 self.loop.run_until_complete( 

295 self.n2vc.add_relation(relation_endpoint_1, relation_endpoint_2) 

296 ) 

297 self.n2vc.libjuju.offer.assert_called_once_with(relation_endpoint_1) 

298 self.n2vc.libjuju.consume.assert_called_once() 

299 self.n2vc.libjuju.add_relation.assert_called_once_with( 

300 "model-1", "app2:endpoint2", "saas" 

301 ) 

302 

303 def test_relation_exception(self): 

304 relation_endpoint_1 = RelationEndpoint("model-1.app1.0", None, "endpoint") 

305 relation_endpoint_2 = RelationEndpoint("model-2.app2.1", None, "endpoint") 

306 self.n2vc.libjuju.offer.side_effect = Exception() 

307 with self.assertRaises(N2VCException): 

308 self.loop.run_until_complete( 

309 self.n2vc.add_relation(relation_endpoint_1, relation_endpoint_2) 

310 ) 

311 

312 

313class UpgradeCharmTest(N2VCJujuConnTestCase): 

314 def setUp(self): 

315 super(UpgradeCharmTest, self).setUp() 

316 self.n2vc._get_libjuju = AsyncMock(return_value=self.n2vc.libjuju) 

317 N2VCJujuConnector._get_ee_id_components = Mock() 

318 self.n2vc.libjuju.upgrade_charm = AsyncMock() 

319 

320 def test_empty_ee_id(self): 

321 with self.assertRaises(N2VCBadArgumentsException): 

322 self.loop.run_until_complete( 

323 self.n2vc.upgrade_charm( 

324 "", "/sample_charm_path", "sample_charm_id", "native-charm", None 

325 ) 

326 ) 

327 self.n2vc._get_libjuju.assert_called() 

328 self.n2vc._get_ee_id_components.assert_not_called() 

329 self.n2vc.libjuju.upgrade_charm.assert_not_called() 

330 

331 def test_wrong_ee_id(self): 

332 N2VCJujuConnector._get_ee_id_components.side_effect = Exception 

333 with self.assertRaises(N2VCBadArgumentsException): 

334 self.loop.run_until_complete( 

335 self.n2vc.upgrade_charm( 

336 "ns-id-k8s.app-vnf-vnf-id-vdu-vdu-random.k8s", 

337 "/sample_charm_path", 

338 "sample_charm_id", 

339 "native-charm", 

340 500, 

341 ) 

342 ) 

343 self.n2vc._get_libjuju.assert_called() 

344 self.n2vc._get_ee_id_components.assert_called() 

345 self.n2vc.libjuju.upgrade_charm.assert_not_called() 

346 

347 def test_charm_upgrade_succeded(self): 

348 N2VCJujuConnector._get_ee_id_components.return_value = ( 

349 "sample_model", 

350 "sample_app", 

351 "sample_machine_id", 

352 ) 

353 self.loop.run_until_complete( 

354 self.n2vc.upgrade_charm( 

355 "ns-id-k8s.app-vnf-vnf-id-vdu-vdu-random.k8s", 

356 "/sample_charm_path", 

357 "sample_charm_id", 

358 "native-charm", 

359 500, 

360 ) 

361 ) 

362 self.n2vc._get_libjuju.assert_called() 

363 self.n2vc._get_ee_id_components.assert_called() 

364 self.n2vc.libjuju.upgrade_charm.assert_called_with( 

365 application_name="sample_app", 

366 path="/sample_charm_path", 

367 model_name="sample_model", 

368 total_timeout=500, 

369 ) 

370 

371 def test_charm_upgrade_failed(self): 

372 N2VCJujuConnector._get_ee_id_components.return_value = ( 

373 "sample_model", 

374 "sample_app", 

375 "sample_machine_id", 

376 ) 

377 self.n2vc.libjuju.upgrade_charm.side_effect = JujuApplicationNotFound 

378 with self.assertRaises(N2VCException): 

379 self.loop.run_until_complete( 

380 self.n2vc.upgrade_charm( 

381 "ns-id-k8s.app-vnf-vnf-id-vdu-vdu-random.k8s", 

382 "/sample_charm_path", 

383 "sample_charm_id", 

384 "native-charm", 

385 None, 

386 ) 

387 ) 

388 self.n2vc._get_libjuju.assert_called() 

389 self.n2vc._get_ee_id_components.assert_called() 

390 self.n2vc.libjuju.upgrade_charm.assert_called_with( 

391 application_name="sample_app", 

392 path="/sample_charm_path", 

393 model_name="sample_model", 

394 total_timeout=None, 

395 ) 

396 

397 

398class GenerateApplicationNameTest(N2VCJujuConnTestCase): 

399 vnf_id = "dbfbd751-3de4-4e68-bd40-ec5ae0a53898" 

400 

401 def setUp(self): 

402 super(GenerateApplicationNameTest, self).setUp() 

403 self.db = MagicMock(DbMemory) 

404 

405 @patch( 

406 "n2vc.n2vc_juju_conn.generate_random_alfanum_string", 

407 **{"return_value": "random"} 

408 ) 

409 def test_generate_backward_compatible_application_name( 

410 self, mock_generate_random_alfanum 

411 ): 

412 vdu_id = "mgmtVM" 

413 vdu_count = "0" 

414 expected_result = "app-vnf-ec5ae0a53898-vdu-mgmtVM-cnt-0-random" 

415 

416 application_name = self.n2vc._generate_backward_compatible_application_name( 

417 GenerateApplicationNameTest.vnf_id, vdu_id, vdu_count 

418 ) 

419 self.assertEqual(application_name, expected_result) 

420 

421 @patch( 

422 "n2vc.n2vc_juju_conn.generate_random_alfanum_string", 

423 **{"return_value": "random"} 

424 ) 

425 def test_generate_backward_compatible_application_name_without_vnf_id_vdu_id( 

426 self, mock_generate_random_alfanum 

427 ): 

428 vnf_id = None 

429 vdu_id = "" 

430 vdu_count = None 

431 expected_result = "app--random" 

432 application_name = self.n2vc._generate_backward_compatible_application_name( 

433 vnf_id, vdu_id, vdu_count 

434 ) 

435 

436 self.assertEqual(application_name, expected_result) 

437 self.assertLess(len(application_name), 50) 

438 

439 def test_find_charm_level_with_vnf_id(self): 

440 vdu_id = "" 

441 expected_result = "vnf-level" 

442 charm_level = self.n2vc._find_charm_level( 

443 GenerateApplicationNameTest.vnf_id, vdu_id 

444 ) 

445 self.assertEqual(charm_level, expected_result) 

446 

447 def test_find_charm_level_with_vdu_id(self): 

448 vnf_id = "" 

449 vdu_id = "mgmtVM" 

450 with self.assertRaises(N2VCException): 

451 self.n2vc._find_charm_level(vnf_id, vdu_id) 

452 

453 def test_find_charm_level_with_vnf_id_and_vdu_id(self): 

454 vdu_id = "mgmtVM" 

455 expected_result = "vdu-level" 

456 charm_level = self.n2vc._find_charm_level( 

457 GenerateApplicationNameTest.vnf_id, vdu_id 

458 ) 

459 self.assertEqual(charm_level, expected_result) 

460 

461 def test_find_charm_level_without_vnf_id_and_vdu_id(self): 

462 vnf_id = "" 

463 vdu_id = "" 

464 expected_result = "ns-level" 

465 charm_level = self.n2vc._find_charm_level(vnf_id, vdu_id) 

466 self.assertEqual(charm_level, expected_result) 

467 

468 def test_generate_application_name_ns_charm(self): 

469 charm_level = "ns-level" 

470 vnfrs = {} 

471 vca_records = [ 

472 { 

473 "target_element": "ns", 

474 "member-vnf-index": "", 

475 "vdu_id": None, 

476 "kdu_name": None, 

477 "vdu_count_index": None, 

478 "vnfd_id": None, 

479 "vdu_name": None, 

480 "type": "proxy_charm", 

481 "ee_descriptor_id": None, 

482 "charm_name": "simple-ns-charm-abc-000-rrrr-nnnn-4444-hhh-3333-yyyy-333-hhh", 

483 "ee_id": None, 

484 "application": "", 

485 "model": "dbfbd751-3de4-4e68-bd40-ec5ae0a53898", 

486 } 

487 ] 

488 vnf_count = "" 

489 vdu_count = "" 

490 vdu_id = None 

491 expected_result = "simple-ns-charm-abc-000-rrrr-nnnn-4444-h-ns" 

492 application_name = self.n2vc._generate_application_name( 

493 charm_level, 

494 vnfrs, 

495 vca_records, 

496 vnf_count=vnf_count, 

497 vdu_id=vdu_id, 

498 vdu_count=vdu_count, 

499 ) 

500 self.assertEqual(application_name, expected_result) 

501 self.assertLess(len(application_name), 50) 

502 

503 def test_generate_application_name_ns_charm_empty_vca_records(self): 

504 charm_level = "ns-level" 

505 vnfrs = {} 

506 vca_records = [] 

507 vnf_count = "" 

508 vdu_count = "" 

509 vdu_id = None 

510 with self.assertRaises(N2VCException): 

511 self.n2vc._generate_application_name( 

512 charm_level, 

513 vnfrs, 

514 vca_records, 

515 vnf_count=vnf_count, 

516 vdu_id=vdu_id, 

517 vdu_count=vdu_count, 

518 ) 

519 

520 def test_generate_application_name_vnf_charm(self): 

521 charm_level = "vnf-level" 

522 vnfrs = {"member-vnf-index-ref": "vnf111-xxx-yyy-zzz"} 

523 vca_records = [ 

524 { 

525 "target_element": "vnf/vnf1", 

526 "member-vnf-index": "vnf111-xxx-yyy-zzz", 

527 "vnfd_id": "r7fbd751-3de4-4e68-bd40-ec5ae0a53898", 

528 "ee_descriptor_id": "simple-ee-abc-000-rrrr-nnnn-4444-hhh-3333-yyyy-333-hhh-ttt-444", 

529 "charm_name": "", 

530 "model": "dbfbd751-3de4-4e68-bd40-ec5ae0a53898", 

531 } 

532 ] 

533 vnf_count = "1" 

534 vdu_count = "" 

535 vdu_id = None 

536 expected_result = "simple-ee-ab-1-vnf111-xxx-y-vnf" 

537 application_name = self.n2vc._generate_application_name( 

538 charm_level, 

539 vnfrs, 

540 vca_records, 

541 vnf_count=vnf_count, 

542 vdu_id=vdu_id, 

543 vdu_count=vdu_count, 

544 ) 

545 self.assertEqual(application_name, expected_result) 

546 self.assertLess(len(application_name), 50) 

547 

548 def test_generate_application_name_vdu_charm_kdu_name_in_vca_record_is_none(self): 

549 charm_level = "vdu-level" 

550 vnfrs = { 

551 "member-vnf-index-ref": "vnf111-xxx-yyy-zzz", 

552 "vdur": [ 

553 {"_id": "38912ff7-5bdd-4228-911f-c2bee259c44a", "vdu-id-ref": "mgmtVM"}, 

554 {"_id": "45512ff7-5bdd-4228-911f-c2bee259c44a", "vdu-id-ref": "dataVM"}, 

555 ], 

556 } 

557 vca_records = [ 

558 { 

559 "target_element": "vnf/vnf1/mgmtvm", 

560 "member-vnf-index": "vnf111-xxx-yyy-zzz", 

561 "vdu_id": "mgmtVM", 

562 "kdu_name": None, 

563 "vdu_count_index": None, 

564 "vnfd_id": "r7fbd751-3de4-4e68-bd40-ec5ae0a53898", 

565 "vdu_name": "mgmtvm", 

566 "ee_descriptor_id": "simple-ee-abc-000-rrrr-nnnn-4444-hhh-3333-yyyy-333-hhh-ttt-444", 

567 "charm_name": "", 

568 "model": "dbfbd751-3de4-4e68-bd40-ec5ae0a53898", 

569 }, 

570 { 

571 "target_element": "vnf/vnf1/dataVM", 

572 "member-vnf-index": "vnf111-xxx-yyy-zzz", 

573 "vdu_id": "dataVM", 

574 "kdu_name": None, 

575 "vdu_count_index": None, 

576 "vnfd_id": "r7fbd751-3de4-4e68-bd40-ec5ae0a53898", 

577 "vdu_name": "datavm", 

578 "ee_descriptor_id": "simple-ee-abc-000-rrrr-nnnn-8888-hhh-3333-yyyy-888-hhh-ttt-444", 

579 "charm_name": "", 

580 "model": "dbfbd751-3de4-4e68-bd40-ec5ae0a53898", 

581 }, 

582 ] 

583 vnf_count = "2" 

584 vdu_count = "0" 

585 vdu_id = "mgmtVM" 

586 expected_result = "simple-ee-ab-2-vnf111-xxx-y-mgmtVM-0-vdu" 

587 application_name = self.n2vc._generate_application_name( 

588 charm_level, 

589 vnfrs, 

590 vca_records, 

591 vnf_count=vnf_count, 

592 vdu_id=vdu_id, 

593 vdu_count=vdu_count, 

594 ) 

595 self.assertEqual(application_name, expected_result) 

596 self.assertLess(len(application_name), 50) 

597 

598 def test_generate_application_name_vdu_charm_vdu_id_kdu_name_in_vca_record_are_both_set( 

599 self, 

600 ): 

601 charm_level = "vdu-level" 

602 vnfrs = { 

603 "member-vnf-index-ref": "vnf111-xxx-yyy-zzz", 

604 "vdur": [ 

605 {"_id": "38912ff7-5bdd-4228-911f-c2bee259c44a", "vdu-id-ref": "mgmtVM"}, 

606 {"_id": "45512ff7-5bdd-4228-911f-c2bee259c44a", "vdu-id-ref": "dataVM"}, 

607 ], 

608 } 

609 vca_records = [ 

610 { 

611 "target_element": "vnf/vnf1/mgmtVM", 

612 "member-vnf-index": "vnf111-xxx-yyy-zzz", 

613 "vdu_id": "mgmtVM", 

614 "kdu_name": "mgmtVM", 

615 "vdu_count_index": None, 

616 "vnfd_id": "r7fbd751-3de4-4e68-bd40-ec5ae0a53898", 

617 "vdu_name": "mgmtvm", 

618 "ee_descriptor_id": "simple-ee-abc-000-rrrr-nnnn-4444-hhh-3333-yyyy-333-hhh-ttt-444", 

619 "charm_name": "", 

620 "model": "dbfbd751-3de4-4e68-bd40-ec5ae0a53898", 

621 }, 

622 { 

623 "target_element": "vnf/vnf1/dataVM", 

624 "member-vnf-index": "vnf111-xxx-yyy-zzz", 

625 "vdu_id": "dataVM", 

626 "kdu_name": None, 

627 "vdu_count_index": None, 

628 "vnfd_id": "r7fbd751-3de4-4e68-bd40-ec5ae0a53898", 

629 "vdu_name": "datavm", 

630 "ee_descriptor_id": "simple-ee-abc-000-rrrr-nnnn-8888-hhh-3333-yyyy-888-hhh-ttt-444", 

631 "charm_name": "", 

632 "model": "dbfbd751-3de4-4e68-bd40-ec5ae0a53898", 

633 }, 

634 ] 

635 vnf_count = "2" 

636 vdu_count = "0" 

637 vdu_id = "mgmtVM" 

638 expected_result = "simple-ee-ab-2-vnf111-xxx-y-mgmtVM-0-vdu" 

639 application_name = self.n2vc._generate_application_name( 

640 charm_level, 

641 vnfrs, 

642 vca_records, 

643 vnf_count=vnf_count, 

644 vdu_id=vdu_id, 

645 vdu_count=vdu_count, 

646 ) 

647 self.assertEqual(application_name, expected_result) 

648 self.assertLess(len(application_name), 50) 

649 

650 def test_generate_application_name_vdu_charm_both_vdu_id_kdu_name_in_vca_record_are_none( 

651 self, 

652 ): 

653 charm_level = "vdu-level" 

654 vnfrs = {"member-vnf-index-ref": "vnf111-xxx-yyy-zzz"} 

655 vca_records = [ 

656 { 

657 "target_element": "vnf/vnf1/mgmtVM", 

658 "member-vnf-index": "vnf111-xxx-yyy-zzz", 

659 "vdu_id": None, 

660 "kdu_name": None, 

661 "vdu_count_index": None, 

662 "vnfd_id": "r7fbd751-3de4-4e68-bd40-ec5ae0a53898", 

663 "vdu_name": "mgmtvm", 

664 "ee_descriptor_id": "simple-ee-abc-000-rrrr-nnnn-4444-hhh-3333-yyyy-333-hhh-ttt-444", 

665 "charm_name": "", 

666 "model": "dbfbd751-3de4-4e68-bd40-ec5ae0a53898", 

667 } 

668 ] 

669 vnf_count = "2" 

670 vdu_count = "0" 

671 vdu_id = "mgmtVM" 

672 with self.assertRaises(KeyError): 

673 self.n2vc._generate_application_name( 

674 charm_level, 

675 vnfrs, 

676 vca_records, 

677 vnf_count=vnf_count, 

678 vdu_id=vdu_id, 

679 vdu_count=vdu_count, 

680 ) 

681 

682 def test_generate_application_name_vdu_charm_given_vdu_id_is_none(self): 

683 charm_level = "vdu-level" 

684 vnfrs = {"member-vnf-index-ref": "vnf111-xxx-yyy-zzz"} 

685 vca_records = [ 

686 { 

687 "target_element": "vnf/vnf1/mgmtvVM", 

688 "member-vnf-index": "vnf111-xxx-yyy-zzz", 

689 "vdu_id": None, 

690 "kdu_name": "mgmtVM", 

691 "vdu_count_index": None, 

692 "vnfd_id": "r7fbd751-3de4-4e68-bd40-ec5ae0a53898", 

693 "vdu_name": "mgmtvm", 

694 "ee_descriptor_id": "simple-ee-abc-000-rrrr-nnnn-4444-hhh-3333-yyyy-333-hhh-ttt-444", 

695 "charm_name": "", 

696 "model": "dbfbd751-3de4-4e68-bd40-ec5ae0a53898", 

697 } 

698 ] 

699 vnf_count = "2" 

700 vdu_count = "0" 

701 vdu_id = None 

702 with self.assertRaises(N2VCException): 

703 self.n2vc._generate_application_name( 

704 charm_level, 

705 vnfrs, 

706 vca_records, 

707 vnf_count=vnf_count, 

708 vdu_id=vdu_id, 

709 vdu_count=vdu_count, 

710 ) 

711 

712 def test_generate_application_name_vdu_charm_vdu_id_does_not_match_with_the_key_in_vca_record( 

713 self, 

714 ): 

715 charm_level = "vdu-level" 

716 vnfrs = {"member-vnf-index-ref": "vnf111-xxx-yyy-zzz"} 

717 vca_records = [ 

718 { 

719 "target_element": "vnf/vnf1/mgmtVM", 

720 "member-vnf-index": "vnf111-xxx-yyy-zzz", 

721 "vdu_id": None, 

722 "kdu_name": "mgmtVM", 

723 "vdu_count_index": None, 

724 "vnfd_id": "r7fbd751-3de4-4e68-bd40-ec5ae0a53898", 

725 "vdu_name": "mgmtvm", 

726 "ee_descriptor_id": "simple-ee-abc-000-rrrr-nnnn-4444-hhh-3333-yyyy-333-hhh-ttt-444", 

727 "charm_name": "", 

728 "model": "dbfbd751-3de4-4e68-bd40-ec5ae0a53898", 

729 } 

730 ] 

731 vnf_count = "2" 

732 vdu_count = "0" 

733 vdu_id = "mgmtvm" 

734 with self.assertRaises(KeyError): 

735 self.n2vc._generate_application_name( 

736 charm_level, 

737 vnfrs, 

738 vca_records, 

739 vnf_count=vnf_count, 

740 vdu_id=vdu_id, 

741 vdu_count=vdu_count, 

742 ) 

743 

744 def test_generate_application_name_vdu_charm_vdu_id_in_vca_record_is_none(self): 

745 charm_level = "vdu-level" 

746 vnfrs = {"member-vnf-index-ref": "vnf111-xxx-yyy-zzz"} 

747 vca_records = [ 

748 { 

749 "target_element": "vnf/vnf1/mgmtVM", 

750 "member-vnf-index": "vnf111-xxx-yyy-zzz", 

751 "vdu_id": None, 

752 "kdu_name": "mgmtVM", 

753 "vdu_count_index": None, 

754 "vnfd_id": "r7fbd751-3de4-4e68-bd40-ec5ae0a53898", 

755 "vdu_name": "mgmtvm", 

756 "ee_descriptor_id": "simple-ee-abc-000-rrrr-nnnn-4444-hhh-3333-yyyy-333-hhh-ttt-444", 

757 "charm_name": "", 

758 "model": "dbfbd751-3de4-4e68-bd40-ec5ae0a53898", 

759 } 

760 ] 

761 vnf_count = "2" 

762 vdu_count = "0" 

763 vdu_id = "mgmtVM" 

764 expected_result = "simple-ee-ab-2-vnf111-xxx-y-mgmtVM-0-vdu" 

765 application_name = self.n2vc._generate_application_name( 

766 charm_level, 

767 vnfrs, 

768 vca_records, 

769 vnf_count=vnf_count, 

770 vdu_id=vdu_id, 

771 vdu_count=vdu_count, 

772 ) 

773 self.assertEqual(application_name, expected_result) 

774 self.assertLess(len(application_name), 50) 

775 

776 def test_get_vnf_count_db_vnfr_ns_charm(self): 

777 self.db.get_one.return_value = {"member-vnf-index-ref": "sample-ref"} 

778 charm_level = "ns-level" 

779 vnf_id_and_count = "m7fbd751-3de4-4e68-bd40-ec5ae0a53898-4" 

780 with patch.object(self.n2vc, "db", self.db): 

781 vnf_count, db_vnfr = self.n2vc._get_vnf_count_and_record( 

782 charm_level, vnf_id_and_count 

783 ) 

784 self.assertEqual(vnf_count, "") 

785 self.assertEqual(db_vnfr, {}) 

786 

787 def test_get_vnf_count_db_vnfr_vnf_charm(self): 

788 self.db.get_one.return_value = {"member-vnf-index-ref": "sample-ref"} 

789 charm_level = "vnf-level" 

790 vnf_id_and_count = "m7fbd751-3de4-4e68-bd40-ec5ae0a53898-4" 

791 with patch.object(self.n2vc, "db", self.db): 

792 vnf_count, db_vnfr = self.n2vc._get_vnf_count_and_record( 

793 charm_level, vnf_id_and_count 

794 ) 

795 self.assertEqual(vnf_count, "4") 

796 self.assertEqual(db_vnfr, {"member-vnf-index-ref": "sample-ref"}) 

797 

798 def test_get_vnf_count_db_vnfr_vdu_charm(self): 

799 self.db.get_one.return_value = {"member-vnf-index-ref": "sample-ref"} 

800 charm_level = "vdu-level" 

801 vnf_id_and_count = "m7fbd751-3de4-4e68-bd40-ec5ae0a53898-2" 

802 with patch.object(self.n2vc, "db", self.db): 

803 vnf_count, db_vnfr = self.n2vc._get_vnf_count_and_record( 

804 charm_level, vnf_id_and_count 

805 ) 

806 self.assertEqual(vnf_count, "2") 

807 self.assertEqual(db_vnfr, {"member-vnf-index-ref": "sample-ref"}) 

808 

809 def test_get_vca_records_vdu_charm(self): 

810 charm_level = "vdu-level" 

811 db_vnfr = { 

812 "member-vnf-index-ref": "vnf111-xxx-yyy-zzz", 

813 "vdur": [ 

814 {"_id": "38912ff7-5bdd-4228-911f-c2bee259c44a", "vdu-id-ref": "mgmtVM"}, 

815 {"_id": "45512ff7-5bdd-4228-911f-c2bee259c44a", "vdu-id-ref": "dataVM"}, 

816 ], 

817 } 

818 db_nsr = { 

819 "_admin": { 

820 "deployed": { 

821 "VCA": [ 

822 { 

823 "target_element": "vnf/vnf1/mgmtvm", 

824 "member-vnf-index": "vnf111-xxx-yyy-zzz", 

825 "vdu_id": "38912ff7-5bdd-4228-911f-c2bee259c44a", 

826 "kdu_name": None, 

827 "vdu_count_index": None, 

828 "vnfd_id": "r7fbd751-3de4-4e68-bd40-ec5ae0a53898", 

829 "vdu_name": "mgmtvm", 

830 "ee_descriptor_id": "simple-ee-abc-000-rrrr-nnnn-4444-hhh-3333-yyyy-333-hhh-ttt-444", 

831 "charm_name": "", 

832 "model": "dbfbd751-3de4-4e68-bd40-ec5ae0a53898", 

833 }, 

834 { 

835 "target_element": "vnf/vnf2/datavm", 

836 "member-vnf-index": "vnf222-xxx-yyy-zzz", 

837 "vdu_id": "45512ff7-5bdd-4228-911f-c2bee259c44a",