예제 #1
0
    def setUp(self):
        self.db = DbMemory()
        self.fs = Mock(FsBase())
        self.fs.get_params.return_value = {"./fake/folder"}
        self.fs.file_open = mock_open()
        self.msg = Mock(MsgBase())
        # create class
        self.nslcmop_topic = NsLcmOpTopic(self.db, self.fs, self.msg, None)
        self.nslcmop_topic.check_quota = Mock(return_value=None)  # skip quota

        self.db.create_list(
            "vim_accounts", yaml.load(db_vim_accounts_text,
                                      Loader=yaml.Loader))
        self.db.create_list("nsds", yaml.load(db_nsds_text,
                                              Loader=yaml.Loader))
        self.db.create_list("vnfds",
                            yaml.load(db_vnfds_text, Loader=yaml.Loader))
        self.db.create_list("vnfrs",
                            yaml.load(db_vnfrs_text, Loader=yaml.Loader))
        self.db.create_list("nsrs", yaml.load(db_nsrs_text,
                                              Loader=yaml.Loader))
        self.db.create = Mock(return_value="created_id")
        self.db.set_one = Mock(return_value={"updated": 1})
        self.nsd = self.db.get_list("nsds")[0]
        self.nsd_id = self.nsd["_id"]
        self.nsr = self.db.get_list("nsrs")[0]
        self.nsr_id = self.nsr["_id"]
        self.nsr_project = self.nsr["_admin"]["projects_read"][0]

        self.vim = self.db.get_list("vim_accounts")[0]
        self.vim_id = self.vim["_id"]
예제 #2
0
 def setUp(self):
     self.db = DbMemory()
     self.pmjobs_topic = PmJobsTopic(self.db, host="prometheus", port=9091)
     self.db.create_list("nsds", yaml.load(db_nsds_text, Loader=yaml.Loader))
     self.db.create_list("vnfds", yaml.load(db_vnfds_text, Loader=yaml.Loader))
     self.db.create_list("vnfrs", yaml.load(db_vnfrs_text, Loader=yaml.Loader))
     self.db.create_list("nsrs", yaml.load(db_nsrs_text, Loader=yaml.Loader))
     self.nsr = self.db.get_list("nsrs")[0]
     self.nsr_id = self.nsr["_id"]
     project_id = self.nsr["_admin"]["projects_write"]
     """metric_check_list contains the vnf metric name used in descriptor i.e users,load"""
     self.metric_check_list = ['cpu_utilization', 'average_memory_utilization', 'disk_read_ops',
                               'disk_write_ops', 'disk_read_bytes', 'disk_write_bytes',
                               'packets_dropped', 'packets_sent', 'packets_received', 'users', 'load']
     self.session = {"username": "******", "project_id": project_id, "method": None,
                     "admin": True, "force": False, "public": False, "allow_show_user_project_role": True}
예제 #3
0
 async def setUp(self):
     self.db = Mock(DbMemory())
     self.fs = asynctest.Mock(FsLocal())
     self.fs.path = "/app/storage"
     vca_config = {}
     lcm_helm_conn.K8sHelmConnector = asynctest.Mock(
         lcm_helm_conn.K8sHelmConnector)
     self.helm_conn = LCMHelmConn(self.db,
                                  self.fs,
                                  loop=self.loop,
                                  vca_config=vca_config,
                                  log=self.logger)
예제 #4
0
class TestNsLcmOpTopic(unittest.TestCase):
    def setUp(self):
        self.db = DbMemory()
        self.fs = Mock(FsBase())
        self.fs.get_params.return_value = {"./fake/folder"}
        self.fs.file_open = mock_open()
        self.msg = Mock(MsgBase())
        # create class
        self.nslcmop_topic = NsLcmOpTopic(self.db, self.fs, self.msg, None)
        self.nslcmop_topic.check_quota = Mock(return_value=None)  # skip quota

        self.db.create_list(
            "vim_accounts", yaml.load(db_vim_accounts_text,
                                      Loader=yaml.Loader))
        self.db.create_list("nsds", yaml.load(db_nsds_text,
                                              Loader=yaml.Loader))
        self.db.create_list("vnfds",
                            yaml.load(db_vnfds_text, Loader=yaml.Loader))
        self.db.create_list("vnfrs",
                            yaml.load(db_vnfrs_text, Loader=yaml.Loader))
        self.db.create_list("nsrs", yaml.load(db_nsrs_text,
                                              Loader=yaml.Loader))
        self.db.create = Mock(return_value="created_id")
        self.db.set_one = Mock(return_value={"updated": 1})
        self.nsd = self.db.get_list("nsds")[0]
        self.nsd_id = self.nsd["_id"]
        self.nsr = self.db.get_list("nsrs")[0]
        self.nsr_id = self.nsr["_id"]
        self.nsr_project = self.nsr["_admin"]["projects_read"][0]

        self.vim = self.db.get_list("vim_accounts")[0]
        self.vim_id = self.vim["_id"]

    def test_create_instantiate(self):
        session = {
            "force": False,
            "admin": False,
            "public": False,
            "project_id": [self.nsr_project],
            "method": "write"
        }
        indata = {
            "nsdId":
            self.nsd_id,
            "nsInstanceId":
            self.nsr_id,
            "nsName":
            "name",
            "vimAccountId":
            self.vim_id,
            "additionalParamsForVnf": [{
                "member-vnf-index": "1",
                "additionalParams": {
                    "touch_filename": "file"
                }
            }, {
                "member-vnf-index": "2",
                "additionalParams": {
                    "touch_filename": "file"
                }
            }],
            "vnf": [{
                "member-vnf-index":
                "1",
                "vdu": [{
                    "id":
                    "dataVM",
                    "interface": [{
                        "name": "dataVM-eth0",
                        "ip-address": "10.11.12.13",
                        "floating-ip-required": True
                    }]
                }],
                "internal-vld": [{
                    "name": "internal",
                    "vim-network-id": "vim-net-id"
                }]
            }],
            "lcmOperationType":
            "instantiate",
        }
        rollback = []
        headers = {}

        nslcmop_id, _ = self.nslcmop_topic.new(rollback,
                                               session,
                                               indata=indata,
                                               kwargs=None,
                                               headers=headers)

        # check nslcmop is created at database
        self.assertEqual(
            self.db.create.call_count, 1,
            "database create not called, or called more than once")
        _call = self.db.create.call_args_list[0]
        self.assertEqual(_call[0][0], "nslcmops",
                         "must be create a nslcmops entry at database")

        created_nslcmop = _call[0][1]
        self.assertEqual(nslcmop_id, created_nslcmop["_id"],
                         "mismatch between return id and database '_id'")
        self.assertEqual(self.nsr_id, created_nslcmop["nsInstanceId"],
                         "bad reference id from nslcmop to nsr")
        self.assertTrue(created_nslcmop["_admin"].get("projects_read"),
                        "Database record must contain '_amdin.projects_read'")
        self.assertIn("created", created_nslcmop["_admin"],
                      "Database record must contain '_admin.created'")
        self.assertTrue(
            created_nslcmop["lcmOperationType"] == "instantiate",
            "Database record must contain 'lcmOperationType=instantiate'")

        self.assertEqual(
            len(rollback),
            len(self.db.set_one.call_args_list) + 1,
            "rollback mismatch with created/set items at database")

        # test parameters with error
        bad_id = "88d90b0c-faff-4b9f-bccd-aaaaaaaaaaaa"
        test_set = (
            ("nsr not found", {
                "nsInstanceId": bad_id
            }, DbException, HTTPStatus.NOT_FOUND, ("not found", bad_id)),
            # TODO add "nsd"
            # ({"vimAccountId": bad_id}, DbException, HTTPStatus.NOT_FOUND, ("not found", bad_id)),  # TODO add "vim"
            ("bad member-vnf-index", {
                "vnf.0.member-vnf-index": "k"
            }, EngineException, HTTPStatus.BAD_REQUEST, ("k", )),
        )
        for message, kwargs_, expect_exc, expect_code, expect_text_list in test_set:
            with self.assertRaises(expect_exc, msg=message) as e:
                self.nslcmop_topic.new(rollback,
                                       session,
                                       indata=deepcopy(indata),
                                       kwargs=kwargs_,
                                       headers=headers)
            if expect_code:
                self.assertTrue(e.exception.http_code == expect_code)
            if expect_text_list:
                for expect_text in expect_text_list:
                    self.assertIn(
                        expect_text,
                        str(e.exception).lower(),
                        "Expected '{}' at exception text".format(expect_text))

    def test_check_ns_operation_action(self):
        nsrs = self.db.get_list("nsrs")[0]
        session = {}

        indata = {
            "member_vnf_index": "1",
            "vdu_id": None,
            "primitive": "touch",
            "primitive_params": {
                "filename": "file"
            }
        }

        self.nslcmop_topic._check_ns_operation(session, nsrs, "action", indata)
        for k in indata:
            indata_copy = indata.copy()
            if k == "primitive_params":
                continue
            indata_copy[k] = "non_existing"
            with self.assertRaises(EngineException) as exc_manager:
                self.nslcmop_topic._check_ns_operation(session, nsrs, "action",
                                                       indata_copy)
            exc = exc_manager.exception
            self.assertEqual(
                exc.http_code, HTTPStatus.BAD_REQUEST,
                "Engine exception bad http_code with {}".format(indata_copy))
