| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 1 | # |
| 2 | # Licensed under the Apache License, Version 2.0 (the "License"); you may |
| 3 | # not use this file except in compliance with the License. You may obtain |
| 4 | # a copy of the License at |
| 5 | # |
| 6 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 7 | # |
| 8 | # Unless required by applicable law or agreed to in writing, software |
| 9 | # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| 10 | # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| 11 | # License for the specific language governing permissions and limitations |
| 12 | # under the License. |
| 13 | # |
| 14 | # For those usages not covered by the Apache License, Version 2.0 please |
| 15 | # contact: esousa@whitestack.com or alfonso.tiernosepulveda@telefonica.com |
| 16 | ## |
| 17 | |
| aticig | 544a2ae | 2022-04-05 09:00:17 +0300 | [diff] [blame] | 18 | from contextlib import contextmanager |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 19 | import unittest |
| K Sai Kiran | 5758955 | 2021-01-27 21:38:34 +0530 | [diff] [blame] | 20 | from time import time |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 21 | from unittest.mock import Mock, mock_open # patch, MagicMock |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 22 | from osm_common.dbbase import DbException |
| 23 | from osm_nbi.engine import EngineException |
| 24 | from osm_common.dbmemory import DbMemory |
| 25 | from osm_common.fsbase import FsBase |
| 26 | from osm_common.msgbase import MsgBase |
| beierlm | cee2ebf | 2022-03-29 17:42:48 -0400 | [diff] [blame] | 27 | from osm_common import dbbase |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 28 | from http import HTTPStatus |
| 29 | from osm_nbi.instance_topics import NsLcmOpTopic, NsrTopic |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 30 | from osm_nbi.tests.test_db_descriptors import ( |
| 31 | db_vim_accounts_text, |
| 32 | db_nsds_text, |
| 33 | db_vnfds_text, |
| 34 | db_nsrs_text, |
| 35 | db_vnfrs_text, |
| 36 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 37 | from copy import deepcopy |
| 38 | import yaml |
| 39 | |
| 40 | |
| 41 | class TestNsLcmOpTopic(unittest.TestCase): |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 42 | def setUp(self): |
| 43 | self.db = DbMemory() |
| 44 | self.fs = Mock(FsBase()) |
| 45 | self.fs.get_params.return_value = {"./fake/folder"} |
| 46 | self.fs.file_open = mock_open() |
| 47 | self.msg = Mock(MsgBase()) |
| 48 | # create class |
| 49 | self.nslcmop_topic = NsLcmOpTopic(self.db, self.fs, self.msg, None) |
| 50 | self.nslcmop_topic.check_quota = Mock(return_value=None) # skip quota |
| 51 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 52 | self.db.create_list( |
| 53 | "vim_accounts", yaml.load(db_vim_accounts_text, Loader=yaml.Loader) |
| 54 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 55 | self.db.create_list("nsds", yaml.load(db_nsds_text, Loader=yaml.Loader)) |
| 56 | self.db.create_list("vnfds", yaml.load(db_vnfds_text, Loader=yaml.Loader)) |
| 57 | self.db.create_list("vnfrs", yaml.load(db_vnfrs_text, Loader=yaml.Loader)) |
| 58 | self.db.create_list("nsrs", yaml.load(db_nsrs_text, Loader=yaml.Loader)) |
| 59 | self.db.create = Mock(return_value="created_id") |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 60 | self.nsd = self.db.get_list("nsds")[0] |
| 61 | self.nsd_id = self.nsd["_id"] |
| 62 | self.nsr = self.db.get_list("nsrs")[0] |
| 63 | self.nsr_id = self.nsr["_id"] |
| 64 | self.nsr_project = self.nsr["_admin"]["projects_read"][0] |
| 65 | |
| 66 | self.vim = self.db.get_list("vim_accounts")[0] |
| 67 | self.vim_id = self.vim["_id"] |
| 68 | |
| 69 | def test_create_instantiate(self): |
| elumalai | 6380e7c | 2022-04-28 00:15:59 +0530 | [diff] [blame] | 70 | self.db.set_one = Mock(return_value={"updated": 1}) |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 71 | session = { |
| 72 | "force": False, |
| 73 | "admin": False, |
| 74 | "public": False, |
| 75 | "project_id": [self.nsr_project], |
| 76 | "method": "write", |
| 77 | } |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 78 | indata = { |
| 79 | "nsdId": self.nsd_id, |
| 80 | "nsInstanceId": self.nsr_id, |
| 81 | "nsName": "name", |
| 82 | "vimAccountId": self.vim_id, |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 83 | "additionalParamsForVnf": [ |
| 84 | { |
| 85 | "member-vnf-index": "1", |
| 86 | "additionalParams": {"touch_filename": "file"}, |
| 87 | }, |
| 88 | { |
| 89 | "member-vnf-index": "2", |
| 90 | "additionalParams": {"touch_filename": "file"}, |
| 91 | }, |
| 92 | ], |
| 93 | "vnf": [ |
| 94 | { |
| 95 | "member-vnf-index": "1", |
| 96 | "vdu": [ |
| 97 | { |
| 98 | "id": "dataVM", |
| 99 | "interface": [ |
| 100 | { |
| 101 | "name": "dataVM-eth0", |
| 102 | "ip-address": "10.11.12.13", |
| 103 | "floating-ip-required": True, |
| 104 | } |
| 105 | ], |
| 106 | } |
| 107 | ], |
| 108 | "internal-vld": [ |
| 109 | {"name": "internal", "vim-network-id": "vim-net-id"} |
| 110 | ], |
| 111 | } |
| 112 | ], |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 113 | "lcmOperationType": "instantiate", |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 114 | } |
| 115 | rollback = [] |
| 116 | headers = {} |
| 117 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 118 | nslcmop_id, _ = self.nslcmop_topic.new( |
| 119 | rollback, session, indata=deepcopy(indata), kwargs=None, headers=headers |
| 120 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 121 | |
| 122 | # check nslcmop is created at database |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 123 | self.assertEqual( |
| 124 | self.db.create.call_count, |
| 125 | 1, |
| 126 | "database create not called, or called more than once", |
| 127 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 128 | _call = self.db.create.call_args_list[0] |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 129 | self.assertEqual( |
| 130 | _call[0][0], "nslcmops", "must be create a nslcmops entry at database" |
| 131 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 132 | |
| 133 | created_nslcmop = _call[0][1] |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 134 | self.assertEqual( |
| 135 | nslcmop_id, |
| 136 | created_nslcmop["_id"], |
| 137 | "mismatch between return id and database '_id'", |
| 138 | ) |
| 139 | self.assertEqual( |
| 140 | self.nsr_id, |
| 141 | created_nslcmop["nsInstanceId"], |
| 142 | "bad reference id from nslcmop to nsr", |
| 143 | ) |
| 144 | self.assertTrue( |
| 145 | created_nslcmop["_admin"].get("projects_read"), |
| 146 | "Database record must contain '_amdin.projects_read'", |
| 147 | ) |
| 148 | self.assertIn( |
| 149 | "created", |
| 150 | created_nslcmop["_admin"], |
| 151 | "Database record must contain '_admin.created'", |
| 152 | ) |
| 153 | self.assertTrue( |
| 154 | created_nslcmop["lcmOperationType"] == "instantiate", |
| 155 | "Database record must contain 'lcmOperationType=instantiate'", |
| 156 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 157 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 158 | self.assertEqual( |
| 159 | len(rollback), |
| 160 | len(self.db.set_one.call_args_list) + 1, |
| 161 | "rollback mismatch with created/set items at database", |
| 162 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 163 | |
| 164 | # test parameters with error |
| 165 | bad_id = "88d90b0c-faff-4b9f-bccd-aaaaaaaaaaaa" |
| 166 | test_set = ( |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 167 | ( |
| 168 | "nsr not found", |
| 169 | {"nsInstanceId": bad_id}, |
| 170 | DbException, |
| 171 | HTTPStatus.NOT_FOUND, |
| 172 | ("not found", bad_id), |
| 173 | ), |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 174 | # TODO add "nsd" |
| 175 | # ({"vimAccountId": bad_id}, DbException, HTTPStatus.NOT_FOUND, ("not found", bad_id)), # TODO add "vim" |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 176 | ( |
| 177 | "bad member-vnf-index", |
| 178 | {"vnf.0.member-vnf-index": "k"}, |
| 179 | EngineException, |
| 180 | HTTPStatus.BAD_REQUEST, |
| 181 | ("k",), |
| 182 | ), |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 183 | ) |
| 184 | for message, kwargs_, expect_exc, expect_code, expect_text_list in test_set: |
| 185 | with self.assertRaises(expect_exc, msg=message) as e: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 186 | self.nslcmop_topic.new( |
| 187 | rollback, |
| 188 | session, |
| 189 | indata=deepcopy(indata), |
| 190 | kwargs=kwargs_, |
| 191 | headers=headers, |
| 192 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 193 | if expect_code: |
| 194 | self.assertTrue(e.exception.http_code == expect_code) |
| 195 | if expect_text_list: |
| 196 | for expect_text in expect_text_list: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 197 | self.assertIn( |
| 198 | expect_text, |
| 199 | str(e.exception).lower(), |
| 200 | "Expected '{}' at exception text".format(expect_text), |
| 201 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 202 | |
| 203 | def test_check_ns_operation_action(self): |
| 204 | nsrs = self.db.get_list("nsrs")[0] |
| 205 | session = {} |
| 206 | |
| 207 | indata = { |
| 208 | "member_vnf_index": "1", |
| 209 | "vdu_id": None, |
| 210 | "primitive": "touch", |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 211 | "primitive_params": {"filename": "file"}, |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 212 | } |
| 213 | |
| 214 | self.nslcmop_topic._check_ns_operation(session, nsrs, "action", indata) |
| 215 | for k in indata: |
| 216 | indata_copy = indata.copy() |
| 217 | if k == "primitive_params": |
| 218 | continue |
| 219 | indata_copy[k] = "non_existing" |
| 220 | with self.assertRaises(EngineException) as exc_manager: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 221 | self.nslcmop_topic._check_ns_operation( |
| 222 | session, nsrs, "action", indata_copy |
| 223 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 224 | exc = exc_manager.exception |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 225 | self.assertEqual( |
| 226 | exc.http_code, |
| 227 | HTTPStatus.BAD_REQUEST, |
| 228 | "Engine exception bad http_code with {}".format(indata_copy), |
| 229 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 230 | |
| elumalai | 6380e7c | 2022-04-28 00:15:59 +0530 | [diff] [blame] | 231 | def test_update_remove_vnf(self): |
| 232 | vnfr_id = self.db.get_list("vnfrs")[0]["_id"] |
| 233 | session = {} |
| 234 | self.db.set_one( |
| 235 | "nsrs", |
| 236 | {"_id": self.nsr_id}, |
| 237 | {"_admin.nsState": "INSTANTIATED"}, |
| 238 | ) |
| 239 | indata = { |
| 240 | "lcmOperationType": "update", |
| 241 | "updateType": "REMOVE_VNF", |
| 242 | "nsInstanceId": self.nsr_id, |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 243 | "removeVnfInstanceId": vnfr_id, |
| elumalai | 6380e7c | 2022-04-28 00:15:59 +0530 | [diff] [blame] | 244 | } |
| 245 | |
| 246 | session = { |
| 247 | "force": False, |
| 248 | "admin": False, |
| 249 | "public": False, |
| 250 | "project_id": [self.nsr_project], |
| 251 | "method": "write", |
| 252 | } |
| 253 | rollback = [] |
| 254 | headers = {} |
| 255 | |
| 256 | nslcmop_id, _ = self.nslcmop_topic.new( |
| 257 | rollback, session, indata, kwargs=None, headers=headers |
| 258 | ) |
| 259 | |
| 260 | self.assertEqual( |
| 261 | self.db.create.call_count, |
| 262 | 1, |
| 263 | "database create not called, or called more than once", |
| 264 | ) |
| 265 | _call = self.db.create.call_args_list[0] |
| 266 | self.assertEqual( |
| 267 | _call[0][0], "nslcmops", "nslcmops entry must be created at database" |
| 268 | ) |
| 269 | created_nslcmop = _call[0][1] |
| 270 | self.assertEqual( |
| 271 | self.nsr_id, |
| 272 | created_nslcmop["nsInstanceId"], |
| 273 | "mismatch between nsId '_id' in created nslcmop and database nsr", |
| 274 | ) |
| 275 | self.assertTrue( |
| 276 | created_nslcmop["lcmOperationType"] == "update", |
| 277 | "Database record must contain 'lcmOperationType=update'", |
| 278 | ) |
| 279 | self.assertTrue( |
| 280 | created_nslcmop["operationParams"]["updateType"] == "REMOVE_VNF", |
| 281 | "Database record must contain 'updateType=REMOVE_VNF'", |
| 282 | ) |
| 283 | |
| elumalai | 8f92b4b | 2022-06-01 12:57:24 +0530 | [diff] [blame] | 284 | def test_migrate(self): |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 285 | _ = self.db.get_list("vnfrs")[0]["_id"] |
| elumalai | 8f92b4b | 2022-06-01 12:57:24 +0530 | [diff] [blame] | 286 | session = {} |
| 287 | self.db.set_one( |
| 288 | "nsrs", |
| 289 | {"_id": self.nsr_id}, |
| 290 | {"_admin.nsState": "INSTANTIATED"}, |
| 291 | ) |
| 292 | session = { |
| 293 | "force": False, |
| 294 | "admin": False, |
| 295 | "public": False, |
| 296 | "project_id": [self.nsr_project], |
| 297 | "method": "write", |
| 298 | } |
| 299 | rollback = [] |
| 300 | headers = {} |
| 301 | |
| 302 | with self.subTest(i=1, t="Migration for Specific VM"): |
| 303 | indata = { |
| 304 | "lcmOperationType": "migrate", |
| 305 | "nsInstanceId": self.nsr_id, |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 306 | "migrateToHost": "sample02", |
| 307 | "vdu": {"vduCountIndex": 0, "vduId": "mgmtVM"}, |
| 308 | "vnfInstanceId": "9e8006df-cdfa-4f63-bf6a-fce860d71c1f", |
| elumalai | 8f92b4b | 2022-06-01 12:57:24 +0530 | [diff] [blame] | 309 | } |
| 310 | nslcmop_id, _ = self.nslcmop_topic.new( |
| 311 | rollback, session, indata, kwargs=None, headers=headers |
| 312 | ) |
| 313 | |
| 314 | self.assertEqual( |
| 315 | self.db.create.call_count, |
| 316 | 1, |
| 317 | "database create not called, or called more than once", |
| 318 | ) |
| 319 | _call = self.db.create.call_args_list[0] |
| 320 | self.assertEqual( |
| 321 | _call[0][0], "nslcmops", "nslcmops entry must be created at database" |
| 322 | ) |
| 323 | created_nslcmop = _call[0][1] |
| 324 | self.assertEqual( |
| 325 | self.nsr_id, |
| 326 | created_nslcmop["nsInstanceId"], |
| 327 | "mismatch between nsId '_id' in created nslcmop and database nsr", |
| 328 | ) |
| 329 | self.assertTrue( |
| 330 | created_nslcmop["lcmOperationType"] == "migrate", |
| 331 | "Database record must contain 'lcmOperationType=migrate'", |
| 332 | ) |
| 333 | with self.subTest(i=2, t="Migration of all VDUs in a VNF"): |
| 334 | indata = { |
| 335 | "lcmOperationType": "migrate", |
| 336 | "nsInstanceId": self.nsr_id, |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 337 | "vnfInstanceId": "9e8006df-cdfa-4f63-bf6a-fce860d71c1f", |
| elumalai | 8f92b4b | 2022-06-01 12:57:24 +0530 | [diff] [blame] | 338 | } |
| 339 | nslcmop_id, _ = self.nslcmop_topic.new( |
| 340 | rollback, session, indata, kwargs=None, headers=headers |
| 341 | ) |
| 342 | |
| 343 | self.assertEqual( |
| 344 | self.db.create.call_count, |
| 345 | 2, |
| 346 | "database create not called, or called more than once", |
| 347 | ) |
| 348 | _call = self.db.create.call_args_list[0] |
| 349 | self.assertEqual( |
| 350 | _call[0][0], "nslcmops", "nslcmops entry must be created at database" |
| 351 | ) |
| 352 | created_nslcmop = _call[0][1] |
| 353 | self.assertEqual( |
| 354 | self.nsr_id, |
| 355 | created_nslcmop["nsInstanceId"], |
| 356 | "mismatch between nsId '_id' in created nslcmop and database nsr", |
| 357 | ) |
| 358 | self.assertTrue( |
| 359 | created_nslcmop["lcmOperationType"] == "migrate", |
| 360 | "Database record must contain 'lcmOperationType=migrate'", |
| 361 | ) |
| 362 | with self.subTest(i=3, t="Migration failure - vduId not provided in vdu "): |
| 363 | indata = { |
| 364 | "lcmOperationType": "migrate", |
| 365 | "nsInstanceId": self.nsr_id, |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 366 | "migrateToHost": "sample02", |
| 367 | "vdu": {"vduCountIndex": 0}, |
| 368 | "vnfInstanceId": "9e8006df-cdfa-4f63-bf6a-fce860d71c1f", |
| elumalai | 8f92b4b | 2022-06-01 12:57:24 +0530 | [diff] [blame] | 369 | } |
| 370 | |
| 371 | with self.assertRaises(Exception) as e: |
| 372 | nslcmop_id, _ = self.nslcmop_topic.new( |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 373 | rollback, session, indata, kwargs=None, headers=headers |
| 374 | ) |
| 375 | self.assertTrue( |
| 376 | "Format error at 'vdu' ''vduId' is a required property'" |
| 377 | in str(e.exception) |
| elumalai | 8f92b4b | 2022-06-01 12:57:24 +0530 | [diff] [blame] | 378 | ) |
| elumalai | 8f92b4b | 2022-06-01 12:57:24 +0530 | [diff] [blame] | 379 | |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 380 | |
| beierlm | cee2ebf | 2022-03-29 17:42:48 -0400 | [diff] [blame] | 381 | class TestNsLcmOpTopicWithMock(unittest.TestCase): |
| 382 | def setUp(self): |
| 383 | self.db = Mock(dbbase.DbBase()) |
| 384 | self.fs = Mock(FsBase()) |
| 385 | self.fs.get_params.return_value = {"./fake/folder"} |
| 386 | self.fs.file_open = mock_open() |
| 387 | self.msg = Mock(MsgBase()) |
| 388 | # create class |
| 389 | self.nslcmop_topic = NsLcmOpTopic(self.db, self.fs, self.msg, None) |
| 390 | |
| 391 | def test_get_vnfd_from_vnf_member_revision(self): |
| 392 | test_vnfr = yaml.load(db_vnfrs_text, Loader=yaml.Loader)[0] |
| 393 | test_vnfd = yaml.load(db_vnfds_text, Loader=yaml.Loader) |
| 394 | self.db.get_one.side_effect = [test_vnfr, test_vnfd] |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 395 | _ = self.nslcmop_topic._get_vnfd_from_vnf_member_index("1", test_vnfr["_id"]) |
| 396 | self.assertEqual( |
| 397 | self.db.get_one.call_args_list[0][0][0], |
| 398 | "vnfrs", |
| 399 | "Incorrect first DB lookup", |
| 400 | ) |
| 401 | self.assertEqual( |
| 402 | self.db.get_one.call_args_list[1][0][0], |
| 403 | "vnfds", |
| 404 | "Incorrect second DB lookup", |
| 405 | ) |
| beierlm | cee2ebf | 2022-03-29 17:42:48 -0400 | [diff] [blame] | 406 | |
| 407 | def test_get_vnfd_from_vnf_member_no_revision(self): |
| 408 | test_vnfr = yaml.load(db_vnfrs_text, Loader=yaml.Loader)[0] |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 409 | test_vnfr["revision"] = 3 |
| beierlm | cee2ebf | 2022-03-29 17:42:48 -0400 | [diff] [blame] | 410 | test_vnfd = yaml.load(db_vnfds_text, Loader=yaml.Loader) |
| 411 | self.db.get_one.side_effect = [test_vnfr, test_vnfd] |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 412 | _ = self.nslcmop_topic._get_vnfd_from_vnf_member_index("1", test_vnfr["_id"]) |
| 413 | self.assertEqual( |
| 414 | self.db.get_one.call_args_list[0][0][0], |
| 415 | "vnfrs", |
| 416 | "Incorrect first DB lookup", |
| 417 | ) |
| 418 | self.assertEqual( |
| 419 | self.db.get_one.call_args_list[1][0][0], |
| 420 | "vnfds_revisions", |
| 421 | "Incorrect second DB lookup", |
| 422 | ) |
| beierlm | cee2ebf | 2022-03-29 17:42:48 -0400 | [diff] [blame] | 423 | |
| aticig | 544a2ae | 2022-04-05 09:00:17 +0300 | [diff] [blame] | 424 | @contextmanager |
| 425 | def assertNotRaises(self, exception_type): |
| 426 | try: |
| 427 | yield None |
| 428 | except exception_type: |
| 429 | raise self.failureException("{} raised".format(exception_type.__name__)) |
| 430 | |
| 431 | def test_check_ns_update_operation(self): |
| 432 | self.db = DbMemory() |
| 433 | self.nslcmop_topic = NsLcmOpTopic(self.db, self.fs, self.msg, None) |
| 434 | session = {} |
| 435 | |
| 436 | with self.subTest(i=1, t="VNF instance does not belong to NS"): |
| 437 | test_vnfr = yaml.load(db_vnfrs_text, Loader=yaml.Loader) |
| 438 | test_vnfr[0]["revision"] = 2 |
| 439 | test_nsr = yaml.load(db_nsrs_text, Loader=yaml.Loader) |
| 440 | test_nsr[0]["constituent-vnfr-ref"][ |
| 441 | 0 |
| 442 | ] = "99d90b0c-faff-4b9f-bccd-017f33985984" |
| 443 | self.db.create_list("vnfrs", test_vnfr) |
| 444 | self.db.create_list("nsrs", test_nsr) |
| 445 | nsrs = self.db.get_list("nsrs")[0] |
| 446 | indata = { |
| 447 | "updateType": "CHANGE_VNFPKG", |
| 448 | "changeVnfPackageData": { |
| 449 | "vnfInstanceId": "88d90b0c-faff-4b9f-bccd-017f33985984", |
| 450 | "vnfdId": "7637bcf8-cf14-42dc-ad70-c66fcf1e6e77", |
| 451 | }, |
| 452 | "nsInstanceId": "f48163a6-c807-47bc-9682-f72caef5af85", |
| 453 | } |
| 454 | with self.assertRaises(EngineException) as expected_exception: |
| 455 | self.nslcmop_topic._check_ns_operation(session, nsrs, "update", indata) |
| 456 | self.assertEqual( |
| 457 | str(expected_exception.exception), |
| 458 | "Error in validating ns-update request: vnf 88d90b0c-faff-4b9f-bccd-017f33985984" |
| 459 | " does not belong to NS f48163a6-c807-47bc-9682-f72caef5af85", |
| 460 | ) |
| 461 | |
| 462 | with self.subTest(i=2, t="Ns update request validated with no exception"): |
| 463 | test_vnfr = yaml.load(db_vnfrs_text, Loader=yaml.Loader) |
| 464 | test_vnfr[0]["revision"] = 2 |
| 465 | test_nsr = yaml.load(db_nsrs_text, Loader=yaml.Loader) |
| 466 | self.db.create_list("vnfrs", test_vnfr) |
| 467 | self.db.create_list("nsrs", test_nsr) |
| 468 | nsrs = self.db.get_list("nsrs")[1] |
| 469 | indata = { |
| 470 | "updateType": "CHANGE_VNFPKG", |
| 471 | "changeVnfPackageData": { |
| 472 | "vnfInstanceId": "88d90b0c-faff-4b9f-bccd-017f33985984", |
| 473 | "vnfdId": "7637bcf8-cf14-42dc-ad70-c66fcf1e6e77", |
| 474 | }, |
| 475 | "nsInstanceId": "f48163a6-c807-47bc-9682-f72caef5af85", |
| 476 | } |
| 477 | with self.assertNotRaises(EngineException): |
| 478 | self.nslcmop_topic._check_ns_operation(session, nsrs, "update", indata) |
| 479 | |
| 480 | with self.subTest( |
| 481 | i=3, t="Ns update request rejected because of too small timeout" |
| 482 | ): |
| 483 | indata = { |
| 484 | "updateType": "CHANGE_VNFPKG", |
| 485 | "changeVnfPackageData": { |
| 486 | "vnfInstanceId": "88d90b0c-faff-4b9f-bccd-017f33985984", |
| 487 | "vnfdId": "7637bcf8-cf14-42dc-ad70-c66fcf1e6e77", |
| 488 | }, |
| 489 | "nsInstanceId": "f48163a6-c807-47bc-9682-f72caef5af85", |
| 490 | "timeout_ns_update": 50, |
| 491 | } |
| 492 | with self.assertRaises(EngineException) as expected_exception: |
| 493 | self.nslcmop_topic._check_ns_operation(session, nsrs, "update", indata) |
| 494 | self.assertEqual( |
| 495 | str(expected_exception.exception), |
| 496 | "Error in validating ns-update request: 50 second is not enough " |
| 497 | "to upgrade the VNF instance: 88d90b0c-faff-4b9f-bccd-017f33985984", |
| 498 | ) |
| 499 | |
| 500 | with self.subTest(i=4, t="wrong vnfdid is given as an update parameter"): |
| 501 | test_vnfr = yaml.load(db_vnfrs_text, Loader=yaml.Loader) |
| 502 | test_vnfr[0]["revision"] = 2 |
| 503 | test_nsr = yaml.load(db_nsrs_text, Loader=yaml.Loader) |
| 504 | self.db.create_list("vnfrs", test_vnfr) |
| 505 | self.db.create_list("nsrs", test_nsr) |
| 506 | nsrs = self.db.get_list("nsrs")[2] |
| 507 | indata = { |
| 508 | "updateType": "CHANGE_VNFPKG", |
| 509 | "changeVnfPackageData": { |
| 510 | "vnfInstanceId": "88d90b0c-faff-4b9f-bccd-017f33985984", |
| 511 | "vnfdId": "9637bcf8-cf14-42dc-ad70-c66fcf1e6e77", |
| 512 | }, |
| 513 | "nsInstanceId": "f48163a6-c807-47bc-9682-f72caef5af85", |
| 514 | } |
| 515 | with self.assertRaises(EngineException) as expected_exception: |
| 516 | self.nslcmop_topic._check_ns_operation(session, nsrs, "update", indata) |
| 517 | self.assertEqual( |
| 518 | str(expected_exception.exception), |
| 519 | "Error in validating ns-update request: vnfd-id 9637bcf8-cf14-42dc-ad70-c66fcf1e6e77 does not " |
| 520 | "match with the vnfd-id: 7637bcf8-cf14-42dc-ad70-c66fcf1e6e77 of " |
| 521 | "VNF instance: 88d90b0c-faff-4b9f-bccd-017f33985984", |
| 522 | ) |
| 523 | |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 524 | with self.subTest( |
| 525 | i=5, t="Ns update REMOVE_VNF request validated with no exception" |
| 526 | ): |
| elumalai | 6380e7c | 2022-04-28 00:15:59 +0530 | [diff] [blame] | 527 | test_vnfr = yaml.load(db_vnfrs_text, Loader=yaml.Loader) |
| 528 | test_vnfr[0]["revision"] = 2 |
| 529 | test_nsr = yaml.load(db_nsrs_text, Loader=yaml.Loader) |
| 530 | self.db.create_list("vnfrs", test_vnfr) |
| 531 | self.db.create_list("nsrs", test_nsr) |
| 532 | nsrs = self.db.get_list("nsrs")[1] |
| 533 | indata = { |
| 534 | "updateType": "REMOVE_VNF", |
| 535 | "removeVnfInstanceId": "88d90b0c-faff-4b9f-bccd-017f33985984", |
| 536 | "nsInstanceId": "f48163a6-c807-47bc-9682-f72caef5af85", |
| 537 | } |
| 538 | with self.assertNotRaises(EngineException): |
| 539 | self.nslcmop_topic._check_ns_operation(session, nsrs, "update", indata) |
| beierlm | cee2ebf | 2022-03-29 17:42:48 -0400 | [diff] [blame] | 540 | |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 541 | |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 542 | class TestNsrTopic(unittest.TestCase): |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 543 | def setUp(self): |
| 544 | self.db = DbMemory() |
| 545 | self.fs = Mock(FsBase()) |
| 546 | self.fs.get_params.return_value = {"./fake/folder"} |
| 547 | self.fs.file_open = mock_open() |
| 548 | self.msg = Mock(MsgBase()) |
| 549 | # create class |
| 550 | self.nsr_topic = NsrTopic(self.db, self.fs, self.msg, None) |
| 551 | self.nsr_topic.check_quota = Mock(return_value=None) # skip quota |
| 552 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 553 | self.db.create_list( |
| 554 | "vim_accounts", yaml.load(db_vim_accounts_text, Loader=yaml.Loader) |
| 555 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 556 | self.db.create_list("nsds", yaml.load(db_nsds_text, Loader=yaml.Loader)) |
| 557 | self.db.create_list("vnfds", yaml.load(db_vnfds_text, Loader=yaml.Loader)) |
| 558 | self.db.create = Mock(return_value="created_id") |
| 559 | self.nsd = self.db.get_list("nsds")[0] |
| 560 | self.nsd_id = self.nsd["_id"] |
| 561 | self.nsd_project = self.nsd["_admin"]["projects_read"][0] |
| 562 | |
| 563 | self.vim = self.db.get_list("vim_accounts")[0] |
| 564 | self.vim_id = self.vim["_id"] |
| 565 | |
| 566 | def test_create(self): |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 567 | session = { |
| 568 | "force": False, |
| 569 | "admin": False, |
| 570 | "public": False, |
| 571 | "project_id": [self.nsd_project], |
| 572 | "method": "write", |
| 573 | } |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 574 | indata = { |
| 575 | "nsdId": self.nsd_id, |
| 576 | "nsName": "name", |
| 577 | "vimAccountId": self.vim_id, |
| garciaale | 7cbd03c | 2020-11-27 10:38:35 -0300 | [diff] [blame] | 578 | "additionalParamsForVnf": [ |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 579 | { |
| 580 | "member-vnf-index": "hackfest_vnf1", |
| 581 | "additionalParams": {"touch_filename": "file"}, |
| 582 | }, |
| 583 | { |
| 584 | "member-vnf-index": "hackfest_vnf2", |
| 585 | "additionalParams": {"touch_filename": "file"}, |
| 586 | }, |
| 587 | ], |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 588 | } |
| 589 | rollback = [] |
| 590 | headers = {} |
| 591 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 592 | self.nsr_topic.new( |
| 593 | rollback, session, indata=indata, kwargs=None, headers=headers |
| 594 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 595 | |
| 596 | # check vnfrs and nsrs created in whatever order |
| 597 | created_vnfrs = [] |
| 598 | created_nsrs = [] |
| 599 | nsr_id = None |
| 600 | for _call in self.db.create.call_args_list: |
| 601 | assert len(_call[0]) >= 2, "called db.create with few parameters" |
| 602 | created_item = _call[0][1] |
| 603 | if _call[0][0] == "vnfrs": |
| 604 | created_vnfrs.append(created_item) |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 605 | self.assertIn( |
| 606 | "member-vnf-index-ref", |
| 607 | created_item, |
| 608 | "Created item must contain member-vnf-index-ref section", |
| 609 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 610 | if nsr_id: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 611 | self.assertEqual( |
| 612 | nsr_id, |
| 613 | created_item["nsr-id-ref"], |
| 614 | "bad reference id from vnfr to nsr", |
| 615 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 616 | else: |
| 617 | nsr_id = created_item["nsr-id-ref"] |
| 618 | |
| 619 | elif _call[0][0] == "nsrs": |
| 620 | created_nsrs.append(created_item) |
| 621 | if nsr_id: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 622 | self.assertEqual( |
| 623 | nsr_id, created_item["_id"], "bad reference id from vnfr to nsr" |
| 624 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 625 | else: |
| 626 | nsr_id = created_item["_id"] |
| 627 | else: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 628 | assert True, "created an unknown record {} at database".format( |
| 629 | _call[0][0] |
| 630 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 631 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 632 | self.assertTrue( |
| 633 | created_item["_admin"].get("projects_read"), |
| 634 | "Database record must contain '_amdin.projects_read'", |
| 635 | ) |
| 636 | self.assertIn( |
| 637 | "created", |
| 638 | created_item["_admin"], |
| 639 | "Database record must contain '_admin.created'", |
| 640 | ) |
| 641 | self.assertTrue( |
| 642 | created_item["_admin"]["nsState"] == "NOT_INSTANTIATED", |
| 643 | "Database record must contain '_admin.nstate=NOT INSTANTIATE'", |
| 644 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 645 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 646 | self.assertEqual( |
| 647 | len(created_vnfrs), 2, "created a mismatch number of vnfr at database" |
| 648 | ) |
| aticig | c9c0339 | 2022-06-16 01:39:44 +0300 | [diff] [blame] | 649 | |
| 650 | self.assertEqual( |
| 651 | created_vnfrs[0]["vdur"][0]["interfaces"][0]["position"], |
| 652 | 1, |
| 653 | "vdur first interface position does not match", |
| 654 | ) |
| 655 | |
| 656 | self.assertEqual( |
| 657 | created_vnfrs[0]["vdur"][0]["interfaces"][1]["position"], |
| 658 | 2, |
| 659 | "vdur second interface position does not match", |
| 660 | ) |
| 661 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 662 | self.assertEqual( |
| 663 | len(created_nsrs), 1, "Only one nsrs must be created at database" |
| 664 | ) |
| 665 | self.assertEqual( |
| 666 | len(rollback), |
| 667 | len(created_vnfrs) + 1, |
| 668 | "rollback mismatch with created items at database", |
| 669 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 670 | |
| 671 | # test parameters with error |
| 672 | bad_id = "88d90b0c-faff-4b9f-bccd-aaaaaaaaaaaa" |
| 673 | test_set = ( |
| 674 | # TODO add "nsd" |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 675 | ( |
| 676 | "nsd not found", |
| 677 | {"nsdId": bad_id}, |
| 678 | DbException, |
| 679 | HTTPStatus.NOT_FOUND, |
| 680 | ("not found", bad_id), |
| 681 | ), |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 682 | # ({"vimAccountId": bad_id}, DbException, HTTPStatus.NOT_FOUND, ("not found", bad_id)), # TODO add "vim" |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 683 | ( |
| 684 | "additional params not supply", |
| 685 | {"additionalParamsForVnf.0.member-vnf-index": "k"}, |
| 686 | EngineException, |
| 687 | HTTPStatus.BAD_REQUEST, |
| 688 | None, |
| 689 | ), |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 690 | ) |
| 691 | for message, kwargs_, expect_exc, expect_code, expect_text_list in test_set: |
| 692 | with self.assertRaises(expect_exc, msg=message) as e: |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 693 | self.nsr_topic.new( |
| 694 | rollback, |
| 695 | session, |
| 696 | indata=deepcopy(indata), |
| 697 | kwargs=kwargs_, |
| 698 | headers=headers, |
| 699 | ) |
| tierno | 1bfe4e2 | 2019-09-02 16:03:25 +0000 | [diff] [blame] | 700 | if expect_code: |
| 701 | self.assertTrue(e.exception.http_code == expect_code) |
| 702 | if expect_text_list: |
| 703 | for expect_text in expect_text_list: |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 704 | self.assertIn( |
| 705 | expect_text, |
| 706 | str(e.exception).lower(), |
| 707 | "Expected '{}' at exception text".format(expect_text), |
| 708 | ) |
| K Sai Kiran | 5758955 | 2021-01-27 21:38:34 +0530 | [diff] [blame] | 709 | |
| 710 | def test_show_instance(self): |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 711 | session = { |
| 712 | "force": False, |
| 713 | "admin": False, |
| 714 | "public": False, |
| 715 | "project_id": [self.nsd_project], |
| 716 | "method": "write", |
| 717 | } |
| K Sai Kiran | 5758955 | 2021-01-27 21:38:34 +0530 | [diff] [blame] | 718 | filter_q = {} |
| 719 | for refresh_status in ("true", "false"): |
| 720 | self.db.create_list("nsrs", yaml.load(db_nsrs_text, Loader=yaml.Loader)) |
| 721 | actual_nsr = self.db.get_list("nsrs")[0] |
| 722 | nsr_id = actual_nsr["_id"] |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 723 | filter_q["vcaStatus-refresh"] = refresh_status |
| K Sai Kiran | 5758955 | 2021-01-27 21:38:34 +0530 | [diff] [blame] | 724 | expected_nsr = self.nsr_topic.show(session, nsr_id, filter_q=filter_q) |
| 725 | self.nsr_topic.delete(session, nsr_id) |
| 726 | actual_nsr.pop("_admin") |
| 727 | expected_nsr.pop("_admin") |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 728 | self.assertEqual( |
| 729 | expected_nsr, actual_nsr, "Database nsr and show() nsr do not match." |
| 730 | ) |
| K Sai Kiran | 5758955 | 2021-01-27 21:38:34 +0530 | [diff] [blame] | 731 | |
| 732 | def test_vca_status_refresh(self): |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 733 | session = { |
| 734 | "force": False, |
| 735 | "admin": False, |
| 736 | "public": False, |
| 737 | "project_id": [self.nsd_project], |
| 738 | "method": "write", |
| 739 | } |
| 740 | filter_q = {"vcaStatus-refresh": "true"} |
| K Sai Kiran | 5758955 | 2021-01-27 21:38:34 +0530 | [diff] [blame] | 741 | time_delta = 120 |
| 742 | self.db.create_list("nsrs", yaml.load(db_nsrs_text, Loader=yaml.Loader)) |
| 743 | nsr = self.db.get_list("nsrs")[0] |
| 744 | |
| 745 | # When vcaStatus-refresh is true |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 746 | filter_q["vcaStatus-refresh"] = "true" |
| K Sai Kiran | 5758955 | 2021-01-27 21:38:34 +0530 | [diff] [blame] | 747 | self.nsr_topic.vca_status_refresh(session, nsr, filter_q) |
| 748 | msg_args = self.msg.write.call_args[0] |
| 749 | self.assertEqual(msg_args[1], "vca_status_refresh", "Wrong message action") |
| 750 | self.assertGreater(nsr["_admin"]["modified"], time() - time_delta) |
| 751 | |
| 752 | # When vcaStatus-refresh is false but modified time is within threshold |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 753 | filter_q["vcaStatus-refresh"] = "false" |
| K Sai Kiran | 5758955 | 2021-01-27 21:38:34 +0530 | [diff] [blame] | 754 | time_now = time() |
| 755 | nsr["_admin"]["modified"] = time_now |
| 756 | self.nsr_topic.vca_status_refresh(session, nsr, filter_q) |
| 757 | msg_args = self.msg.write.call_args[1] |
| 758 | self.assertEqual(msg_args, {}, "Message should not be sent.") |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 759 | self.assertEqual( |
| 760 | nsr["_admin"]["modified"], time_now, "Modified time should not be changed." |
| 761 | ) |
| K Sai Kiran | 5758955 | 2021-01-27 21:38:34 +0530 | [diff] [blame] | 762 | |
| 763 | # When vcaStatus-refresh is false but modified time is less than threshold |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 764 | filter_q["vcaStatus-refresh"] = "false" |
| 765 | nsr["_admin"]["modified"] = time() - (2 * time_delta) |
| K Sai Kiran | 5758955 | 2021-01-27 21:38:34 +0530 | [diff] [blame] | 766 | self.nsr_topic.vca_status_refresh(session, nsr, filter_q) |
| 767 | msg_args = self.msg.write.call_args[0] |
| 768 | self.assertEqual(msg_args[1], "vca_status_refresh", "Wrong message action") |
| 769 | self.nsr_topic.delete(session, nsr["_id"]) |
| garciadeblas | f2af4a1 | 2023-01-24 16:56:54 +0100 | [diff] [blame] | 770 | self.assertGreater( |
| 771 | nsr["_admin"]["modified"], |
| 772 | time() - time_delta, |
| 773 | "Modified time is not changed.", |
| 774 | ) |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 775 | |
| 776 | def test_delete_ns(self): |
| 777 | self.db.create_list("nsrs", yaml.load(db_nsrs_text, Loader=yaml.Loader)) |
| 778 | self.nsr = self.db.get_list("nsrs")[0] |
| 779 | self.nsr_id = self.nsr["_id"] |
| 780 | self.db_set_one = self.db.set_one |
| 781 | p_id = self.nsd_project |
| 782 | p_other = "other_p" |
| 783 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 784 | session = { |
| 785 | "force": False, |
| 786 | "admin": False, |
| 787 | "public": None, |
| 788 | "project_id": [p_id], |
| 789 | "method": "delete", |
| 790 | } |
| 791 | session2 = { |
| 792 | "force": False, |
| 793 | "admin": False, |
| 794 | "public": None, |
| 795 | "project_id": [p_other], |
| 796 | "method": "delete", |
| 797 | } |
| 798 | session_force = { |
| 799 | "force": True, |
| 800 | "admin": True, |
| 801 | "public": None, |
| 802 | "project_id": [], |
| 803 | "method": "delete", |
| 804 | } |
| 805 | with self.subTest(i=1, t="Normal Deletion"): |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 806 | self.db.del_one = Mock() |
| 807 | self.db.set_one = Mock() |
| 808 | self.nsr_topic.delete(session, self.nsr_id) |
| 809 | |
| tierno | f0441ea | 2020-05-26 15:39:18 +0000 | [diff] [blame] | 810 | db_args_ro_nsrs = self.db.del_one.call_args_list[1][0] |
| 811 | db_args = self.db.del_one.call_args_list[0][0] |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 812 | msg_args = self.msg.write.call_args[0] |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 813 | self.assertEqual( |
| 814 | msg_args[0], self.nsr_topic.topic_msg, "Wrong message topic" |
| 815 | ) |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 816 | self.assertEqual(msg_args[1], "deleted", "Wrong message action") |
| 817 | self.assertEqual(msg_args[2], {"_id": self.nsr_id}, "Wrong message content") |
| tierno | f0441ea | 2020-05-26 15:39:18 +0000 | [diff] [blame] | 818 | self.assertEqual(db_args_ro_nsrs[0], "ro_nsrs", "Wrong DB topic") |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 819 | self.assertEqual(db_args[0], self.nsr_topic.topic, "Wrong DB topic") |
| 820 | self.assertEqual(db_args[1]["_id"], self.nsr_id, "Wrong DB ID") |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 821 | self.assertEqual( |
| 822 | db_args[1]["_admin.projects_read.cont"], [p_id], "Wrong DB filter" |
| 823 | ) |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 824 | self.db.set_one.assert_not_called() |
| 825 | fs_del_calls = self.fs.file_delete.call_args_list |
| 826 | self.assertEqual(fs_del_calls[0][0][0], self.nsr_id, "Wrong FS file id") |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 827 | with self.subTest(i=2, t="No delete because referenced by other project"): |
| 828 | self.db_set_one( |
| 829 | "nsrs", |
| 830 | {"_id": self.nsr_id}, |
| 831 | update_dict=None, |
| 832 | push={ |
| 833 | "_admin.projects_read": p_other, |
| 834 | "_admin.projects_write": p_other, |
| 835 | }, |
| 836 | ) |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 837 | self.db.del_one.reset_mock() |
| 838 | self.db.set_one.reset_mock() |
| 839 | self.msg.write.reset_mock() |
| 840 | self.fs.file_delete.reset_mock() |
| 841 | |
| 842 | self.nsr_topic.delete(session2, self.nsr_id) |
| 843 | self.db.del_one.assert_not_called() |
| 844 | self.msg.write.assert_not_called() |
| 845 | db_s1_args = self.db.set_one.call_args |
| 846 | self.assertEqual(db_s1_args[0][0], self.nsr_topic.topic, "Wrong DB topic") |
| 847 | self.assertEqual(db_s1_args[0][1]["_id"], self.nsr_id, "Wrong DB ID") |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 848 | self.assertIsNone( |
| 849 | db_s1_args[1]["update_dict"], "Wrong DB update dictionary" |
| 850 | ) |
| 851 | self.assertEqual( |
| 852 | db_s1_args[1]["pull_list"], |
| 853 | {"_admin.projects_read": [p_other], "_admin.projects_write": [p_other]}, |
| 854 | "Wrong DB pull_list dictionary", |
| 855 | ) |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 856 | self.fs.file_delete.assert_not_called() |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 857 | with self.subTest(i=4, t="Delete with force and admin"): |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 858 | self.db.del_one.reset_mock() |
| 859 | self.db.set_one.reset_mock() |
| 860 | self.msg.write.reset_mock() |
| 861 | self.fs.file_delete.reset_mock() |
| 862 | self.nsr_topic.delete(session_force, self.nsr_id) |
| 863 | |
| tierno | f0441ea | 2020-05-26 15:39:18 +0000 | [diff] [blame] | 864 | db_args_ro_nsrs = self.db.del_one.call_args_list[1][0] |
| 865 | db_args = self.db.del_one.call_args_list[0][0] |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 866 | msg_args = self.msg.write.call_args[0] |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 867 | self.assertEqual( |
| 868 | msg_args[0], self.nsr_topic.topic_msg, "Wrong message topic" |
| 869 | ) |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 870 | self.assertEqual(msg_args[1], "deleted", "Wrong message action") |
| 871 | self.assertEqual(msg_args[2], {"_id": self.nsr_id}, "Wrong message content") |
| tierno | f0441ea | 2020-05-26 15:39:18 +0000 | [diff] [blame] | 872 | self.assertEqual(db_args_ro_nsrs[0], "ro_nsrs", "Wrong DB topic") |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 873 | self.assertEqual(db_args[0], self.nsr_topic.topic, "Wrong DB topic") |
| 874 | self.assertEqual(db_args[1]["_id"], self.nsr_id, "Wrong DB ID") |
| 875 | self.db.set_one.assert_not_called() |
| 876 | fs_del_calls = self.fs.file_delete.call_args_list |
| 877 | self.assertEqual(fs_del_calls[0][0][0], self.nsr_id, "Wrong FS file id") |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 878 | with self.subTest(i=3, t="Conflict on Delete - NS in INSTANTIATED state"): |
| 879 | self.db_set_one( |
| 880 | "nsrs", |
| 881 | {"_id": self.nsr_id}, |
| 882 | {"_admin.nsState": "INSTANTIATED"}, |
| 883 | pull={ |
| 884 | "_admin.projects_read": p_other, |
| 885 | "_admin.projects_write": p_other, |
| 886 | }, |
| 887 | ) |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 888 | self.db.del_one.reset_mock() |
| 889 | self.db.set_one.reset_mock() |
| 890 | self.msg.write.reset_mock() |
| 891 | self.fs.file_delete.reset_mock() |
| 892 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 893 | with self.assertRaises( |
| 894 | EngineException, msg="Accepted NSR with nsState INSTANTIATED" |
| 895 | ) as e: |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 896 | self.nsr_topic.delete(session, self.nsr_id) |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 897 | self.assertEqual( |
| 898 | e.exception.http_code, HTTPStatus.CONFLICT, "Wrong HTTP status code" |
| 899 | ) |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 900 | self.assertIn("INSTANTIATED", str(e.exception), "Wrong exception text") |
| 901 | # TODOD with self.subTest(i=3, t='Conflict on Delete - NS in use by NSI'): |
| 902 | |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 903 | with self.subTest(i=4, t="Non-existent NS"): |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 904 | self.db.del_one.reset_mock() |
| 905 | self.db.set_one.reset_mock() |
| 906 | self.msg.write.reset_mock() |
| 907 | self.fs.file_delete.reset_mock() |
| 908 | excp_msg = "Not found" |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 909 | with self.assertRaises( |
| 910 | DbException, msg="Accepted non-existent NSD ID" |
| 911 | ) as e: |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 912 | self.nsr_topic.delete(session2, "other_id") |
| garciadeblas | 4568a37 | 2021-03-24 09:19:48 +0100 | [diff] [blame] | 913 | self.assertEqual( |
| 914 | e.exception.http_code, HTTPStatus.NOT_FOUND, "Wrong HTTP status code" |
| 915 | ) |
| tierno | f5f2e3f | 2020-03-23 14:42:10 +0000 | [diff] [blame] | 916 | self.assertIn(excp_msg, str(e.exception), "Wrong exception text") |
| 917 | self.assertIn("other_id", str(e.exception), "Wrong exception text") |
| 918 | return |