osm_lcm/tests/test_ns.py

   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
  18
  19 import asynctest   # pip3 install asynctest --user
  20 import asyncio
  21 import yaml
  22 # import logging
  23 from os import getenv
  24 from osm_lcm.ns import NsLcm
  25 from osm_common.dbmemory import DbMemory
  26 from osm_common.msgkafka import MsgKafka
  27 from osm_common.fslocal import FsLocal
  28 from osm_lcm.lcm_utils import TaskRegistry
  29 from n2vc.vnf import N2VC
  30 # from n2vc.k8s_helm_conn import K8sHelmConnector
  31 from uuid import uuid4
  32 from asynctest.mock import patch
  33
  34 from osm_lcm.tests import test_db_descriptors as descriptors
  35
  36 __author__ = "Alfonso Tierno <alfonso.tiernosepulveda@telefonica.com>"
  37
  38 """ Perform unittests using asynctest of osm_lcm.ns module
  39 It allows, if some testing ENV are supplied, testing without mocking some external libraries for debugging:
  40     OSMLCMTEST_NS_PUBKEY: public ssh-key returned by N2VC to inject to VMs
  41     OSMLCMTEST_NS_NAME: change name of NS
  42     OSMLCMTEST_PACKAGES_PATH: path where the vnf-packages are stored (de-compressed), each one on a 'vnfd_id' folder
  43     OSMLCMTEST_NS_IPADDRESS: IP address where emulated VMs are reached. Comma separate list
  44     OSMLCMTEST_RO_VIMID: VIM id of RO target vim IP. Obtain it with openmano datcenter-list on RO container
  45     OSMLCMTEST_VCA_NOMOCK: Do no mock the VCA, N2VC library, for debugging it
  46     OSMLCMTEST_RO_NOMOCK: Do no mock the ROClient library, for debugging it
  47     OSMLCMTEST_DB_NOMOCK: Do no mock the database library, for debugging it
  48     OSMLCMTEST_FS_NOMOCK: Do no mock the File Storage library, for debugging it
  49     OSMLCMTEST_LOGGING_NOMOCK: Do no mock the logging
  50     OSMLCM_VCA_XXX: configuration of N2VC
  51     OSMLCM_RO_XXX: configuration of RO
  52 """
  53
  54
  55 vca_config = {   # TODO replace with os.get_env to get other configurations
  56     "host": getenv("OSMLCM_VCA_HOST", "vca"),
  57     "port": getenv("OSMLCM_VCA_PORT", 17070),
  58     "user": getenv("OSMLCM_VCA_USER", "admin"),
  59     "secret": getenv("OSMLCM_VCA_SECRET", "vca"),
  60     "pubkey": getenv("OSMLCM_VCA_PUBKEY", None),
  61     'cacert': getenv("OSMLCM_VCA_CACERT", None)
  62 }
  63
  64 ro_config = {
  65     "endpoint_url": "http://{}:{}/openmano".format(getenv("OSMLCM_RO_HOST", "ro"), getenv("OSMLCM_RO_PORT", "9090")),
  66     "tenant": getenv("OSMLCM_RO_TENANT", "osm"),
  67     "logger_name": "lcm.ROclient",
  68     "loglevel": "DEBUG",
  69 }
  70
  71
  72 class TestMyNS(asynctest.TestCase):
  73
  74     async def _n2vc_DeployCharms(self, model_name, application_name, vnfd, charm_path, params={}, machine_spec={},
  75                                  callback=None, *callback_args):
  76         if callback:
  77             for status, message in (("maintenance", "installing sofwware"), ("active", "Ready!")):
  78                 # call callback after some time
  79                 asyncio.sleep(5, loop=self.loop)
  80                 callback(model_name, application_name, status, message, *callback_args)
  81
  82     @staticmethod
  83     def _n2vc_FormatApplicationName(*args):
  84         num_calls = 0
  85         while True:
  86             yield "app_name-{}".format(num_calls)
  87             num_calls += 1
  88
  89     def _n2vc_CreateExecutionEnvironment(self, namespace, reuse_ee_id, db_dict):
  90         k_list = namespace.split(".")
  91         ee_id = k_list[1] + "."
  92         if len(k_list) >= 2:
  93             for k in k_list[2:4]:
  94                 ee_id += k[:8]
  95         else:
  96             ee_id += "_NS_"
  97         return ee_id, {}
  98
  99     def _ro_show(self, *args, **kwargs):
 100         ro_ns_desc = yaml.load(descriptors.db_ro_ns_text, Loader=yaml.Loader)
 101
 102         # if ip address provided, replace descriptor
 103         ip_addresses = getenv("OSMLCMTEST_NS_IPADDRESS", "")
 104         if ip_addresses:
 105             ip_addresses_list = ip_addresses.split(",")
 106             for vnf in ro_ns_desc["vnfs"]:
 107                 if not ip_addresses_list:
 108                     break
 109                 vnf["ip_address"] = ip_addresses_list[0]
 110                 for vm in vnf["vms"]:
 111                     if not ip_addresses_list:
 112                         break
 113                     vm["ip_address"] = ip_addresses_list.pop(0)
 114
 115         while True:
 116             yield ro_ns_desc
 117             for net in ro_ns_desc["nets"]:
 118                 if net["status"] != "ACTIVE":
 119                     net["status"] = "ACTIVE"
 120                     break
 121             else:
 122                 for vnf in ro_ns_desc["vnfs"]:
 123                     for vm in vnf["vms"]:
 124                         if vm["status"] != "ACTIVE":
 125                             vm["status"] = "ACTIVE"
 126                             break
 127
 128     def _ro_create(self, *args, **kwargs):
 129         while True:
 130             yield {"uuid": str(uuid4())}
 131
 132     def _return_uuid(self, *args, **kwargs):
 133         return str(uuid4())
 134
 135     @patch("osm_lcm.ns.N2VCJujuConnector")
 136     @patch("osm_lcm.ns.K8sHelmConnector")
 137     async def setUp(self, k8s_mock, n2vc_mock):
 138         # Mock DB
 139         if not getenv("OSMLCMTEST_DB_NOMOCK"):
 140             self.db = DbMemory()
 141             self.db.create_list("vnfds", yaml.load(descriptors.db_vnfds_text, Loader=yaml.Loader))
 142             self.db.create_list("nsds", yaml.load(descriptors.db_nsds_text, Loader=yaml.Loader))
 143             self.db.create_list("nsrs", yaml.load(descriptors.db_nsrs_text, Loader=yaml.Loader))
 144             self.db.create_list("vim_accounts", yaml.load(descriptors.db_vim_accounts_text, Loader=yaml.Loader))
 145             self.db.create_list("k8sclusters", yaml.load(descriptors.db_k8sclusters_text, Loader=yaml.Loader))
 146             self.db.create_list("nslcmops", yaml.load(descriptors.db_nslcmops_text, Loader=yaml.Loader))
 147             self.db.create_list("vnfrs", yaml.load(descriptors.db_vnfrs_text, Loader=yaml.Loader))
 148
 149             self.db_vim_accounts = yaml.load(descriptors.db_vim_accounts_text, Loader=yaml.Loader)
 150
 151         # Mock kafka
 152         self.msg = asynctest.Mock(MsgKafka())
 153
 154         # Mock filesystem
 155         if not getenv("OSMLCMTEST_FS_NOMOCK"):
 156             self.fs = asynctest.Mock(FsLocal())
 157             self.fs.get_params.return_value = {"path": getenv("OSMLCMTEST_PACKAGES_PATH", "./test/temp/packages")}
 158             self.fs.file_open = asynctest.mock_open()
 159             # self.fs.file_open.return_value.__enter__.return_value = asynctest.MagicMock()  # called on a python "with"
 160             # self.fs.file_open.return_value.__enter__.return_value.read.return_value = ""   # empty file
 161
 162         # Mock TaskRegistry
 163         self.lcm_tasks = asynctest.Mock(TaskRegistry())
 164         self.lcm_tasks.lock_HA.return_value = True
 165         self.lcm_tasks.waitfor_related_HA.return_value = None
 166         self.lcm_tasks.lookfor_related.return_value = ("", [])
 167
 168         # Create NsLCM class
 169         self.my_ns = NsLcm(self.db, self.msg, self.fs, self.lcm_tasks, ro_config, vca_config, self.loop)
 170
 171         # Mock logging
 172         if not getenv("OSMLCMTEST_LOGGING_NOMOCK"):
 173             self.my_ns.logger = asynctest.Mock(self.my_ns.logger)
 174
 175         # Mock VCA - N2VC
 176         if not getenv("OSMLCMTEST_VCA_NOMOCK"):
 177             pub_key = getenv("OSMLCMTEST_NS_PUBKEY", "ssh-rsa test-pub-key t@osm.com")
 178             self.my_ns.n2vc = asynctest.Mock(N2VC())
 179             self.my_ns.n2vc.GetPublicKey.return_value = getenv("OSMLCM_VCA_PUBKEY", "public_key")
 180             # allow several versions of n2vc
 181             self.my_ns.n2vc.FormatApplicationName = asynctest.Mock(side_effect=self._n2vc_FormatApplicationName())
 182             self.my_ns.n2vc.DeployCharms = asynctest.CoroutineMock(side_effect=self._n2vc_DeployCharms)
 183             self.my_ns.n2vc.create_execution_environment = asynctest.CoroutineMock(
 184                 side_effect=self._n2vc_CreateExecutionEnvironment)
 185             self.my_ns.n2vc.install_configuration_sw = asynctest.CoroutineMock(return_value=pub_key)
 186             self.my_ns.n2vc.get_ee_ssh_public__key = asynctest.CoroutineMock(return_value=pub_key)
 187             self.my_ns.n2vc.exec_primitive = asynctest.CoroutineMock(side_effect=self._return_uuid)
 188             self.my_ns.n2vc.GetPrimitiveStatus = asynctest.CoroutineMock(return_value="completed")
 189             self.my_ns.n2vc.GetPrimitiveOutput = asynctest.CoroutineMock(return_value={"result": "ok",
 190                                                                                        "pubkey": pub_key})
 191             self.my_ns.n2vc.get_public_key = asynctest.CoroutineMock(
 192                 return_value=getenv("OSMLCM_VCA_PUBKEY", "public_key"))
 193
 194         # # Mock VCA - K8s
 195         # if not getenv("OSMLCMTEST_VCA_K8s_NOMOCK"):
 196         #     pub_key = getenv("OSMLCMTEST_NS_PUBKEY", "ssh-rsa test-pub-key t@osm.com")
 197         #     self.my_ns.k8sclusterhelm = asynctest.Mock(K8sHelmConnector())
 198
 199         # Mock RO
 200         if not getenv("OSMLCMTEST_RO_NOMOCK"):
 201             # self.my_ns.RO = asynctest.Mock(ROclient.ROClient(self.loop, **ro_config))
 202             # TODO first time should be empty list, following should return a dict
 203             self.my_ns.RO.get_list = asynctest.CoroutineMock(self.my_ns.RO.get_list, return_value=[])
 204             self.my_ns.RO.create = asynctest.CoroutineMock(self.my_ns.RO.create, side_effect=self._ro_create())
 205             self.my_ns.RO.show = asynctest.CoroutineMock(self.my_ns.RO.show, side_effect=self._ro_show())
 206             self.my_ns.RO.create_action = asynctest.CoroutineMock(self.my_ns.RO.create_action,
 207                                                                   return_value={"vm-id": {"vim_result": 200,
 208                                                                                           "description": "done"}})
 209             # self.my_ns.wait_vm_up_insert_key_ro = asynctest.CoroutineMock(return_value="ip-address")
 210
 211     @asynctest.fail_on(active_handles=True)   # all async tasks must be completed
 212     async def test_instantiate(self):
 213         self.db.set_one = asynctest.Mock()
 214         nsr_id = self.db.get_list("nsrs")[0]["_id"]
 215         nslcmop_id = self.db.get_list("nslcmops")[0]["_id"]
 216         # print("Test instantiate started")
 217
 218         # delete deployed information of database
 219         if not getenv("OSMLCMTEST_DB_NOMOCK"):
 220             if self.db.get_list("nsrs")[0]["_admin"].get("deployed"):
 221                 del self.db.get_list("nsrs")[0]["_admin"]["deployed"]
 222             for db_vnfr in self.db.get_list("vnfrs"):
 223                 db_vnfr.pop("ip_address", None)
 224                 for db_vdur in db_vnfr["vdur"]:
 225                     db_vdur.pop("ip_address", None)
 226                     db_vdur.pop("mac_address", None)
 227             if getenv("OSMLCMTEST_RO_VIMID"):
 228                 self.db.get_list("vim_accounts")[0]["_admin"]["deployed"]["RO"] = getenv("OSMLCMTEST_RO_VIMID")
 229             if getenv("OSMLCMTEST_RO_VIMID"):
 230                 self.db.get_list("nsrs")[0]["_admin"]["deployed"]["RO"] = getenv("OSMLCMTEST_RO_VIMID")
 231
 232         await self.my_ns.instantiate(nsr_id, nslcmop_id)
 233
 234         # print("instantiate_result: {}".format(self.db.get_one("nslcmops",
 235         # {"_id": nslcmop_id}).get("detailed-status")))
 236
 237         self.msg.aiowrite.assert_called_once_with("ns", "instantiated",
 238                                                   {"nsr_id": nsr_id, "nslcmop_id": nslcmop_id,
 239                                                    "operationState": "COMPLETED"},
 240                                                   loop=self.loop)
 241         self.lcm_tasks.lock_HA.assert_called_once_with('ns', 'nslcmops', nslcmop_id)
 242         if not getenv("OSMLCMTEST_LOGGING_NOMOCK"):
 243             self.assertTrue(self.my_ns.logger.debug.called, "Debug method not called")
 244             self.my_ns.logger.error.assert_not_called()
 245             self.my_ns.logger.exception().assert_not_called()
 246
 247         if not getenv("OSMLCMTEST_DB_NOMOCK"):
 248             self.assertTrue(self.db.set_one.called, "db.set_one not called")
 249
 250         # TODO add more checks of called methods
 251         # TODO add a terminate
 252
 253     def test_ns_params_2_RO(self):
 254         vim = self.db.get_list("vim_accounts")[0]
 255         vim_id = vim["_id"]
 256         ro_vim_id = vim["_admin"]["deployed"]["RO"]
 257         ns_params = {"vimAccountId": vim_id}
 258         mgmt_interface = {"cp": "cp"}
 259         vdu = [{"id": "vdu_id", "interface": [{"external-connection-point-ref": "cp"}]}]
 260         vnfd_dict = {
 261             "1": {"vdu": vdu, "mgmt-interface": mgmt_interface},
 262             "2": {"vdu": vdu, "mgmt-interface": mgmt_interface, "vnf-configuration": None},
 263             "3": {"vdu": vdu, "mgmt-interface": mgmt_interface, "vnf-configuration": {"config-access": None}},
 264             "4": {"vdu": vdu, "mgmt-interface": mgmt_interface,
 265                   "vnf-configuration": {"config-access": {"ssh-access": None}}},
 266             "5": {"vdu": vdu, "mgmt-interface": mgmt_interface,
 267                   "vnf-configuration": {"config-access": {"ssh-access": {"required": True, "default_user": "U"}}}},
 268         }
 269         nsd = {"constituent-vnfd": []}
 270         for k in vnfd_dict.keys():
 271             nsd["constituent-vnfd"].append({"vnfd-id-ref": k, "member-vnf-index": k})
 272
 273         n2vc_key_list = ["key"]
 274         ro_ns_params = self.my_ns.ns_params_2_RO(ns_params, nsd, vnfd_dict, n2vc_key_list)
 275         ro_params_expected = {'wim_account': None, "datacenter": ro_vim_id,
 276                               "vnfs": {"5": {"vdus": {"vdu_id": {"mgmt_keys": n2vc_key_list}}}}}
 277         self.assertEqual(ro_ns_params, ro_params_expected)
 278
 279     # Test scale() and related methods
 280     @asynctest.fail_on(active_handles=True)   # all async tasks must be completed
 281     async def test_scale(self):
 282         # print("Test scale started")
 283
 284         # TODO: Add more higher-lever tests here, for example:
 285         # scale-out/scale-in operations with success/error result
 286
 287         # Test scale() with missing 'scaleVnfData', should return operationState = 'FAILED'
 288         nsr_id = self.db.get_list("nsrs")[0]["_id"]
 289         nslcmop_id = self.db.get_list("nslcmops")[0]["_id"]
 290         await self.my_ns.scale(nsr_id, nslcmop_id)
 291         expected_value = 'FAILED'
 292         return_value = self.db.get_one("nslcmops", {"_id": nslcmop_id}).get("operationState")
 293         self.assertEqual(return_value, expected_value)
 294         # print("scale_result: {}".format(self.db.get_one("nslcmops", {"_id": nslcmop_id}).get("detailed-status")))
 295
 296     # Test _reintent_or_skip_suboperation()
 297     # Expected result:
 298     # - if a suboperation's 'operationState' is marked as 'COMPLETED', SUBOPERATION_STATUS_SKIP is expected
 299     # - if marked as anything but 'COMPLETED', the suboperation index is expected
 300     def test_scale_reintent_or_skip_suboperation(self):
 301         # Load an alternative 'nslcmops' YAML for this test
 302         db_nslcmop = self.db.get_list('nslcmops')[0]
 303         op_index = 2
 304         # Test when 'operationState' is 'COMPLETED'
 305         db_nslcmop['_admin']['operations'][op_index]['operationState'] = 'COMPLETED'
 306         return_value = self.my_ns._reintent_or_skip_suboperation(db_nslcmop, op_index)
 307         expected_value = self.my_ns.SUBOPERATION_STATUS_SKIP
 308         self.assertEqual(return_value, expected_value)
 309         # Test when 'operationState' is not 'COMPLETED'
 310         db_nslcmop['_admin']['operations'][op_index]['operationState'] = None
 311         return_value = self.my_ns._reintent_or_skip_suboperation(db_nslcmop, op_index)
 312         expected_value = op_index
 313         self.assertEqual(return_value, expected_value)
 314
 315     # Test _find_suboperation()
 316     # Expected result: index of the found sub-operation, or SUBOPERATION_STATUS_NOT_FOUND if not found
 317     def test_scale_find_suboperation(self):
 318         # Load an alternative 'nslcmops' YAML for this test
 319         db_nslcmop = self.db.get_list('nslcmops')[0]
 320         # Find this sub-operation
 321         op_index = 2
 322         vnf_index = db_nslcmop['_admin']['operations'][op_index]['member_vnf_index']
 323         primitive = db_nslcmop['_admin']['operations'][op_index]['primitive']
 324         primitive_params = db_nslcmop['_admin']['operations'][op_index]['primitive_params']
 325         match = {
 326             'member_vnf_index': vnf_index,
 327             'primitive': primitive,
 328             'primitive_params': primitive_params,
 329         }
 330         found_op_index = self.my_ns._find_suboperation(db_nslcmop, match)
 331         self.assertEqual(found_op_index, op_index)
 332         # Test with not-matching params
 333         match = {
 334             'member_vnf_index': vnf_index,
 335             'primitive': '',
 336             'primitive_params': primitive_params,
 337         }
 338         found_op_index = self.my_ns._find_suboperation(db_nslcmop, match)
 339         self.assertEqual(found_op_index, self.my_ns.SUBOPERATION_STATUS_NOT_FOUND)
 340         # Test with None
 341         match = None
 342         found_op_index = self.my_ns._find_suboperation(db_nslcmop, match)
 343         self.assertEqual(found_op_index, self.my_ns.SUBOPERATION_STATUS_NOT_FOUND)
 344
 345     # Test _update_suboperation_status()
 346     def test_scale_update_suboperation_status(self):
 347         self.db.set_one = asynctest.Mock()
 348         db_nslcmop = self.db.get_list('nslcmops')[0]
 349         op_index = 0
 350         # Force the initial values to be distinct from the updated ones
 351         q_filter = {"_id": db_nslcmop["_id"]}
 352         # Test to change 'operationState' and 'detailed-status'
 353         operationState = 'COMPLETED'
 354         detailed_status = 'Done'
 355         expected_update_dict = {'_admin.operations.0.operationState': operationState,
 356                                 '_admin.operations.0.detailed-status': detailed_status,
 357                                 }
 358         self.my_ns._update_suboperation_status(db_nslcmop, op_index, operationState, detailed_status)
 359         self.db.set_one.assert_called_once_with("nslcmops", q_filter=q_filter, update_dict=expected_update_dict,
 360                                                 fail_on_empty=False)
 361
 362     def test_scale_add_suboperation(self):
 363         db_nslcmop = self.db.get_list('nslcmops')[0]
 364         vnf_index = '1'
 365         num_ops_before = len(db_nslcmop.get('_admin', {}).get('operations', [])) - 1
 366         vdu_id = None
 367         vdu_count_index = None
 368         vdu_name = None
 369         primitive = 'touch'
 370         mapped_primitive_params = {'parameter':
 371                                    [{'data-type': 'STRING',
 372                                      'name': 'filename',
 373                                      'default-value': '<touch_filename2>'}],
 374                                    'name': 'touch'}
 375         operationState = 'PROCESSING'
 376         detailed_status = 'In progress'
 377         operationType = 'PRE-SCALE'
 378         # Add a 'pre-scale' suboperation
 379         op_index_after = self.my_ns._add_suboperation(db_nslcmop, vnf_index, vdu_id, vdu_count_index,
 380                                                       vdu_name, primitive, mapped_primitive_params,
 381                                                       operationState, detailed_status, operationType)
 382         self.assertEqual(op_index_after, num_ops_before + 1)
 383
 384         # Delete all suboperations and add the same operation again
 385         del db_nslcmop['_admin']['operations']
 386         op_index_zero = self.my_ns._add_suboperation(db_nslcmop, vnf_index, vdu_id, vdu_count_index,
 387                                                      vdu_name, primitive, mapped_primitive_params,
 388                                                      operationState, detailed_status, operationType)
 389         self.assertEqual(op_index_zero, 0)
 390
 391         # Add a 'RO' suboperation
 392         RO_nsr_id = '1234567890'
 393         RO_scaling_info = [{'type': 'create', 'count': 1, 'member-vnf-index': '1', 'osm_vdu_id': 'dataVM'}]
 394         op_index = self.my_ns._add_suboperation(db_nslcmop, vnf_index, vdu_id, vdu_count_index,
 395                                                 vdu_name, primitive, mapped_primitive_params,
 396                                                 operationState, detailed_status, operationType,
 397                                                 RO_nsr_id, RO_scaling_info)
 398         db_RO_nsr_id = db_nslcmop['_admin']['operations'][op_index]['RO_nsr_id']
 399         self.assertEqual(op_index, 1)
 400         self.assertEqual(RO_nsr_id, db_RO_nsr_id)
 401
 402         # Try to add an invalid suboperation, should return SUBOPERATION_STATUS_NOT_FOUND
 403         op_index_invalid = self.my_ns._add_suboperation(None, None, None, None, None,
 404                                                         None, None, None,
 405                                                         None, None, None)
 406         self.assertEqual(op_index_invalid, self.my_ns.SUBOPERATION_STATUS_NOT_FOUND)
 407
 408     # Test _check_or_add_scale_suboperation() and _check_or_add_scale_suboperation_RO()
 409     # check the possible return values:
 410     # - SUBOPERATION_STATUS_NEW: This is a new sub-operation
 411     # - op_index (non-negative number): This is an existing sub-operation, operationState != 'COMPLETED'
 412     # - SUBOPERATION_STATUS_SKIP: This is an existing sub-operation, operationState == 'COMPLETED'
 413     def test_scale_check_or_add_scale_suboperation(self):
 414         db_nslcmop = self.db.get_list('nslcmops')[0]
 415         operationType = 'PRE-SCALE'
 416         vnf_index = '1'
 417         primitive = 'touch'
 418         primitive_params = {'parameter':
 419                             [{'data-type': 'STRING',
 420                               'name': 'filename',
 421                               'default-value': '<touch_filename2>'}],
 422                             'name': 'touch'}
 423
 424         # Delete all sub-operations to be sure this is a new sub-operation
 425         del db_nslcmop['_admin']['operations']
 426
 427         # Add a new sub-operation
 428         # For new sub-operations, operationState is set to 'PROCESSING' by default
 429         op_index_new = self.my_ns._check_or_add_scale_suboperation(
 430             db_nslcmop, vnf_index, primitive, primitive_params, operationType)
 431         self.assertEqual(op_index_new, self.my_ns.SUBOPERATION_STATUS_NEW)
 432
 433         # Use the same parameters again to match the already added sub-operation
 434         # which has status 'PROCESSING' (!= 'COMPLETED') by default
 435         # The expected return value is a non-negative number
 436         op_index_existing = self.my_ns._check_or_add_scale_suboperation(
 437             db_nslcmop, vnf_index, primitive, primitive_params, operationType)
 438         self.assertTrue(op_index_existing >= 0)
 439
 440         # Change operationState 'manually' for this sub-operation
 441         db_nslcmop['_admin']['operations'][op_index_existing]['operationState'] = 'COMPLETED'
 442         # Then use the same parameters again to match the already added sub-operation,
 443         # which now has status 'COMPLETED'
 444         # The expected return value is SUBOPERATION_STATUS_SKIP
 445         op_index_skip = self.my_ns._check_or_add_scale_suboperation(
 446             db_nslcmop, vnf_index, primitive, primitive_params, operationType)
 447         self.assertEqual(op_index_skip, self.my_ns.SUBOPERATION_STATUS_SKIP)
 448
 449         # RO sub-operation test:
 450         # Repeat tests for the very similar _check_or_add_scale_suboperation_RO(),
 451         RO_nsr_id = '1234567890'
 452         RO_scaling_info = [{'type': 'create', 'count': 1, 'member-vnf-index': '1', 'osm_vdu_id': 'dataVM'}]
 453         op_index_new_RO = self.my_ns._check_or_add_scale_suboperation(
 454             db_nslcmop, vnf_index, None, None, 'SCALE-RO', RO_nsr_id, RO_scaling_info)
 455         self.assertEqual(op_index_new_RO, self.my_ns.SUBOPERATION_STATUS_NEW)
 456
 457         # Use the same parameters again to match the already added RO sub-operation
 458         op_index_existing_RO = self.my_ns._check_or_add_scale_suboperation(
 459             db_nslcmop, vnf_index, None, None, 'SCALE-RO', RO_nsr_id, RO_scaling_info)
 460         self.assertTrue(op_index_existing_RO >= 0)
 461
 462         # Change operationState 'manually' for this RO sub-operation
 463         db_nslcmop['_admin']['operations'][op_index_existing_RO]['operationState'] = 'COMPLETED'
 464         # Then use the same parameters again to match the already added sub-operation,
 465         # which now has status 'COMPLETED'
 466         # The expected return value is SUBOPERATION_STATUS_SKIP
 467         op_index_skip_RO = self.my_ns._check_or_add_scale_suboperation(
 468             db_nslcmop, vnf_index, None, None, 'SCALE-RO', RO_nsr_id, RO_scaling_info)
 469         self.assertEqual(op_index_skip_RO, self.my_ns.SUBOPERATION_STATUS_SKIP)
 470
 471     async def test_deploy_kdus(self):
 472         db_nsr = self.db.get_list("nsrs")[1]
 473         db_vnfr = self.db.get_list("vnfrs")[2]
 474         db_vnfrs = {"multikdu": db_vnfr}
 475         nsr_id = db_nsr["_id"]
 476         # nslcmop_id = self.db.get_list("nslcmops")[1]["_id"]
 477         logging_text = "KDU"
 478         self.my_ns.k8sclusterhelm.install = asynctest.CoroutineMock(return_value="k8s_id")
 479         await self.my_ns.deploy_kdus(logging_text, nsr_id, db_nsr, db_vnfrs)
 480         db_nsr = self.db.get_list("nsrs")[1]
 481         self.assertIn("K8s", db_nsr["_admin"]["deployed"], "K8s entry not created at '_admin.deployed'")
 482         self.assertIsInstance(db_nsr["_admin"]["deployed"]["K8s"], list, "K8s entry is not of type list")
 483         self.assertEqual(len(db_nsr["_admin"]["deployed"]["K8s"]), 2, "K8s entry is not of type list")
 484         k8s_instace_info = {"kdu-instance": "k8s_id", "k8scluster-uuid": "73d96432-d692-40d2-8440-e0c73aee209c",
 485                             "k8scluster-type": "chart",
 486                             "kdu-name": "ldap", "kdu-model": "stable/openldap:1.2.1"}
 487
 488         self.assertEqual(db_nsr["_admin"]["deployed"]["K8s"][0], k8s_instace_info)
 489         k8s_instace_info["kdu-name"] = "mongo"
 490         k8s_instace_info["kdu-model"] = "stable/mongodb"
 491         self.assertEqual(db_nsr["_admin"]["deployed"]["K8s"][1], k8s_instace_info)
 492
 493
 494 if __name__ == '__main__':
 495     asynctest.main()