예제 #5
0
class TestNsrTopic(unittest.TestCase):
    def setUp(self):
        self.db = DbMemory()
        self.fs = Mock(FsBase())
        self.fs.get_params.return_value = {"./fake/folder"}
        self.fs.file_open = mock_open()
        self.msg = Mock(MsgBase())
        # create class
        self.nsr_topic = NsrTopic(self.db, self.fs, self.msg, None)
        self.nsr_topic.check_quota = Mock(return_value=None)  # skip quota

        self.db.create_list(
            "vim_accounts", yaml.load(db_vim_accounts_text,
                                      Loader=yaml.Loader))
        self.db.create_list("nsds", yaml.load(db_nsds_text,
                                              Loader=yaml.Loader))
        self.db.create_list("vnfds",
                            yaml.load(db_vnfds_text, Loader=yaml.Loader))
        self.db.create = Mock(return_value="created_id")
        self.nsd = self.db.get_list("nsds")[0]
        self.nsd_id = self.nsd["_id"]
        self.nsd_project = self.nsd["_admin"]["projects_read"][0]

        self.vim = self.db.get_list("vim_accounts")[0]
        self.vim_id = self.vim["_id"]

    def test_create(self):
        session = {
            "force": False,
            "admin": False,
            "public": False,
            "project_id": [self.nsd_project],
            "method": "write"
        }
        indata = {
            "nsdId":
            self.nsd_id,
            "nsName":
            "name",
            "vimAccountId":
            self.vim_id,
            "additionalParamsForVnf": [{
                "member-vnf-index": "1",
                "additionalParams": {
                    "touch_filename": "file"
                }
            }, {
                "member-vnf-index": "2",
                "additionalParams": {
                    "touch_filename": "file"
                }
            }]
        }
        rollback = []
        headers = {}

        self.nsr_topic.new(rollback,
                           session,
                           indata=indata,
                           kwargs=None,
                           headers=headers)

        # check vnfrs and nsrs created in whatever order
        created_vnfrs = []
        created_nsrs = []
        nsr_id = None
        for _call in self.db.create.call_args_list:
            assert len(_call[0]) >= 2, "called db.create with few parameters"
            created_item = _call[0][1]
            if _call[0][0] == "vnfrs":
                created_vnfrs.append(created_item)
                self.assertIn(
                    "member-vnf-index-ref", created_item,
                    "Created item must contain member-vnf-index-ref section")
                if nsr_id:
                    self.assertEqual(nsr_id, created_item["nsr-id-ref"],
                                     "bad reference id from vnfr to nsr")
                else:
                    nsr_id = created_item["nsr-id-ref"]

            elif _call[0][0] == "nsrs":
                created_nsrs.append(created_item)
                if nsr_id:
                    self.assertEqual(nsr_id, created_item["_id"],
                                     "bad reference id from vnfr to nsr")
                else:
                    nsr_id = created_item["_id"]
            else:
                assert True, "created an unknown record {} at database".format(
                    _call[0][0])

            self.assertTrue(
                created_item["_admin"].get("projects_read"),
                "Database record must contain '_amdin.projects_read'")
            self.assertIn("created", created_item["_admin"],
                          "Database record must contain '_admin.created'")
            self.assertTrue(
                created_item["_admin"]["nsState"] == "NOT_INSTANTIATED",
                "Database record must contain '_admin.nstate=NOT INSTANTIATE'")

        self.assertEqual(len(created_vnfrs), len(self.nsd["constituent-vnfd"]),
                         "created a mismatch number of vnfr at database")
        self.assertEqual(len(created_nsrs), 1,
                         "Only one nsrs must be created at database")
        self.assertEqual(len(rollback),
                         len(created_vnfrs) + 1,
                         "rollback mismatch with created items at database")

        # test parameters with error
        bad_id = "88d90b0c-faff-4b9f-bccd-aaaaaaaaaaaa"
        test_set = (
            # TODO add "nsd"
            ("nsd not found", {
                "nsdId": bad_id
            }, DbException, HTTPStatus.NOT_FOUND, ("not found", bad_id)),
            # ({"vimAccountId": bad_id}, DbException, HTTPStatus.NOT_FOUND, ("not found", bad_id)),  # TODO add "vim"
            ("additional params not supply", {
                "additionalParamsForVnf.0.member-vnf-index": "k"
            }, EngineException, HTTPStatus.BAD_REQUEST, None),
        )
        for message, kwargs_, expect_exc, expect_code, expect_text_list in test_set:
            with self.assertRaises(expect_exc, msg=message) as e:
                self.nsr_topic.new(rollback,
                                   session,
                                   indata=deepcopy(indata),
                                   kwargs=kwargs_,
                                   headers=headers)
            if expect_code:
                self.assertTrue(e.exception.http_code == expect_code)
            if expect_text_list:
                for expect_text in expect_text_list:
                    self.assertIn(
                        expect_text,
                        str(e.exception).lower(),
                        "Expected '{}' at exception text".format(expect_text))
예제 #6
0
class PmJobsTopicTest(asynctest.TestCase):

    def setUp(self):
        self.db = DbMemory()
        self.pmjobs_topic = PmJobsTopic(self.db, host="prometheus", port=9091)
        self.db.create_list("nsds", yaml.load(db_nsds_text, Loader=yaml.Loader))
        self.db.create_list("vnfds", yaml.load(db_vnfds_text, Loader=yaml.Loader))
        self.db.create_list("vnfrs", yaml.load(db_vnfrs_text, Loader=yaml.Loader))
        self.db.create_list("nsrs", yaml.load(db_nsrs_text, Loader=yaml.Loader))
        self.nsr = self.db.get_list("nsrs")[0]
        self.nsr_id = self.nsr["_id"]
        project_id = self.nsr["_admin"]["projects_write"]
        """metric_check_list contains the vnf metric name used in descriptor i.e users,load"""
        self.metric_check_list = ['cpu_utilization', 'average_memory_utilization', 'disk_read_ops',
                                  'disk_write_ops', 'disk_read_bytes', 'disk_write_bytes',
                                  'packets_dropped', 'packets_sent', 'packets_received', 'users', 'load']
        self.session = {"username": "******", "project_id": project_id, "method": None,
                        "admin": True, "force": False, "public": False, "allow_show_user_project_role": True}

    def set_get_mock_res(self, mock_res, ns_id, metric_list):
        site = "http://prometheus:9091/api/v1/query?query=osm_metric_name{ns_id='nsr'}"
        site = re.sub(r'nsr', ns_id, site)
        for metric in metric_list:
            endpoint = re.sub(r'metric_name', metric, site)
            if metric == 'cpu_utilization':
                response = yaml.load(cpu_utilization, Loader=yaml.Loader)
            elif metric == 'users':
                response = yaml.load(users, Loader=yaml.Loader)
            elif metric == 'load':
                response = yaml.load(load, Loader=yaml.Loader)
            else:
                response = yaml.load(empty, Loader=yaml.Loader)
            mock_res.get(endpoint, payload=response)

    def test_get_vnf_metric_list(self):
        with self.subTest("Test case1 failed in test_get_vnf_metric_list"):
            metric_list = self.pmjobs_topic._get_vnf_metric_list(self.nsr_id)
            self.assertCountEqual(metric_list, self.metric_check_list,
                                  "VNF metric list is not correctly fetched")
        with self.subTest("Test case2 failed in test_get_vnf_metric_list"):
            wrong_ns_id = "88d90b0c-faff-4bbc-cccc-aaaaaaaaaaaa"
            with self.assertRaises(EngineException, msg="ns not found") as e:
                self.pmjobs_topic._get_vnf_metric_list(wrong_ns_id)
            self.assertEqual(e.exception.http_code, HTTPStatus.NOT_FOUND, "Wrong HTTP status code")
            self.assertIn("NS not found with id {}".format(wrong_ns_id),
                          str(e.exception), "Wrong exception text")

    async def test_prom_metric_request(self):
        with self.subTest("Test case1 failed in test_prom"):
            prom_response = yaml.load(prom_res, Loader=yaml.Loader)
            with aioresponses() as mock_res:
                self.set_get_mock_res(mock_res, self.nsr_id, self.metric_check_list)
                result = await self.pmjobs_topic._prom_metric_request(self.nsr_id, self.metric_check_list)
            self.assertCountEqual(result, prom_response, "Metric Data is valid")
        with self.subTest("Test case2 failed in test_prom"):
            with self.assertRaises(EngineException, msg="Prometheus not reachable") as e:
                await self.pmjobs_topic._prom_metric_request(self.nsr_id, self.metric_check_list)
            self.assertIn("Connection to ", str(e.exception), "Wrong exception text")

    def test_show(self):
        with self.subTest("Test case1 failed in test_show"):
            show_response = yaml.load(show_res, Loader=yaml.Loader)
            with aioresponses() as mock_res:
                self.set_get_mock_res(mock_res, self.nsr_id, self.metric_check_list)
                result = self.pmjobs_topic.show(self.session, self.nsr_id)
            self.assertEqual(len(result['entries']), 3, "Number of metrics returned")
            self.assertCountEqual(result, show_response, "Response is valid")
        with self.subTest("Test case2 failed in test_show"):
            wrong_ns_id = "88d90b0c-faff-4bbc-cccc-aaaaaaaaaaaa"
            with aioresponses() as mock_res:
                self.set_get_mock_res(mock_res, wrong_ns_id, self.metric_check_list)
                with self.assertRaises(EngineException, msg="ns not found") as e:
                    self.pmjobs_topic.show(self.session, wrong_ns_id)
                self.assertEqual(e.exception.http_code, HTTPStatus.NOT_FOUND, "Wrong HTTP status code")
                self.assertIn("NS not found with id {}".format(wrong_ns_id), str(e.exception),
                              "Wrong exception text")
예제 #7
0
 async def setUp(self):
     config = {'uri': 'http:prometheus:9090', 'path': '/etc/prometheus'}
     self.db = Mock(DbMemory())
     self.p = Prometheus(config, worker_id='1', db=self.db, loop=self.loop)
예제 #8
0
파일: test_ns.py 프로젝트: OSMadmin/LCM
class TestMyNS(asynctest.TestCase):
    async def _n2vc_DeployCharms(self,
                                 model_name,
                                 application_name,
                                 vnfd,
                                 charm_path,
                                 params={},
                                 machine_spec={},
                                 callback=None,
                                 *callback_args):
        if callback:
            for status, message in (("maintenance", "installing sofwware"),
                                    ("active", "Ready!")):
                # call callback after some time
                asyncio.sleep(5, loop=self.loop)
                callback(model_name, application_name, status, message,
                         *callback_args)

    @staticmethod
    def _n2vc_FormatApplicationName(*args):
        num_calls = 0
        while True:
            yield "app_name-{}".format(num_calls)
            num_calls += 1

    def _n2vc_CreateExecutionEnvironment(self, namespace, reuse_ee_id, db_dict,
                                         *args, **kwargs):
        k_list = namespace.split(".")
        ee_id = k_list[1] + "."
        if len(k_list) >= 2:
            for k in k_list[2:4]:
                ee_id += k[:8]
        else:
            ee_id += "_NS_"
        return ee_id, {}

    def _ro_show(self, *args, **kwargs):
        if kwargs.get("delete"):
            ro_ns_desc = yaml.load(descriptors.ro_delete_action_text,
                                   Loader=yaml.Loader)
            while True:
                yield ro_ns_desc

        ro_ns_desc = yaml.load(descriptors.ro_ns_text, Loader=yaml.Loader)

        # if ip address provided, replace descriptor
        ip_addresses = getenv("OSMLCMTEST_NS_IPADDRESS", "")
        if ip_addresses:
            ip_addresses_list = ip_addresses.split(",")
            for vnf in ro_ns_desc["vnfs"]:
                if not ip_addresses_list:
                    break
                vnf["ip_address"] = ip_addresses_list[0]
                for vm in vnf["vms"]:
                    if not ip_addresses_list:
                        break
                    vm["ip_address"] = ip_addresses_list.pop(0)

        while True:
            yield ro_ns_desc
            for net in ro_ns_desc["nets"]:
                if net["status"] != "ACTIVE":
                    net["status"] = "ACTIVE"
                    break
            else:
                for vnf in ro_ns_desc["vnfs"]:
                    for vm in vnf["vms"]:
                        if vm["status"] != "ACTIVE":
                            vm["status"] = "ACTIVE"
                            break

    def _ro_create(self, *args, **kwargs):
        while True:
            yield {"uuid": str(uuid4())}

    def _return_uuid(self, *args, **kwargs):
        return str(uuid4())

    async def setUp(self):

        # Mock DB
        if not getenv("OSMLCMTEST_DB_NOMOCK"):
            self.db = DbMemory()
            self.db.create_list(
                "vnfds",
                yaml.load(descriptors.db_vnfds_text, Loader=yaml.Loader))
            self.db.create_list(
                "nsds", yaml.load(descriptors.db_nsds_text,
                                  Loader=yaml.Loader))
            self.db.create_list(
                "nsrs", yaml.load(descriptors.db_nsrs_text,
                                  Loader=yaml.Loader))
            self.db.create_list(
                "vim_accounts",
                yaml.load(descriptors.db_vim_accounts_text,
                          Loader=yaml.Loader))
            self.db.create_list(
                "k8sclusters",
                yaml.load(descriptors.db_k8sclusters_text, Loader=yaml.Loader))
            self.db.create_list(
                "nslcmops",
                yaml.load(descriptors.db_nslcmops_text, Loader=yaml.Loader))
            self.db.create_list(
                "vnfrs",
                yaml.load(descriptors.db_vnfrs_text, Loader=yaml.Loader))
            self.db_vim_accounts = yaml.load(descriptors.db_vim_accounts_text,
                                             Loader=yaml.Loader)

        # Mock kafka
        self.msg = asynctest.Mock(MsgKafka())

        # Mock filesystem
        if not getenv("OSMLCMTEST_FS_NOMOCK"):
            self.fs = asynctest.Mock(FsLocal())
            self.fs.get_params.return_value = {
                "path": getenv("OSMLCMTEST_PACKAGES_PATH",
                               "./test/temp/packages")
            }
            self.fs.file_open = asynctest.mock_open()
            # self.fs.file_open.return_value.__enter__.return_value = asynctest.MagicMock()  # called on a python "with"
            # self.fs.file_open.return_value.__enter__.return_value.read.return_value = ""   # empty file

        # Mock TaskRegistry
        self.lcm_tasks = asynctest.Mock(TaskRegistry())
        self.lcm_tasks.lock_HA.return_value = True
        self.lcm_tasks.waitfor_related_HA.return_value = None
        self.lcm_tasks.lookfor_related.return_value = ("", [])

        # Mock VCA - K8s
        if not getenv("OSMLCMTEST_VCA_K8s_NOMOCK"):
            ns.K8sJujuConnector = asynctest.MagicMock(ns.K8sJujuConnector)
            ns.K8sHelmConnector = asynctest.MagicMock(ns.K8sHelmConnector)

        if not getenv("OSMLCMTEST_VCA_NOMOCK"):
            ns.N2VCJujuConnector = asynctest.MagicMock(ns.N2VCJujuConnector)
            ns.N2VCJujuConnectorLCM = asynctest.MagicMock(
                ns.N2VCJujuConnectorLCM)
            ns.LCMHelmConn = asynctest.MagicMock(ns.LCMHelmConn)

        # Create NsLCM class
        self.my_ns = ns.NsLcm(self.db, self.msg, self.fs, self.lcm_tasks,
                              lcm_config, self.loop)
        self.my_ns._wait_dependent_n2vc = asynctest.CoroutineMock()

        # Mock logging
        if not getenv("OSMLCMTEST_LOGGING_NOMOCK"):
            self.my_ns.logger = asynctest.Mock(self.my_ns.logger)

        # Mock VCA - N2VC
        if not getenv("OSMLCMTEST_VCA_NOMOCK"):
            pub_key = getenv("OSMLCMTEST_NS_PUBKEY",
                             "ssh-rsa test-pub-key [email protected]")
            # self.my_ns.n2vc = asynctest.Mock(N2VC())
            self.my_ns.n2vc.GetPublicKey.return_value = getenv(
                "OSMLCM_VCA_PUBKEY", "public_key")
            # allow several versions of n2vc
            self.my_ns.n2vc.FormatApplicationName = asynctest.Mock(
                side_effect=self._n2vc_FormatApplicationName())
            self.my_ns.n2vc.DeployCharms = asynctest.CoroutineMock(
                side_effect=self._n2vc_DeployCharms)
            self.my_ns.n2vc.create_execution_environment = asynctest.CoroutineMock(
                side_effect=self._n2vc_CreateExecutionEnvironment)
            self.my_ns.n2vc.install_configuration_sw = asynctest.CoroutineMock(
                return_value=pub_key)
            self.my_ns.n2vc.get_ee_ssh_public__key = asynctest.CoroutineMock(
                return_value=pub_key)
            self.my_ns.n2vc.exec_primitive = asynctest.CoroutineMock(
                side_effect=self._return_uuid)
            self.my_ns.n2vc.GetPrimitiveStatus = asynctest.CoroutineMock(
                return_value="completed")
            self.my_ns.n2vc.GetPrimitiveOutput = asynctest.CoroutineMock(
                return_value={
                    "result": "ok",
                    "pubkey": pub_key
                })
            self.my_ns.n2vc.delete_execution_environment = asynctest.CoroutineMock(
                return_value=None)
            self.my_ns.n2vc.get_public_key = asynctest.CoroutineMock(
                return_value=getenv("OSMLCM_VCA_PUBKEY", "public_key"))
            self.my_ns.n2vc.delete_namespace = asynctest.CoroutineMock(
                return_value=None)

        # Mock RO
        if not getenv("OSMLCMTEST_RO_NOMOCK"):
            # self.my_ns.RO = asynctest.Mock(ROclient.ROClient(self.loop, **lcm_config["ro_config"]))
            # TODO first time should be empty list, following should return a dict
            self.my_ns.RO.get_list = asynctest.CoroutineMock(
                self.my_ns.RO.get_list, return_value=[])
            self.my_ns.RO.create = asynctest.CoroutineMock(
                self.my_ns.RO.create, side_effect=self._ro_create())
            self.my_ns.RO.show = asynctest.CoroutineMock(
                self.my_ns.RO.show, side_effect=self._ro_show())
            self.my_ns.RO.create_action = asynctest.CoroutineMock(
                self.my_ns.RO.create_action,
                return_value={
                    "vm-id": {
                        "vim_result": 200,
                        "description": "done"
                    }
                })
            self.my_ns.RO.delete = asynctest.CoroutineMock(
                self.my_ns.RO.delete, return_value={"action_id": "del"})
            # self.my_ns.wait_vm_up_insert_key_ro = asynctest.CoroutineMock(return_value="ip-address")

    @asynctest.fail_on(active_handles=True
                       )  # all async tasks must be completed
    async def test_instantiate(self):
        self.db.set_one = asynctest.Mock()
        nsr_id = descriptors.test_ids["TEST-A"]["ns"]
        nslcmop_id = descriptors.test_ids["TEST-A"]["instantiate"]
        # print("Test instantiate started")

        # delete deployed information of database
        if not getenv("OSMLCMTEST_DB_NOMOCK"):
            if self.db.get_list("nsrs")[0]["_admin"].get("deployed"):
                del self.db.get_list("nsrs")[0]["_admin"]["deployed"]
            for db_vnfr in self.db.get_list("vnfrs"):
                db_vnfr.pop("ip_address", None)
                for db_vdur in db_vnfr["vdur"]:
                    db_vdur.pop("ip_address", None)
                    db_vdur.pop("mac_address", None)
            if getenv("OSMLCMTEST_RO_VIMID"):
                self.db.get_list("vim_accounts")[0]["_admin"]["deployed"][
                    "RO"] = getenv("OSMLCMTEST_RO_VIMID")
            if getenv("OSMLCMTEST_RO_VIMID"):
                self.db.get_list("nsrs")[0]["_admin"]["deployed"][
                    "RO"] = getenv("OSMLCMTEST_RO_VIMID")

        await self.my_ns.instantiate(nsr_id, nslcmop_id)

        # print("instantiate_result: {}".format(self.db.get_one("nslcmops",
        # {"_id": nslcmop_id}).get("detailed-status")))

        self.msg.aiowrite.assert_called_once_with(
            "ns",
            "instantiated", {
                "nsr_id": nsr_id,
                "nslcmop_id": nslcmop_id,
                "operationState": "COMPLETED"
            },
            loop=self.loop)
        self.lcm_tasks.lock_HA.assert_called_once_with('ns', 'nslcmops',
                                                       nslcmop_id)
        if not getenv("OSMLCMTEST_LOGGING_NOMOCK"):
            self.assertTrue(self.my_ns.logger.debug.called,
                            "Debug method not called")
            self.my_ns.logger.error.assert_not_called()
            self.my_ns.logger.exception().assert_not_called()

        if not getenv("OSMLCMTEST_DB_NOMOCK"):
            self.assertTrue(self.db.set_one.called, "db.set_one not called")
            db_nsr = self.db.get_one("nsrs", {"_id": nsr_id})
            db_vnfrs_list = self.db.get_list("vnfrs", {"nsr-id-ref": nsr_id})
            self.assertEqual(db_nsr["_admin"].get("nsState"), "INSTANTIATED",
                             "Not instantiated")
            for vnfr in db_vnfrs_list:
                self.assertEqual(vnfr["_admin"].get("nsState"), "INSTANTIATED",
                                 "Not instantiated")

        if not getenv("OSMLCMTEST_VCA_NOMOCK"):
            # check intial-primitives called
            self.assertTrue(
                self.my_ns.n2vc.exec_primitive.called,
                "Exec primitive not called for initial config primitive")
            for _call in self.my_ns.n2vc.exec_primitive.call_args_list:
                self.assertIn(
                    _call[1]["primitive_name"], ("config", "touch"),
                    "called exec primitive with a primitive different than config or touch"
                )

        # TODO add more checks of called methods
        # TODO add a terminate

    async def test_instantiate_ee_list(self):
        # Using modern IM where configuration is in the new format of execution_environment_list
        ee_descriptor_id = "charm_simple"
        non_used_initial_primitive = {
            "name": "not_to_be_called",
            "seq": 3,
            "execution-environment-ref": "not_used_ee"
        }
        ee_list = [
            {
                "id": ee_descriptor_id,
                "juju": {
                    "charm": "simple"
                },
            },
        ]

        self.db.set_one(
            "vnfds",
            q_filter={"_id": "7637bcf8-cf14-42dc-ad70-c66fcf1e6e77"},
            update_dict={
                "vnf-configuration.execution-environment-list":
                ee_list,
                "vnf-configuration.initial-config-primitive.0.execution-environment-ref":
                ee_descriptor_id,
                "vnf-configuration.initial-config-primitive.1.execution-environment-ref":
                ee_descriptor_id,
                "vnf-configuration.initial-config-primitive.2":
                non_used_initial_primitive,
                "vnf-configuration.config-primitive.0.execution-environment-ref":
                ee_descriptor_id,
                "vnf-configuration.config-primitive.0.execution-environment-primitive":
                "touch_charm",
            },
            unset={"vnf-configuration.juju": None})
        await self.test_instantiate()
        # this will check that the initial-congig-primitive 'not_to_be_called' is not called

    def test_ns_params_2_RO(self):
        vims = self.db.get_list("vim_accounts")
        vim_id = vims[0]["_id"]
        ro_vim_id = vims[0]["_admin"]["deployed"]["RO"]
        vim_id2 = vims[1]["_id"]
        ro_vim_id2 = vims[1]["_admin"]["deployed"]["RO"]

        ns_params = {"vimAccountId": vim_id}
        mgmt_interface = {"cp": "cp"}
        vdu = [{
            "id": "vdu_id",
            "interface": [{
                "external-connection-point-ref": "cp"
            }]
        }]
        vnfd_dict = {
            "1": {
                "vdu": vdu,
                "mgmt-interface": mgmt_interface
            },
            "2": {
                "vdu": vdu,
                "mgmt-interface": mgmt_interface,
                "vnf-configuration": None
            },
            "3": {
                "vdu": vdu,
                "mgmt-interface": mgmt_interface,
                "vnf-configuration": {
                    "config-access": None
                }
            },
            "4": {
                "vdu": vdu,
                "mgmt-interface": mgmt_interface,
                "vnf-configuration": {
                    "config-access": {
                        "ssh-access": None
                    }
                }
            },
            "5": {
                "vdu": vdu,
                "mgmt-interface": mgmt_interface,
                "vnf-configuration": {
                    "config-access": {
                        "ssh-access": {
                            "required": True,
                            "default_user": "******"
                        }
                    }
                }
            },
        }
        nsd = {"constituent-vnfd": []}
        db_vnfrs = {}
        for k in vnfd_dict.keys():
            nsd["constituent-vnfd"].append({
                "vnfd-id-ref": k,
                "member-vnf-index": "member " + k
            })
            db_vnfrs["member " + k] = {
                "vim-account-id": vim_id2 if k == "1" else vim_id
            }

        n2vc_key_list = ["key"]
        ro_ns_params = self.my_ns._ns_params_2_RO(ns_params, nsd, vnfd_dict,
                                                  db_vnfrs, n2vc_key_list)
        ro_params_expected = {
            'wim_account': None,
            "datacenter": ro_vim_id,
            "vnfs": {
                "member 5": {
                    "vdus": {
                        "vdu_id": {
                            "mgmt_keys": n2vc_key_list
                        }
                    }
                },
                "member 1": {
                    "datacenter": ro_vim_id2
                }
            }
        }
        self.assertEqual(ro_ns_params, ro_params_expected)

    # Test scale() and related methods
    @asynctest.fail_on(active_handles=True
                       )  # all async tasks must be completed
    async def test_scale(self):
        # print("Test scale started")

        # TODO: Add more higher-lever tests here, for example:
        # scale-out/scale-in operations with success/error result

        # Test scale() with missing 'scaleVnfData', should return operationState = 'FAILED'
        nsr_id = descriptors.test_ids["TEST-A"]["ns"]
        nslcmop_id = descriptors.test_ids["TEST-A"]["instantiate"]
        await self.my_ns.scale(nsr_id, nslcmop_id)
        expected_value = 'FAILED'
        return_value = self.db.get_one("nslcmops", {
            "_id": nslcmop_id
        }).get("operationState")
        self.assertEqual(return_value, expected_value)
        # print("scale_result: {}".format(self.db.get_one("nslcmops", {"_id": nslcmop_id}).get("detailed-status")))

    # Test _retry_or_skip_suboperation()
    # Expected result:
    # - if a suboperation's 'operationState' is marked as 'COMPLETED', SUBOPERATION_STATUS_SKIP is expected
    # - if marked as anything but 'COMPLETED', the suboperation index is expected
    def test_scale_retry_or_skip_suboperation(self):
        # Load an alternative 'nslcmops' YAML for this test
        nslcmop_id = descriptors.test_ids["TEST-A"]["instantiate"]
        db_nslcmop = self.db.get_one('nslcmops', {"_id": nslcmop_id})
        op_index = 2
        # Test when 'operationState' is 'COMPLETED'
        db_nslcmop['_admin']['operations'][op_index][
            'operationState'] = 'COMPLETED'
        return_value = self.my_ns._retry_or_skip_suboperation(
            db_nslcmop, op_index)
        expected_value = self.my_ns.SUBOPERATION_STATUS_SKIP
        self.assertEqual(return_value, expected_value)
        # Test when 'operationState' is not 'COMPLETED'
        db_nslcmop['_admin']['operations'][op_index]['operationState'] = None
        return_value = self.my_ns._retry_or_skip_suboperation(
            db_nslcmop, op_index)
        expected_value = op_index
        self.assertEqual(return_value, expected_value)

    # Test _find_suboperation()
    # Expected result: index of the found sub-operation, or SUBOPERATION_STATUS_NOT_FOUND if not found
    def test_scale_find_suboperation(self):
        # Load an alternative 'nslcmops' YAML for this test
        nslcmop_id = descriptors.test_ids["TEST-A"]["instantiate"]
        db_nslcmop = self.db.get_one('nslcmops', {"_id": nslcmop_id})
        # Find this sub-operation
        op_index = 2
        vnf_index = db_nslcmop['_admin']['operations'][op_index][
            'member_vnf_index']
        primitive = db_nslcmop['_admin']['operations'][op_index]['primitive']
        primitive_params = db_nslcmop['_admin']['operations'][op_index][
            'primitive_params']
        match = {
            'member_vnf_index': vnf_index,
            'primitive': primitive,
            'primitive_params': primitive_params,
        }
        found_op_index = self.my_ns._find_suboperation(db_nslcmop, match)
        self.assertEqual(found_op_index, op_index)
        # Test with not-matching params
        match = {
            'member_vnf_index': vnf_index,
            'primitive': '',
            'primitive_params': primitive_params,
        }
        found_op_index = self.my_ns._find_suboperation(db_nslcmop, match)
        self.assertEqual(found_op_index,
                         self.my_ns.SUBOPERATION_STATUS_NOT_FOUND)
        # Test with None
        match = None
        found_op_index = self.my_ns._find_suboperation(db_nslcmop, match)
        self.assertEqual(found_op_index,
                         self.my_ns.SUBOPERATION_STATUS_NOT_FOUND)

    # Test _update_suboperation_status()
    def test_scale_update_suboperation_status(self):
        self.db.set_one = asynctest.Mock()
        nslcmop_id = descriptors.test_ids["TEST-A"]["instantiate"]
        db_nslcmop = self.db.get_one('nslcmops', {"_id": nslcmop_id})
        op_index = 0
        # Force the initial values to be distinct from the updated ones
        q_filter = {"_id": db_nslcmop["_id"]}
        # Test to change 'operationState' and 'detailed-status'
        operationState = 'COMPLETED'
        detailed_status = 'Done'
        expected_update_dict = {
            '_admin.operations.0.operationState': operationState,
            '_admin.operations.0.detailed-status': detailed_status,
        }
        self.my_ns._update_suboperation_status(db_nslcmop, op_index,
                                               operationState, detailed_status)
        self.db.set_one.assert_called_once_with(
            "nslcmops",
            q_filter=q_filter,
            update_dict=expected_update_dict,
            fail_on_empty=False)

    def test_scale_add_suboperation(self):
        nslcmop_id = descriptors.test_ids["TEST-A"]["instantiate"]
        db_nslcmop = self.db.get_one('nslcmops', {"_id": nslcmop_id})
        vnf_index = '1'
        num_ops_before = len(
            db_nslcmop.get('_admin', {}).get('operations', [])) - 1
        vdu_id = None
        vdu_count_index = None
        vdu_name = None
        primitive = 'touch'
        mapped_primitive_params = {
            'parameter': [{
                'data-type': 'STRING',
                'name': 'filename',
                'default-value': '<touch_filename2>'
            }],
            'name':
            'touch'
        }
        operationState = 'PROCESSING'
        detailed_status = 'In progress'
        operationType = 'PRE-SCALE'
        # Add a 'pre-scale' suboperation
        op_index_after = self.my_ns._add_suboperation(
            db_nslcmop, vnf_index, vdu_id, vdu_count_index, vdu_name,
            primitive, mapped_primitive_params, operationState,
            detailed_status, operationType)
        self.assertEqual(op_index_after, num_ops_before + 1)

        # Delete all suboperations and add the same operation again
        del db_nslcmop['_admin']['operations']
        op_index_zero = self.my_ns._add_suboperation(
            db_nslcmop, vnf_index, vdu_id, vdu_count_index, vdu_name,
            primitive, mapped_primitive_params, operationState,
            detailed_status, operationType)
        self.assertEqual(op_index_zero, 0)

        # Add a 'RO' suboperation
        RO_nsr_id = '1234567890'
        RO_scaling_info = [{
            'type': 'create',
            'count': 1,
            'member-vnf-index': '1',
            'osm_vdu_id': 'dataVM'
        }]
        op_index = self.my_ns._add_suboperation(
            db_nslcmop, vnf_index, vdu_id, vdu_count_index, vdu_name,
            primitive, mapped_primitive_params, operationState,
            detailed_status, operationType, RO_nsr_id, RO_scaling_info)
        db_RO_nsr_id = db_nslcmop['_admin']['operations'][op_index][
            'RO_nsr_id']
        self.assertEqual(op_index, 1)
        self.assertEqual(RO_nsr_id, db_RO_nsr_id)

        # Try to add an invalid suboperation, should return SUBOPERATION_STATUS_NOT_FOUND
        op_index_invalid = self.my_ns._add_suboperation(
            None, None, None, None, None, None, None, None, None, None, None)
        self.assertEqual(op_index_invalid,
                         self.my_ns.SUBOPERATION_STATUS_NOT_FOUND)

    # Test _check_or_add_scale_suboperation() and _check_or_add_scale_suboperation_RO()
    # check the possible return values:
    # - SUBOPERATION_STATUS_NEW: This is a new sub-operation
    # - op_index (non-negative number): This is an existing sub-operation, operationState != 'COMPLETED'
    # - SUBOPERATION_STATUS_SKIP: This is an existing sub-operation, operationState == 'COMPLETED'
    def test_scale_check_or_add_scale_suboperation(self):
        nslcmop_id = descriptors.test_ids["TEST-A"]["instantiate"]
        db_nslcmop = self.db.get_one('nslcmops', {"_id": nslcmop_id})
        operationType = 'PRE-SCALE'
        vnf_index = '1'
        primitive = 'touch'
        primitive_params = {
            'parameter': [{
                'data-type': 'STRING',
                'name': 'filename',
                'default-value': '<touch_filename2>'
            }],
            'name':
            'touch'
        }

        # Delete all sub-operations to be sure this is a new sub-operation
        del db_nslcmop['_admin']['operations']

        # Add a new sub-operation
        # For new sub-operations, operationState is set to 'PROCESSING' by default
        op_index_new = self.my_ns._check_or_add_scale_suboperation(
            db_nslcmop, vnf_index, primitive, primitive_params, operationType)
        self.assertEqual(op_index_new, self.my_ns.SUBOPERATION_STATUS_NEW)

        # Use the same parameters again to match the already added sub-operation
        # which has status 'PROCESSING' (!= 'COMPLETED') by default
        # The expected return value is a non-negative number
        op_index_existing = self.my_ns._check_or_add_scale_suboperation(
            db_nslcmop, vnf_index, primitive, primitive_params, operationType)
        self.assertTrue(op_index_existing >= 0)

        # Change operationState 'manually' for this sub-operation
        db_nslcmop['_admin']['operations'][op_index_existing][
            'operationState'] = 'COMPLETED'
        # Then use the same parameters again to match the already added sub-operation,
        # which now has status 'COMPLETED'
        # The expected return value is SUBOPERATION_STATUS_SKIP
        op_index_skip = self.my_ns._check_or_add_scale_suboperation(
            db_nslcmop, vnf_index, primitive, primitive_params, operationType)
        self.assertEqual(op_index_skip, self.my_ns.SUBOPERATION_STATUS_SKIP)

        # RO sub-operation test:
        # Repeat tests for the very similar _check_or_add_scale_suboperation_RO(),
        RO_nsr_id = '1234567890'
        RO_scaling_info = [{
            'type': 'create',
            'count': 1,
            'member-vnf-index': '1',
            'osm_vdu_id': 'dataVM'
        }]
        op_index_new_RO = self.my_ns._check_or_add_scale_suboperation(
            db_nslcmop, vnf_index, None, None, 'SCALE-RO', RO_nsr_id,
            RO_scaling_info)
        self.assertEqual(op_index_new_RO, self.my_ns.SUBOPERATION_STATUS_NEW)

        # Use the same parameters again to match the already added RO sub-operation
        op_index_existing_RO = self.my_ns._check_or_add_scale_suboperation(
            db_nslcmop, vnf_index, None, None, 'SCALE-RO', RO_nsr_id,
            RO_scaling_info)
        self.assertTrue(op_index_existing_RO >= 0)

        # Change operationState 'manually' for this RO sub-operation
        db_nslcmop['_admin']['operations'][op_index_existing_RO][
            'operationState'] = 'COMPLETED'
        # Then use the same parameters again to match the already added sub-operation,
        # which now has status 'COMPLETED'
        # The expected return value is SUBOPERATION_STATUS_SKIP
        op_index_skip_RO = self.my_ns._check_or_add_scale_suboperation(
            db_nslcmop, vnf_index, None, None, 'SCALE-RO', RO_nsr_id,
            RO_scaling_info)
        self.assertEqual(op_index_skip_RO, self.my_ns.SUBOPERATION_STATUS_SKIP)

    async def test_deploy_kdus(self):
        nsr_id = descriptors.test_ids["TEST-KDU"]["ns"]
        nslcmop_id = descriptors.test_ids["TEST-KDU"]["instantiate"]
        db_nsr = self.db.get_one("nsrs", {"_id": nsr_id})
        db_vnfr = self.db.get_one("vnfrs", {
            "nsr-id-ref": nsr_id,
            "member-vnf-index-ref": "multikdu"
        })
        db_vnfrs = {"multikdu": db_vnfr}
        db_vnfd = self.db.get_one("vnfds", {"_id": db_vnfr["vnfd-id"]})
        db_vnfds = {db_vnfd["_id"]: db_vnfd}
        task_register = {}
        logging_text = "KDU"
        self.my_ns.k8sclusterhelm.install = asynctest.CoroutineMock(
            return_value="k8s_id")
        self.my_ns.k8sclusterhelm.synchronize_repos = asynctest.CoroutineMock(
            return_value=("", ""))
        self.my_ns.k8sclusterhelm.get_services = asynctest.CoroutineMock(
            return_value=([]))
        await self.my_ns.deploy_kdus(logging_text, nsr_id, nslcmop_id,
                                     db_vnfrs, db_vnfds, task_register)
        await asyncio.wait(list(task_register.keys()), timeout=100)
        db_nsr = self.db.get_list("nsrs")[1]
        self.assertIn("K8s", db_nsr["_admin"]["deployed"],
                      "K8s entry not created at '_admin.deployed'")
        self.assertIsInstance(db_nsr["_admin"]["deployed"]["K8s"], list,
                              "K8s entry is not of type list")
        self.assertEqual(len(db_nsr["_admin"]["deployed"]["K8s"]), 2,
                         "K8s entry is not of type list")
        k8s_instace_info = {
            "kdu-instance": "k8s_id",
            "k8scluster-uuid": "73d96432-d692-40d2-8440-e0c73aee209c",
            "k8scluster-type": "helm-chart",
            "kdu-name": "ldap",
            "kdu-model": "stable/openldap:1.2.1",
            "member-vnf-index": "multikdu",
            "namespace": None
        }

        self.assertEqual(db_nsr["_admin"]["deployed"]["K8s"][0],
                         k8s_instace_info)
        k8s_instace_info["kdu-name"] = "mongo"
        k8s_instace_info["kdu-model"] = "stable/mongodb"
        self.assertEqual(db_nsr["_admin"]["deployed"]["K8s"][1],
                         k8s_instace_info)

    async def test_instantiate_pdu(self):
        nsr_id = descriptors.test_ids["TEST-A"]["ns"]
        nslcmop_id = descriptors.test_ids["TEST-A"]["instantiate"]
        # Modify vnfd/vnfr to change KDU for PDU. Adding keys that NBI will already set
        self.db.set_one("vnfrs", {
            "nsr-id-ref": nsr_id,
            "member-vnf-index-ref": "1"
        },
                        update_dict={
                            "ip-address": "10.205.1.46",
                            "vdur.0.pdu-id":
                            "53e1ec21-2464-451e-a8dc-6e311d45b2c8",
                            "vdur.0.pdu-type": "PDU-TYPE-1",
                            "vdur.0.ip-address": "10.205.1.46",
                        },
                        unset={"vdur.status": None})
        self.db.set_one("vnfrs", {
            "nsr-id-ref": nsr_id,
            "member-vnf-index-ref": "2"
        },
                        update_dict={
                            "ip-address": "10.205.1.47",
                            "vdur.0.pdu-id":
                            "53e1ec21-2464-451e-a8dc-6e311d45b2c8",
                            "vdur.0.pdu-type": "PDU-TYPE-1",
                            "vdur.0.ip-address": "10.205.1.47",
                        },
                        unset={"vdur.status": None})

        await self.my_ns.instantiate(nsr_id, nslcmop_id)
        db_nsr = self.db.get_one("nsrs", {"_id": nsr_id})
        self.assertEqual(db_nsr.get("nsState"), "READY",
                         str(db_nsr.get("errorDescription ")))
        self.assertEqual(db_nsr.get("currentOperation"), "IDLE",
                         "currentOperation different than 'IDLE'")
        self.assertEqual(db_nsr.get("currentOperationID"), None,
                         "currentOperationID different than None")
        self.assertEqual(db_nsr.get("errorDescription "), None,
                         "errorDescription different than None")
        self.assertEqual(db_nsr.get("errorDetail"), None,
                         "errorDetail different than None")

    @asynctest.fail_on(active_handles=True
                       )  # all async tasks must be completed
    async def test_terminate_without_configuration(self):
        nsr_id = descriptors.test_ids["TEST-A"]["ns"]
        nslcmop_id = descriptors.test_ids["TEST-A"]["terminate"]
        # set instantiation task as completed
        self.db.set_list("nslcmops", {
            "nsInstanceId": nsr_id,
            "_id.ne": nslcmop_id
        },
                         update_dict={"operationState": "COMPLETED"})
        self.my_ns.RO.show = asynctest.CoroutineMock(
            ROClient.show, side_effect=self._ro_show(delete=nslcmop_id))
        self.db.set_one("nsrs", {"_id": nsr_id},
                        update_dict={
                            "_admin.deployed.VCA.0": None,
                            "_admin.deployed.VCA.1": None
                        })

        await self.my_ns.terminate(nsr_id, nslcmop_id)
        db_nslcmop = self.db.get_one("nslcmops", {"_id": nslcmop_id})
        self.assertEqual(db_nslcmop.get("operationState"), 'COMPLETED',
                         db_nslcmop.get("detailed-status"))
        db_nsr = self.db.get_one("nsrs", {"_id": nsr_id})
        self.assertEqual(db_nsr.get("nsState"), "NOT_INSTANTIATED",
                         str(db_nsr.get("errorDescription ")))
        self.assertEqual(db_nsr["_admin"].get("nsState"), "NOT_INSTANTIATED",
                         str(db_nsr.get("errorDescription ")))
        self.assertEqual(db_nsr.get("currentOperation"), "IDLE",
                         "currentOperation different than 'IDLE'")
        self.assertEqual(db_nsr.get("currentOperationID"), None,
                         "currentOperationID different than None")
        self.assertEqual(db_nsr.get("errorDescription "), None,
                         "errorDescription different than None")
        self.assertEqual(db_nsr.get("errorDetail"), None,
                         "errorDetail different than None")
        db_vnfrs_list = self.db.get_list("vnfrs", {"nsr-id-ref": nsr_id})
        for vnfr in db_vnfrs_list:
            self.assertEqual(vnfr["_admin"].get("nsState"), "NOT_INSTANTIATED",
                             "Not instantiated")

    @asynctest.fail_on(active_handles=True
                       )  # all async tasks must be completed
    async def test_terminate_primitive(self):
        nsr_id = descriptors.test_ids["TEST-A"]["ns"]
        nslcmop_id = descriptors.test_ids["TEST-A"]["terminate"]
        self.my_ns.RO.show = asynctest.CoroutineMock(
            ROClient.show, side_effect=self._ro_show(delete=nslcmop_id))
        # set instantiation task as completed
        self.db.set_list("nslcmops", {
            "nsInstanceId": nsr_id,
            "_id.ne": nslcmop_id
        },
                         update_dict={"operationState": "COMPLETED"})

        # modify vnfd descriptor to include terminate_primitive
        terminate_primitive = [{
            "name":
            "touch",
            "parameter": [{
                "name": "filename",
                "value": "terminate_filename"
            }],
            "seq":
            '1'
        }]
        db_vnfr = self.db.get_one("vnfrs", {
            "nsr-id-ref": nsr_id,
            "member-vnf-index-ref": "1"
        })
        self.db.set_one("vnfds", {"_id": db_vnfr["vnfd-id"]}, {
            "vnf-configuration.terminate-config-primitive":
            terminate_primitive
        })

        await self.my_ns.terminate(nsr_id, nslcmop_id)
        db_nslcmop = self.db.get_one("nslcmops", {"_id": nslcmop_id})
        self.assertEqual(db_nslcmop.get("operationState"), 'COMPLETED',
                         db_nslcmop.get("detailed-status"))
        db_nsr = self.db.get_one("nsrs", {"_id": nsr_id})
        self.assertEqual(db_nsr.get("nsState"), "NOT_INSTANTIATED",
                         str(db_nsr.get("errorDescription ")))
        self.assertEqual(db_nsr["_admin"].get("nsState"), "NOT_INSTANTIATED",
                         str(db_nsr.get("errorDescription ")))
        self.assertEqual(db_nsr.get("currentOperation"), "IDLE",
                         "currentOperation different than 'IDLE'")
        self.assertEqual(db_nsr.get("currentOperationID"), None,
                         "currentOperationID different than None")
        self.assertEqual(db_nsr.get("errorDescription "), None,
                         "errorDescription different than None")
        self.assertEqual(db_nsr.get("errorDetail"), None,
                         "errorDetail different than None")
예제 #9
0
파일: test_ns.py 프로젝트: OSMadmin/LCM
    async def setUp(self):

        # Mock DB
        if not getenv("OSMLCMTEST_DB_NOMOCK"):
            self.db = DbMemory()
            self.db.create_list(
                "vnfds",
                yaml.load(descriptors.db_vnfds_text, Loader=yaml.Loader))
            self.db.create_list(
                "nsds", yaml.load(descriptors.db_nsds_text,
                                  Loader=yaml.Loader))
            self.db.create_list(
                "nsrs", yaml.load(descriptors.db_nsrs_text,
                                  Loader=yaml.Loader))
            self.db.create_list(
                "vim_accounts",
                yaml.load(descriptors.db_vim_accounts_text,
                          Loader=yaml.Loader))
            self.db.create_list(
                "k8sclusters",
                yaml.load(descriptors.db_k8sclusters_text, Loader=yaml.Loader))
            self.db.create_list(
                "nslcmops",
                yaml.load(descriptors.db_nslcmops_text, Loader=yaml.Loader))
            self.db.create_list(
                "vnfrs",
                yaml.load(descriptors.db_vnfrs_text, Loader=yaml.Loader))
            self.db_vim_accounts = yaml.load(descriptors.db_vim_accounts_text,
                                             Loader=yaml.Loader)

        # Mock kafka
        self.msg = asynctest.Mock(MsgKafka())

        # Mock filesystem
        if not getenv("OSMLCMTEST_FS_NOMOCK"):
            self.fs = asynctest.Mock(FsLocal())
            self.fs.get_params.return_value = {
                "path": getenv("OSMLCMTEST_PACKAGES_PATH",
                               "./test/temp/packages")
            }
            self.fs.file_open = asynctest.mock_open()
            # self.fs.file_open.return_value.__enter__.return_value = asynctest.MagicMock()  # called on a python "with"
            # self.fs.file_open.return_value.__enter__.return_value.read.return_value = ""   # empty file

        # Mock TaskRegistry
        self.lcm_tasks = asynctest.Mock(TaskRegistry())
        self.lcm_tasks.lock_HA.return_value = True
        self.lcm_tasks.waitfor_related_HA.return_value = None
        self.lcm_tasks.lookfor_related.return_value = ("", [])

        # Mock VCA - K8s
        if not getenv("OSMLCMTEST_VCA_K8s_NOMOCK"):
            ns.K8sJujuConnector = asynctest.MagicMock(ns.K8sJujuConnector)
            ns.K8sHelmConnector = asynctest.MagicMock(ns.K8sHelmConnector)

        if not getenv("OSMLCMTEST_VCA_NOMOCK"):
            ns.N2VCJujuConnector = asynctest.MagicMock(ns.N2VCJujuConnector)
            ns.N2VCJujuConnectorLCM = asynctest.MagicMock(
                ns.N2VCJujuConnectorLCM)
            ns.LCMHelmConn = asynctest.MagicMock(ns.LCMHelmConn)

        # Create NsLCM class
        self.my_ns = ns.NsLcm(self.db, self.msg, self.fs, self.lcm_tasks,
                              lcm_config, self.loop)
        self.my_ns._wait_dependent_n2vc = asynctest.CoroutineMock()

        # Mock logging
        if not getenv("OSMLCMTEST_LOGGING_NOMOCK"):
            self.my_ns.logger = asynctest.Mock(self.my_ns.logger)

        # Mock VCA - N2VC
        if not getenv("OSMLCMTEST_VCA_NOMOCK"):
            pub_key = getenv("OSMLCMTEST_NS_PUBKEY",
                             "ssh-rsa test-pub-key [email protected]")
            # self.my_ns.n2vc = asynctest.Mock(N2VC())
            self.my_ns.n2vc.GetPublicKey.return_value = getenv(
                "OSMLCM_VCA_PUBKEY", "public_key")
            # allow several versions of n2vc
            self.my_ns.n2vc.FormatApplicationName = asynctest.Mock(
                side_effect=self._n2vc_FormatApplicationName())
            self.my_ns.n2vc.DeployCharms = asynctest.CoroutineMock(
                side_effect=self._n2vc_DeployCharms)
            self.my_ns.n2vc.create_execution_environment = asynctest.CoroutineMock(
                side_effect=self._n2vc_CreateExecutionEnvironment)
            self.my_ns.n2vc.install_configuration_sw = asynctest.CoroutineMock(
                return_value=pub_key)
            self.my_ns.n2vc.get_ee_ssh_public__key = asynctest.CoroutineMock(
                return_value=pub_key)
            self.my_ns.n2vc.exec_primitive = asynctest.CoroutineMock(
                side_effect=self._return_uuid)
            self.my_ns.n2vc.GetPrimitiveStatus = asynctest.CoroutineMock(
                return_value="completed")
            self.my_ns.n2vc.GetPrimitiveOutput = asynctest.CoroutineMock(
                return_value={
                    "result": "ok",
                    "pubkey": pub_key
                })
            self.my_ns.n2vc.delete_execution_environment = asynctest.CoroutineMock(
                return_value=None)
            self.my_ns.n2vc.get_public_key = asynctest.CoroutineMock(
                return_value=getenv("OSMLCM_VCA_PUBKEY", "public_key"))
            self.my_ns.n2vc.delete_namespace = asynctest.CoroutineMock(
                return_value=None)

        # Mock RO
        if not getenv("OSMLCMTEST_RO_NOMOCK"):
            # self.my_ns.RO = asynctest.Mock(ROclient.ROClient(self.loop, **lcm_config["ro_config"]))
            # TODO first time should be empty list, following should return a dict
            self.my_ns.RO.get_list = asynctest.CoroutineMock(
                self.my_ns.RO.get_list, return_value=[])
            self.my_ns.RO.create = asynctest.CoroutineMock(
                self.my_ns.RO.create, side_effect=self._ro_create())
            self.my_ns.RO.show = asynctest.CoroutineMock(
                self.my_ns.RO.show, side_effect=self._ro_show())
            self.my_ns.RO.create_action = asynctest.CoroutineMock(
                self.my_ns.RO.create_action,
                return_value={
                    "vm-id": {
                        "vim_result": 200,
                        "description": "done"
                    }
                })
            self.my_ns.RO.delete = asynctest.CoroutineMock(
                self.my_ns.RO.delete, return_value={"action_id": "del"})