From 8b3fc62050a344fe9a9c8909e4c672cb9aa3281d Mon Sep 17 00:00:00 2001 From: efiacor Date: Fri, 24 Jan 2020 13:19:01 +0000 Subject: Adding DB Init and setup Signed-off-by: efiacor Change-Id: Ie32efcf007c9b6fa25b0072019f4a91a841d9d0c Issue-ID: DCAEGEN2-1828 --- components/pm-subscription-handler/Dockerfile | 10 +- .../pmsh_service/mod/__init__.py | 48 ++++++++- .../pmsh_service/mod/aai_client.py | 71 ++++++------- .../pmsh_service/mod/config_handler.py | 2 +- .../pmsh_service/mod/db_models.py | 88 ++++++++++++++++ .../pmsh_service/mod/network_function.py | 73 ++++++++++++++ .../pmsh_service/mod/pmsh_logging.py | 2 +- .../pmsh_service/mod/subscription.py | 97 ++++++++++++++++-- .../pmsh_service/pmsh_service.py | 23 ++++- components/pm-subscription-handler/setup.py | 8 +- .../tests/config_handler_test.py | 112 --------------------- .../tests/data/cbs_data.json | 109 -------------------- .../tests/data/cbs_data_1.json | 109 ++++++++++++++++++++ .../tests/data/cbs_data_2.json | 68 +++++++++++++ .../tests/expected_config.json | 68 ------------- .../tests/test_aai_service.py | 10 +- .../tests/test_config_handler.py | 95 +++++++++++++++++ .../tests/test_network_function.py | 61 +++++++++++ .../tests/test_pmsh_utils.py | 19 +++- .../tests/test_subscription.py | 104 ++++++++++++++++--- 20 files changed, 805 insertions(+), 372 deletions(-) create mode 100755 components/pm-subscription-handler/pmsh_service/mod/db_models.py create mode 100755 components/pm-subscription-handler/pmsh_service/mod/network_function.py delete mode 100755 components/pm-subscription-handler/tests/config_handler_test.py delete mode 100644 components/pm-subscription-handler/tests/data/cbs_data.json create mode 100644 components/pm-subscription-handler/tests/data/cbs_data_1.json create mode 100755 components/pm-subscription-handler/tests/data/cbs_data_2.json delete mode 100755 components/pm-subscription-handler/tests/expected_config.json create mode 100755 components/pm-subscription-handler/tests/test_config_handler.py create mode 100755 components/pm-subscription-handler/tests/test_network_function.py mode change 100644 => 100755 components/pm-subscription-handler/tests/test_subscription.py (limited to 'components') diff --git a/components/pm-subscription-handler/Dockerfile b/components/pm-subscription-handler/Dockerfile index 4f03a306..bb3f63c6 100644 --- a/components/pm-subscription-handler/Dockerfile +++ b/components/pm-subscription-handler/Dockerfile @@ -1,5 +1,5 @@ # ============LICENSE_START=================================================== -# Copyright (C) 2019 Nordix Foundation. +# Copyright (C) 2020 Nordix Foundation. # ============================================================================ # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -24,15 +24,13 @@ ENV PMSHUSER=pmsh \ # set PATH & PYTHONPATH vars PATH=/usr/local/lib/python3.7/bin:$PATH:$APPDIR/bin \ PYTHONPATH=/usr/local/lib/python3.7/site-packages:./mod:./:$PYTHONPATH:$APPDIR/bin \ - REQUESTS_CA_BUNDLE=/etc/ssl/certs/ca-certificates.crt + REQUESTS_CA_BUNDLE=/etc/ssl/certs/ca-certificates.crt \ + LOGS_PATH="/var/log/ONAP/dcaegen2/services/pmsh" WORKDIR $APPDIR # add non root user & group -RUN addgroup --system $PMSHUSER && adduser --ingroup $PMSHUSER --system $PMSHUSER && \ - # create logs dir and change permissions - mkdir -p /var/log/ONAP/$PMSHUSER/logs && \ - chmod a+w /var/log/ONAP/$PMSHUSER/logs +RUN addgroup --system $PMSHUSER && adduser --ingroup $PMSHUSER --system $PMSHUSER COPY setup.py ./ COPY requirements.txt ./ diff --git a/components/pm-subscription-handler/pmsh_service/mod/__init__.py b/components/pm-subscription-handler/pmsh_service/mod/__init__.py index d2f6f2fd..f8b1c598 100644 --- a/components/pm-subscription-handler/pmsh_service/mod/__init__.py +++ b/components/pm-subscription-handler/pmsh_service/mod/__init__.py @@ -1,5 +1,5 @@ # ============LICENSE_START=================================================== -# Copyright (C) 2019 Nordix Foundation. +# Copyright (C) 2019-2020 Nordix Foundation. # ============================================================================ # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -15,7 +15,49 @@ # # SPDX-License-Identifier: Apache-2.0 # ============LICENSE_END===================================================== +import os +from urllib.parse import quote +from connexion import App +from flask_sqlalchemy import SQLAlchemy -# empty __init__.py so that pytest can add correct path to coverage report, -# -- per pytest best practice guideline +import mod.pmsh_logging as logger + +db = SQLAlchemy() +basedir = os.path.abspath(os.path.dirname(__file__)) + + +def create_prod_app(): + logger.create_loggers(os.getenv('LOGS_PATH')) + connex_app = App(__name__, specification_dir=basedir) + app = connex_app.app + app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False + app.config['SQLALCHEMY_RECORD_QUERIES'] = True + app.config['SQLALCHEMY_DATABASE_URI'] = get_db_connection_url() + db.init_app(app) + return app + + +def create_test_app(): + logger.create_loggers('./unit_test_logs') + connex_app = App(__name__, specification_dir=basedir) + app = connex_app.app + app.config['TESTING'] = True + app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False + app.config['SQLALCHEMY_RECORD_QUERIES'] = True + app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get('TEST_DB_URL', 'sqlite://') + db.init_app(app) + return app + + +def get_db_connection_url(): + pg_host = os.getenv('PMSH_PG_URL') + pg_user = os.getenv('PMSH_PG_USERNAME') + pg_user_pass = os.getenv('PMSH_PG_PASSWORD') + pmsh_db_name = os.getenv('PMSH_DB_NAME', 'pmsh') + pmsh_db_port = os.getenv('PMSH_PG_PORT', '5432') + db_url = f'postgres+psycopg2://{quote(str(pg_user), safe="")}:' \ + f'{quote(str(pg_user_pass), safe="")}@{pg_host}:{pmsh_db_port}/{pmsh_db_name}' + if 'None' in db_url: + raise Exception(f'Invalid DB connection URL: {db_url} .. exiting app!') + return db_url diff --git a/components/pm-subscription-handler/pmsh_service/mod/aai_client.py b/components/pm-subscription-handler/pmsh_service/mod/aai_client.py index 8b51a712..747846f1 100755 --- a/components/pm-subscription-handler/pmsh_service/mod/aai_client.py +++ b/components/pm-subscription-handler/pmsh_service/mod/aai_client.py @@ -16,14 +16,15 @@ # SPDX-License-Identifier: Apache-2.0 # ============LICENSE_END===================================================== import json -import os import uuid +from os import environ import requests from requests.auth import HTTPBasicAuth import mod.pmsh_logging as logger -from mod.subscription import Subscription, XnfFilter +from mod.network_function import NetworkFunction +from mod.subscription import Subscription, NetworkFunctionFilter def get_pmsh_subscription_data(cbs_data): @@ -34,28 +35,29 @@ def get_pmsh_subscription_data(cbs_data): cbs_data: json app config from the Config Binding Service. Returns: - Subscription, set(Xnf): `Subscription` object, set of XNFs to be added. + Subscription, set(NetworkFunctions): `Subscription` object, + set of NetworkFunctions to be added. Raises: RuntimeError: if AAI data cannot be retrieved. """ - aai_xnf_data = _get_all_aai_xnf_data() - if aai_xnf_data: + aai_nf_data = _get_all_aai_nf_data() + if aai_nf_data: sub = Subscription(**cbs_data['policy']['subscription']) - xnfs = _filter_xnf_data(aai_xnf_data, XnfFilter(**sub.nfFilter)) + nfs = _filter_nf_data(aai_nf_data, NetworkFunctionFilter(**sub.nfFilter)) else: raise RuntimeError('Failed to get data from AAI') - return sub, xnfs + return sub, nfs -def _get_all_aai_xnf_data(): +def _get_all_aai_nf_data(): """ - Return queried xnf data from the AAI service. + Return queried nf data from the AAI service. Returns: json: the json response from AAI query, else None. """ - xnf_data = None + nf_data = None try: session = requests.Session() aai_endpoint = f'{_get_aai_service_url()}{"/aai/v16/query"}' @@ -74,10 +76,10 @@ def _get_all_aai_xnf_data(): data=json_data, params=params, verify=False) response.raise_for_status() if response.ok: - xnf_data = json.loads(response.text) + nf_data = json.loads(response.text) except Exception as e: logger.debug(e) - return xnf_data + return nf_data def _get_aai_service_url(): @@ -91,52 +93,37 @@ def _get_aai_service_url(): KeyError: if AAI env vars not found. """ try: - aai_service = os.environ['AAI_SERVICE_HOST'] - aai_ssl_port = os.environ['AAI_SERVICE_PORT_AAI_SSL'] + aai_service = environ['AAI_SERVICE_HOST'] + aai_ssl_port = environ['AAI_SERVICE_PORT_AAI_SSL'] return f'https://{aai_service}:{aai_ssl_port}' except KeyError as e: logger.debug(f'Failed to get AAI env vars: {e}') raise -def _filter_xnf_data(xnf_data, xnf_filter): +def _filter_nf_data(nf_data, nf_filter): """ - Returns a list of filtered xnfs using the xnf_filter . + Returns a list of filtered NetworkFunctions using the nf_filter. Args: - xnf_data: the xnf json data from AAI. - xnf_filter: the `XnfFilter ` to be applied. + nf_data : the nf json data from AAI. + nf_filter: the `NetworkFunctionFilter ` to be applied. Returns: - set: a set of filtered xnfs. + set: a set of filtered NetworkFunctions. Raises: KeyError: if AAI data cannot be parsed. """ - xnf_set = set() + nf_set = set() try: - for xnf in xnf_data['results']: - name_identifier = 'pnf-name' if xnf['node-type'] == 'pnf' else 'vnf-name' - if xnf_filter.is_xnf_in_filter(xnf['properties'].get(name_identifier)): - xnf_set.add(Xnf(xnf_name=xnf['properties'].get('name_identifier'), - orchestration_status=xnf['properties'].get('orchestration-status'))) + for nf in nf_data['results']: + name_identifier = 'pnf-name' if nf['node-type'] == 'pnf' else 'vnf-name' + if nf_filter.is_nf_in_filter(nf['properties'].get(name_identifier)): + nf_set.add(NetworkFunction( + nf_name=nf['properties'].get(name_identifier), + orchestration_status=nf['properties'].get('orchestration-status'))) except KeyError as e: logger.debug(f'Failed to parse AAI data: {e}') raise - return xnf_set - - -class Xnf: - def __init__(self, **kwargs): - """ - Object representation of the XNF. - """ - self.xnf_name = kwargs.get('xnf_name') - self.orchestration_status = kwargs.get('orchestration_status') - - @classmethod - def xnf_def(cls): - return cls(xnf_name=None, orchestration_status=None) - - def __str__(self): - return f'xnf-name: {self.xnf_name}, orchestration-status: {self.orchestration_status}' + return nf_set diff --git a/components/pm-subscription-handler/pmsh_service/mod/config_handler.py b/components/pm-subscription-handler/pmsh_service/mod/config_handler.py index e9edbca4..1ce4b701 100755 --- a/components/pm-subscription-handler/pmsh_service/mod/config_handler.py +++ b/components/pm-subscription-handler/pmsh_service/mod/config_handler.py @@ -22,7 +22,7 @@ from os import environ import requests from tenacity import retry, wait_fixed, stop_after_attempt -from pmsh_service.mod import pmsh_logging as logger +import mod.pmsh_logging as logger class ConfigHandler: diff --git a/components/pm-subscription-handler/pmsh_service/mod/db_models.py b/components/pm-subscription-handler/pmsh_service/mod/db_models.py new file mode 100755 index 00000000..479d40e5 --- /dev/null +++ b/components/pm-subscription-handler/pmsh_service/mod/db_models.py @@ -0,0 +1,88 @@ +# ============LICENSE_START=================================================== +# Copyright (C) 2019-2020 Nordix Foundation. +# ============================================================================ +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# +# SPDX-License-Identifier: Apache-2.0 +# ============LICENSE_END===================================================== + +from sqlalchemy import Column, Integer, String, ForeignKey +from sqlalchemy.orm import relationship + +from mod import db + + +class SubscriptionModel(db.Model): + __tablename__ = 'subscriptions' + id = Column(Integer, primary_key=True, autoincrement=True) + subscription_name = Column(String(100), unique=True) + status = Column(String(20)) + + nfs = relationship( + 'NfSubRelationalModel', + cascade='all, delete-orphan', + backref='subscription') + + def __init__(self, subscription_name, status): + self.subscription_name = subscription_name + self.status = status + + def __repr__(self): + return f'Subscription: {self.subscription_name} {self.status}' + + def __eq__(self, other): + if isinstance(self, other.__class__): + return self.subscription_name == other.subscription_name + return False + + +class NetworkFunctionModel(db.Model): + __tablename__ = 'network_functions' + id = Column(Integer, primary_key=True, autoincrement=True) + nf_name = Column(String(100), unique=True) + orchestration_status = Column(String(100)) + + subscriptions = relationship( + 'NfSubRelationalModel', + cascade='all, delete-orphan', + backref='nf') + + def __init__(self, nf_name, orchestration_status): + self.nf_name = nf_name + self.orchestration_status = orchestration_status + + def __repr__(self): + return f'NetworkFunctionModel: {self.nf_name}, {self.orchestration_status}' + + +class NfSubRelationalModel(db.Model): + __tablename__ = 'nf_to_sub_rel' + id = Column(Integer, primary_key=True, autoincrement=True) + subscription_name = Column( + String, + ForeignKey(SubscriptionModel.subscription_name, ondelete='cascade', onupdate='cascade') + ) + nf_name = Column( + String, + ForeignKey(NetworkFunctionModel.nf_name, ondelete='cascade', onupdate='cascade') + ) + nf_sub_status = Column(String(20)) + + def __init__(self, subscription_name, nf_name, nf_sub_status=None): + self.subscription_name = subscription_name + self.nf_name = nf_name + self.nf_sub_status = nf_sub_status + + def __repr__(self): + return f'NetworkFunctionSubscriptions: {self.subscription_name}, ' \ + f'{self.nf_name}, {self.nf_sub_status}' diff --git a/components/pm-subscription-handler/pmsh_service/mod/network_function.py b/components/pm-subscription-handler/pmsh_service/mod/network_function.py new file mode 100755 index 00000000..64f614af --- /dev/null +++ b/components/pm-subscription-handler/pmsh_service/mod/network_function.py @@ -0,0 +1,73 @@ +# ============LICENSE_START=================================================== +# Copyright (C) 2020 Nordix Foundation. +# ============================================================================ +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# +# SPDX-License-Identifier: Apache-2.0 +# ============LICENSE_END===================================================== + +from mod import pmsh_logging as logger, db +from mod.db_models import NetworkFunctionModel + + +class NetworkFunction: + def __init__(self, **kwargs): + """ + Object representation of the NetworkFunction. + """ + self.nf_name = kwargs.get('nf_name') + self.orchestration_status = kwargs.get('orchestration_status') + + @classmethod + def nf_def(cls): + return cls(nf_name=None, orchestration_status=None) + + def __str__(self): + return f'nf-name: {self.nf_name}, orchestration-status: {self.orchestration_status}' + + def create(self): + """ Creates a NetworkFunction database entry + """ + existing_nf = NetworkFunctionModel.query.filter( + NetworkFunctionModel.nf_name == self.nf_name).one_or_none() + + if existing_nf is None: + new_nf = NetworkFunctionModel(nf_name=self.nf_name, + orchestration_status=self.orchestration_status) + db.session.add(new_nf) + db.session.commit() + + return new_nf + else: + logger.debug(f'Network function {existing_nf} already exists,' + f' returning this network function..') + return existing_nf + + @staticmethod + def get(nf_name): + """ Retrieves a network function + Args: + nf_name (str): The network function name + Returns: + NetworkFunctionModel object else None + """ + return NetworkFunctionModel.query.filter( + NetworkFunctionModel.nf_name == nf_name).one_or_none() + + @staticmethod + def get_all(): + """ Retrieves all network functions + Returns: + list: NetworkFunctionModel objects else empty + """ + return NetworkFunctionModel.query.all() diff --git a/components/pm-subscription-handler/pmsh_service/mod/pmsh_logging.py b/components/pm-subscription-handler/pmsh_service/mod/pmsh_logging.py index f88ea137..f2d11d49 100644 --- a/components/pm-subscription-handler/pmsh_service/mod/pmsh_logging.py +++ b/components/pm-subscription-handler/pmsh_service/mod/pmsh_logging.py @@ -64,7 +64,7 @@ def get_module_logger(mod_name): return logger -def create_loggers(logs_path="/var/log/ONAP/pmsh/logs"): +def create_loggers(logs_path=''): """ Public method to set the global logger, launched from Run This is *not* launched during unit testing, so unit tests do not diff --git a/components/pm-subscription-handler/pmsh_service/mod/subscription.py b/components/pm-subscription-handler/pmsh_service/mod/subscription.py index aa3318ac..265d90b8 100755 --- a/components/pm-subscription-handler/pmsh_service/mod/subscription.py +++ b/components/pm-subscription-handler/pmsh_service/mod/subscription.py @@ -1,5 +1,5 @@ # ============LICENSE_START=================================================== -# Copyright (C) 2019 Nordix Foundation. +# Copyright (C) 2019-2020 Nordix Foundation. # ============================================================================ # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -17,6 +17,10 @@ # ============LICENSE_END===================================================== import re +import mod.pmsh_logging as logger +from mod import db +from mod.db_models import SubscriptionModel, NfSubRelationalModel + class Subscription: def __init__(self, **kwargs): @@ -41,21 +45,100 @@ class Subscription: clean_sub.update({'nfName': xnf_name, 'policyName': f'OP-{self.subscriptionName}'}) return clean_sub + def create(self): + """ Creates a subscription database entry + + Returns: + Subscription object + """ + existing_subscription = (SubscriptionModel.query.filter( + SubscriptionModel.subscription_name == self.subscriptionName).one_or_none()) + + if existing_subscription is None: + new_subscription = SubscriptionModel(subscription_name=self.subscriptionName, + status=self.administrativeState) + + db.session.add(new_subscription) + db.session.commit() + + return new_subscription + + else: + logger.debug(f'Subscription {self.subscriptionName} already exists,' + f' returning this subscription..') + return existing_subscription + + def add_network_functions_to_subscription(self, nf_list): + """ Associates network functions to a Subscription -class XnfFilter: + Args: + nf_list : A list of NetworkFunction objects. + """ + current_sub = self.create() + logger.debug(f'Adding network functions to subscription {current_sub.subscription_name}') + + for nf in nf_list: + current_nf = nf.create() + + existing_entry = NfSubRelationalModel.query.filter( + NfSubRelationalModel.subscription_name == current_sub.subscription_name, + NfSubRelationalModel.nf_name == current_nf.nf_name).one_or_none() + if existing_entry is None: + new_nf_sub = NfSubRelationalModel(current_sub.subscription_name, nf.nf_name) + new_nf_sub.nf = current_nf + logger.debug(current_nf) + current_sub.nfs.append(new_nf_sub) + + db.session.add(current_sub) + db.session.commit() + + @staticmethod + def get(subscription_name): + """ Retrieves a subscription + + Args: + subscription_name (str): The subscription name + + Returns: + Subscription object else None + """ + return SubscriptionModel.query.filter( + SubscriptionModel.subscription_name == subscription_name).one_or_none() + + @staticmethod + def get_all(): + """ Retrieves a list of subscriptions + + Returns: + list: Subscription list else empty + """ + return SubscriptionModel.query.all() + + @staticmethod + def get_all_nfs_subscription_relations(): + """ Retrieves all network function to subscription relations + + Returns: + list: NetworkFunctions per Subscription list else empty + """ + nf_per_subscriptions = NfSubRelationalModel.query.all() + + return nf_per_subscriptions + + +class NetworkFunctionFilter: def __init__(self, **kwargs): self.nf_sw_version = kwargs.get('swVersions') self.nf_names = kwargs.get('nfNames') self.regex_matcher = re.compile('|'.join(raw_regex for raw_regex in self.nf_names)) - def is_xnf_in_filter(self, xnf_name): - """Match the xnf name against regex values in Subscription.nfFilter.nfNames + def is_nf_in_filter(self, nf_name): + """Match the nf name against regex values in Subscription.nfFilter.nfNames Args: - xnf_name: the AAI xnf name. + nf_name: the AAI nf name. Returns: bool: True if matched, else False. """ - - return self.regex_matcher.search(xnf_name) + return self.regex_matcher.search(nf_name) diff --git a/components/pm-subscription-handler/pmsh_service/pmsh_service.py b/components/pm-subscription-handler/pmsh_service/pmsh_service.py index d8a593fb..99689d01 100755 --- a/components/pm-subscription-handler/pmsh_service/pmsh_service.py +++ b/components/pm-subscription-handler/pmsh_service/pmsh_service.py @@ -15,17 +15,34 @@ # # SPDX-License-Identifier: Apache-2.0 # ============LICENSE_END===================================================== +import sys import time +import mod.aai_client as aai_client import mod.pmsh_logging as logger +from mod import db, create_prod_app +from mod.config_handler import ConfigHandler +from mod.subscription import Subscription def main(): - logger.create_loggers() + + try: + app = create_prod_app() + app.app_context().push() + db.create_all(app=app) + + config_handler = ConfigHandler() + cbs_data = config_handler.get_config() + subscription, xnfs = aai_client.get_pmsh_subscription_data(cbs_data) + subscription.add_network_functions_to_subscription(xnfs) + except Exception as e: + logger.debug(f'Failed to Init PMSH: {e}') + sys.exit(e) while True: - time.sleep(30) - logger.debug("He's not the messiah, he's a very naughty boy!") + logger.debug(Subscription.get_all_nfs_subscription_relations()) + time.sleep(5) if __name__ == '__main__': diff --git a/components/pm-subscription-handler/setup.py b/components/pm-subscription-handler/setup.py index a4d9ada4..056a5d82 100644 --- a/components/pm-subscription-handler/setup.py +++ b/components/pm-subscription-handler/setup.py @@ -1,5 +1,5 @@ # ============LICENSE_START======================================================= -# Copyright (C) 2019 Nordix Foundation. +# Copyright (C) 2019-2020 Nordix Foundation. # ================================================================================ # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -30,5 +30,9 @@ setup( python_requires='>=3', install_requires=[ "requests==2.22.0", - "tenacity==6.0.0"], + "tenacity==6.0.0", + "connexion==2.5.0", + "flask_sqlalchemy==2.4.1", + "Flask==1.1.1", + "psycopg2-binary==2.8.4"] ) diff --git a/components/pm-subscription-handler/tests/config_handler_test.py b/components/pm-subscription-handler/tests/config_handler_test.py deleted file mode 100755 index fcc25d60..00000000 --- a/components/pm-subscription-handler/tests/config_handler_test.py +++ /dev/null @@ -1,112 +0,0 @@ -# ============LICENSE_START=================================================== -# Copyright (C) 2019-2020 Nordix Foundation. -# ============================================================================ -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. -# -# SPDX-License-Identifier: Apache-2.0 -# ============LICENSE_END===================================================== - -import json -import unittest -from os import environ -from os import path -from unittest.mock import patch - -import requests -import responses -from tenacity import wait_none - -from pmsh_service.mod.config_handler import ConfigHandler - - -class ConfigHandlerTest(unittest.TestCase): - - def setUp(self): - self.env_vars = {'CONFIG_BINDING_SERVICE_SERVICE_HOST': 'cbs_hostname', - 'CONFIG_BINDING_SERVICE_SERVICE_PORT': '10000', - 'HOSTNAME': 'hostname'} - for key, value in self.env_vars.items(): - environ[key] = value - self.cbs_url = 'http://cbs_hostname:10000/service_component_all/hostname' - self.expected_config = self._get_expected_config() - - def test_missing_environment_variable(self): - for key, value in self.env_vars.items(): - with self.assertRaises(KeyError): - environ.pop(key) - test_value = globals()[value] - test_value() - environ[key] = value - - @responses.activate - def test_get_config_success(self): - responses.add(responses.GET, self.cbs_url, json=json.dumps(self.expected_config), - status=200) - - config_handler = ConfigHandler() - config_handler.get_config.retry.wait = wait_none() - - self.assertEqual(self.expected_config, config_handler.get_config()) - - def test_get_config_already_exists(self): - config_handler = ConfigHandler() - expected_config = self._get_expected_config() - config_handler._config = expected_config - - with patch.object(requests, 'get') as mock_get_request: - actual_config = config_handler.get_config() - - self.assertEqual(0, mock_get_request.call_count) - self.assertEqual(expected_config, actual_config) - - @responses.activate - def test_get_config_error(self): - responses.add(responses.GET, self.cbs_url, status=404) - config_handler = ConfigHandler() - config_handler.get_config.retry.wait = wait_none() - - with self.assertRaises(Exception): - config_handler.get_config() - - @responses.activate - def test_get_config_max_retries_error(self): - retry_limit = 5 - config_handler = ConfigHandler() - config_handler.get_config.retry.wait = wait_none() - - for __ in range(retry_limit): - responses.add(responses.GET, self.cbs_url, status=500) - - with self.assertRaises(Exception): - config_handler.get_config() - self.assertEqual(retry_limit, len(responses.calls)) - - @responses.activate - def test_get_config_less_than_5_retries_success(self): - retry_attempts = 4 - responses.add(responses.GET, self.cbs_url, status=500) - responses.add(responses.GET, self.cbs_url, status=400) - responses.add(responses.GET, self.cbs_url, status=300) - responses.add(responses.GET, self.cbs_url, json=json.dumps(self.expected_config), - status=200) - - config_handler = ConfigHandler() - config_handler.get_config.retry.wait = wait_none() - config_handler.get_config() - - self.assertEqual(retry_attempts, len(responses.calls)) - - @staticmethod - def _get_expected_config(): - with open(path.join(path.dirname(__file__), 'expected_config.json'))as json_file: - return json.load(json_file) diff --git a/components/pm-subscription-handler/tests/data/cbs_data.json b/components/pm-subscription-handler/tests/data/cbs_data.json deleted file mode 100644 index ccc0626d..00000000 --- a/components/pm-subscription-handler/tests/data/cbs_data.json +++ /dev/null @@ -1,109 +0,0 @@ -{ - "policy":{ - "subscription":{ - "subscriptionName":"ExtraPM-All-gNB-R2B", - "administrativeState":"UNLOCKED", - "fileBasedGP":15, - "fileLocation":"\/pm\/pm.xml", - "nfTypeModelInvariantId":"2829292", - "nfFilter":{ - "swVersions":[ - "1.0.0", - "1.0.1" - ], - "nfNames":[ - "^pnf.*", - "^vnf.*" - ] - }, - "measurementGroups":[ - { - "measurementGroup":{ - "measurementTypes":[ - { - "measurementType":"countera" - }, - { - "measurementType":"counterb" - } - ], - "managedObjectDNsBasic":[ - { - "DN":"dna" - }, - { - "DN":"dnb" - } - ] - } - }, - { - "measurementGroup":{ - "measurementTypes":[ - { - "measurementType":"counterc" - }, - { - "measurementType":"counterd" - } - ], - "managedObjectDNsBasic":[ - { - "DN":"dnc" - }, - { - "DN":"dnd" - } - ] - } - } - ] - } - }, - "config":{ - "aaf_password":"demo123456!", - "aaf_identity":"dcae@dcae.onap.org", - "cert_path":"/opt/app/pm-mapper/etc/certs/cert.pem", - "key_path":"/opt/app/pm-mapper/etc/certs/key.pem", - "streams_subscribes":{ - "aai_subscriber":{ - "type":"message_router", - "dmaap_info":{ - "topic_url":"https://node:30226/events/AAI_EVENT", - "client_role":"org.onap.dcae.aaiSub", - "location":"san-francisco", - "client_id":"1575976809466" - } - }, - "policy_pm_subscriber":{ - "type":"message_router", - "dmaap_info":{ - "topic_url":"https://node:30226/events/org.onap.dmaap.mr.PM_SUBSCRIPTIONS", - "client_role":"org.onap.dcae.pmSubscriber", - "location":"san-francisco", - "client_id":"1575876809456" - } - } - }, - "streams_publishes":{ - "policy_pm_publisher":{ - "type":"message_router", - "dmaap_info":{ - "topic_url":"https://node:30226/events/org.onap.dmaap.mr.PM_SUBSCRIPTIONS", - "client_role":"org.onap.dcae.pmPublisher", - "location":"san-francisco", - "client_id":"1475976809466" - } - }, - "other_publisher":{ - "type":"message_router", - "dmaap_info":{ - "topic_url":"https://node:30226/events/org.onap.dmaap.mr.SOME_OTHER_TOPIC", - "client_role":"org.onap.dcae.pmControlPub", - "location":"san-francisco", - "client_id":"1875976809466" - } - } - } - } -} \ No newline at end of file diff --git a/components/pm-subscription-handler/tests/data/cbs_data_1.json b/components/pm-subscription-handler/tests/data/cbs_data_1.json new file mode 100644 index 00000000..ccc0626d --- /dev/null +++ b/components/pm-subscription-handler/tests/data/cbs_data_1.json @@ -0,0 +1,109 @@ +{ + "policy":{ + "subscription":{ + "subscriptionName":"ExtraPM-All-gNB-R2B", + "administrativeState":"UNLOCKED", + "fileBasedGP":15, + "fileLocation":"\/pm\/pm.xml", + "nfTypeModelInvariantId":"2829292", + "nfFilter":{ + "swVersions":[ + "1.0.0", + "1.0.1" + ], + "nfNames":[ + "^pnf.*", + "^vnf.*" + ] + }, + "measurementGroups":[ + { + "measurementGroup":{ + "measurementTypes":[ + { + "measurementType":"countera" + }, + { + "measurementType":"counterb" + } + ], + "managedObjectDNsBasic":[ + { + "DN":"dna" + }, + { + "DN":"dnb" + } + ] + } + }, + { + "measurementGroup":{ + "measurementTypes":[ + { + "measurementType":"counterc" + }, + { + "measurementType":"counterd" + } + ], + "managedObjectDNsBasic":[ + { + "DN":"dnc" + }, + { + "DN":"dnd" + } + ] + } + } + ] + } + }, + "config":{ + "aaf_password":"demo123456!", + "aaf_identity":"dcae@dcae.onap.org", + "cert_path":"/opt/app/pm-mapper/etc/certs/cert.pem", + "key_path":"/opt/app/pm-mapper/etc/certs/key.pem", + "streams_subscribes":{ + "aai_subscriber":{ + "type":"message_router", + "dmaap_info":{ + "topic_url":"https://node:30226/events/AAI_EVENT", + "client_role":"org.onap.dcae.aaiSub", + "location":"san-francisco", + "client_id":"1575976809466" + } + }, + "policy_pm_subscriber":{ + "type":"message_router", + "dmaap_info":{ + "topic_url":"https://node:30226/events/org.onap.dmaap.mr.PM_SUBSCRIPTIONS", + "client_role":"org.onap.dcae.pmSubscriber", + "location":"san-francisco", + "client_id":"1575876809456" + } + } + }, + "streams_publishes":{ + "policy_pm_publisher":{ + "type":"message_router", + "dmaap_info":{ + "topic_url":"https://node:30226/events/org.onap.dmaap.mr.PM_SUBSCRIPTIONS", + "client_role":"org.onap.dcae.pmPublisher", + "location":"san-francisco", + "client_id":"1475976809466" + } + }, + "other_publisher":{ + "type":"message_router", + "dmaap_info":{ + "topic_url":"https://node:30226/events/org.onap.dmaap.mr.SOME_OTHER_TOPIC", + "client_role":"org.onap.dcae.pmControlPub", + "location":"san-francisco", + "client_id":"1875976809466" + } + } + } + } +} \ No newline at end of file diff --git a/components/pm-subscription-handler/tests/data/cbs_data_2.json b/components/pm-subscription-handler/tests/data/cbs_data_2.json new file mode 100755 index 00000000..43f67e88 --- /dev/null +++ b/components/pm-subscription-handler/tests/data/cbs_data_2.json @@ -0,0 +1,68 @@ +{ + "config":{}, + "policy": { + "subscription": { + "subscriptionName": "someExtraPM-AllKista-gNB-R2B", + "administrativeState": "UNLOCKED", + "fileBasedGP": 15, + "fileLocation": "c:\/\/PM", + "nfTypeModelInvariantId": "2829292", + "nfFilter": { + "swVersions": [ + "A21", + "B" + ], + "nfNames": [ + "ABC", + "DEF", + "foo.*" + ] + }, + "measurementGroups": [ + { + "measurementGroup": { + "measurementTypes": [ + { + "measurementType": "countera" + }, + { + "measurementType": "counterb" + } + ], + "managedObjectDNsBasic": [ + { + "DN": "dna" + }, + { + "DN": "dnb" + } + ] + } + }, + { + "measurementGroup": { + "measurementTypes": [ + { + "measurementType": "counterc" + }, + { + "measurementType": "counterd" + } + ], + "managedObjectDNsBasic": [ + { + "DN": "dnc" + }, + { + "DN": "dnd" + } + ] + } + } + ] + } +} + + + +} \ No newline at end of file diff --git a/components/pm-subscription-handler/tests/expected_config.json b/components/pm-subscription-handler/tests/expected_config.json deleted file mode 100755 index 43f67e88..00000000 --- a/components/pm-subscription-handler/tests/expected_config.json +++ /dev/null @@ -1,68 +0,0 @@ -{ - "config":{}, - "policy": { - "subscription": { - "subscriptionName": "someExtraPM-AllKista-gNB-R2B", - "administrativeState": "UNLOCKED", - "fileBasedGP": 15, - "fileLocation": "c:\/\/PM", - "nfTypeModelInvariantId": "2829292", - "nfFilter": { - "swVersions": [ - "A21", - "B" - ], - "nfNames": [ - "ABC", - "DEF", - "foo.*" - ] - }, - "measurementGroups": [ - { - "measurementGroup": { - "measurementTypes": [ - { - "measurementType": "countera" - }, - { - "measurementType": "counterb" - } - ], - "managedObjectDNsBasic": [ - { - "DN": "dna" - }, - { - "DN": "dnb" - } - ] - } - }, - { - "measurementGroup": { - "measurementTypes": [ - { - "measurementType": "counterc" - }, - { - "measurementType": "counterd" - } - ], - "managedObjectDNsBasic": [ - { - "DN": "dnc" - }, - { - "DN": "dnd" - } - ] - } - } - ] - } -} - - - -} \ No newline at end of file diff --git a/components/pm-subscription-handler/tests/test_aai_service.py b/components/pm-subscription-handler/tests/test_aai_service.py index 7b71d3e8..7f4735a9 100644 --- a/components/pm-subscription-handler/tests/test_aai_service.py +++ b/components/pm-subscription-handler/tests/test_aai_service.py @@ -1,5 +1,5 @@ # ============LICENSE_START=================================================== -# Copyright (C) 2019 Nordix Foundation. +# Copyright (C) 2019-2020 Nordix Foundation. # ============================================================================ # Licensed under the Apache License, Version 2.0 (the 'License'); # you may not use this file except in compliance with the License. @@ -33,7 +33,7 @@ class AaiClientTestCase(unittest.TestCase): self.env = EnvironmentVarGuard() self.env.set('AAI_SERVICE_HOST', '1.2.3.4') self.env.set('AAI_SERVICE_PORT_AAI_SSL', '8443') - with open(os.path.join(os.path.dirname(__file__), 'data/cbs_data.json'), 'r') as data: + with open(os.path.join(os.path.dirname(__file__), 'data/cbs_data_1.json'), 'r') as data: self.cbs_data = json.load(data) with open(os.path.join(os.path.dirname(__file__), 'data/aai_xnfs.json'), 'r') as data: self.aai_response_data = data.read() @@ -50,7 +50,7 @@ class AaiClientTestCase(unittest.TestCase): @mock.patch.object(Session, 'put') def test_aai_client_get_pm_sub_data_fail(self, mock_session): mock_session.return_value.status_code = 404 - with mock.patch('aai_client._get_all_aai_xnf_data', return_value=None): + with mock.patch('mod.aai_client._get_all_aai_nf_data', return_value=None): with self.assertRaises(RuntimeError): aai_client.get_pmsh_subscription_data(self.cbs_data) @@ -59,14 +59,14 @@ class AaiClientTestCase(unittest.TestCase): responses.add(responses.PUT, 'https://1.2.3.4:8443/aai/v16/query?format=simple&nodesOnly=true', json={'error': 'not found'}, status=404) - self.assertIsNone(aai_client._get_all_aai_xnf_data()) + self.assertIsNone(aai_client._get_all_aai_nf_data()) @responses.activate def test_aai_client_get_all_aai_xnf_data_success(self): responses.add(responses.PUT, 'https://1.2.3.4:8443/aai/v16/query?format=simple&nodesOnly=true', json={'dummy_data': 'blah_blah'}, status=200) - self.assertIsNotNone(aai_client._get_all_aai_xnf_data()) + self.assertIsNotNone(aai_client._get_all_aai_nf_data()) def test_aai_client_get_aai_service_url_fail(self): self.env.clear() diff --git a/components/pm-subscription-handler/tests/test_config_handler.py b/components/pm-subscription-handler/tests/test_config_handler.py new file mode 100755 index 00000000..5e80db5d --- /dev/null +++ b/components/pm-subscription-handler/tests/test_config_handler.py @@ -0,0 +1,95 @@ +# ============LICENSE_START=================================================== +# Copyright (C) 2019-2020 Nordix Foundation. +# ============================================================================ +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# +# SPDX-License-Identifier: Apache-2.0 +# ============LICENSE_END===================================================== + +import json +import unittest +from os import environ +from os import path + +import responses +from tenacity import wait_none + +from pmsh_service.mod.config_handler import ConfigHandler + + +class ConfigHandlerTestCase(unittest.TestCase): + + def setUp(self): + self.env_vars = {'CONFIG_BINDING_SERVICE_SERVICE_HOST': 'cbs_hostname', + 'CONFIG_BINDING_SERVICE_SERVICE_PORT': '10000', + 'HOSTNAME': 'hostname'} + for key, value in self.env_vars.items(): + environ[key] = value + self.cbs_url = 'http://cbs_hostname:10000/service_component_all/hostname' + with open(path.join(path.dirname(__file__), 'data/cbs_data_2.json'))as json_file: + self.expected_config = json.load(json_file) + + def test_missing_environment_variable(self): + for key, value in self.env_vars.items(): + with self.assertRaises(KeyError): + environ.pop(key) + test_value = globals()[value] + test_value() + environ[key] = value + + @responses.activate + def test_get_config_success(self): + responses.add(responses.GET, self.cbs_url, json=json.dumps(self.expected_config), + status=200) + + config_handler = ConfigHandler() + config_handler.get_config.retry.wait = wait_none() + + self.assertEqual(self.expected_config, config_handler.get_config()) + + @responses.activate + def test_get_config_error(self): + responses.add(responses.GET, self.cbs_url, status=404) + config_handler = ConfigHandler() + config_handler.get_config.retry.wait = wait_none() + + with self.assertRaises(Exception): + config_handler.get_config() + + @responses.activate + def test_get_config_max_retries_error(self): + retry_limit = 5 + config_handler = ConfigHandler() + config_handler.get_config.retry.wait = wait_none() + + for __ in range(retry_limit): + responses.add(responses.GET, self.cbs_url, status=500) + + with self.assertRaises(Exception): + config_handler.get_config() + self.assertEqual(retry_limit, len(responses.calls)) + + @responses.activate + def test_get_config_less_than_5_retries_success(self): + retry_attempts = 4 + responses.add(responses.GET, self.cbs_url, status=500) + responses.add(responses.GET, self.cbs_url, status=400) + responses.add(responses.GET, self.cbs_url, status=300) + responses.add(responses.GET, self.cbs_url, json=json.dumps(self.expected_config), + status=200) + + config_handler = ConfigHandler() + config_handler.get_config.retry.wait = wait_none() + config_handler.get_config() + + self.assertEqual(retry_attempts, len(responses.calls)) diff --git a/components/pm-subscription-handler/tests/test_network_function.py b/components/pm-subscription-handler/tests/test_network_function.py new file mode 100755 index 00000000..2af1489b --- /dev/null +++ b/components/pm-subscription-handler/tests/test_network_function.py @@ -0,0 +1,61 @@ +# ============LICENSE_START=================================================== +# Copyright (C) 2019-2020 Nordix Foundation. +# ============================================================================ +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# +# SPDX-License-Identifier: Apache-2.0 +# ============LICENSE_END===================================================== +import unittest + +from mod import db, create_test_app +from mod.network_function import NetworkFunction + + +class NetworkFunctionTests(unittest.TestCase): + + def setUp(self): + self.nf_1 = NetworkFunction(nf_name='pnf_1', orchestration_status='Inventoried') + self.nf_2 = NetworkFunction(nf_name='pnf_2', orchestration_status='Active') + self.app = create_test_app() + self.app_context = self.app.app_context() + self.app_context.push() + db.create_all() + + def tearDown(self): + db.session.remove() + db.drop_all() + self.app_context.pop() + + def test_get_network_function(self): + self.nf_1.create() + nf = NetworkFunction.get('pnf_1') + self.assertEqual(self.nf_1.nf_name, nf.nf_name) + + def test_get_network_function_no_match(self): + self.nf_1.create() + nf_name = 'nf2_does_not_exist' + nf = NetworkFunction.get(nf_name) + self.assertEqual(nf, None) + + def test_get_network_functions(self): + self.nf_1.create() + self.nf_2.create() + nfs = NetworkFunction.get_all() + + self.assertEqual(2, len(nfs)) + + def test_create_existing_network_function(self): + nf = self.nf_1.create() + same_nf = self.nf_1.create() + + self.assertEqual(nf, same_nf) diff --git a/components/pm-subscription-handler/tests/test_pmsh_utils.py b/components/pm-subscription-handler/tests/test_pmsh_utils.py index ee79d523..8df2c62a 100644 --- a/components/pm-subscription-handler/tests/test_pmsh_utils.py +++ b/components/pm-subscription-handler/tests/test_pmsh_utils.py @@ -18,12 +18,14 @@ import json import os import unittest +from test.support import EnvironmentVarGuard from unittest import mock from unittest.mock import patch import responses from requests import Session +from mod import get_db_connection_url from mod.pmsh_utils import AppConfig from mod.subscription import Subscription @@ -31,7 +33,7 @@ from mod.subscription import Subscription class PmshUtilsTestCase(unittest.TestCase): def setUp(self): - with open(os.path.join(os.path.dirname(__file__), 'data/cbs_data.json'), 'r') as data: + with open(os.path.join(os.path.dirname(__file__), 'data/cbs_data_1.json'), 'r') as data: self.cbs_data = json.load(data) self.app_conf = AppConfig(**self.cbs_data['config']) self.sub = Subscription(**self.cbs_data['policy']['subscription']) @@ -98,3 +100,18 @@ class PmshUtilsTestCase(unittest.TestCase): mr_policy_sub = self.app_conf.get_mr_sub('policy_pm_subscriber') mr_topic_data = mr_policy_sub.get_from_topic(1) self.assertIsNone(mr_topic_data) + + def test_get_db_connection_url_success(self): + self.env = EnvironmentVarGuard() + self.env.set('PMSH_PG_URL', '1.2.3.4') + self.env.set('PMSH_PG_USERNAME', 'pmsh') + self.env.set('PMSH_PG_PASSWORD', 'pass') + db_url = get_db_connection_url() + self.assertEqual(db_url, 'postgres+psycopg2://pmsh:pass@1.2.3.4:5432/pmsh') + + def test_get_db_connection_url_fail(self): + self.env = EnvironmentVarGuard() + self.env.set('PMSH_PG_USERNAME', 'pmsh') + self.env.set('PMSH_PG_PASSWORD', 'pass') + with self.assertRaises(Exception): + get_db_connection_url() diff --git a/components/pm-subscription-handler/tests/test_subscription.py b/components/pm-subscription-handler/tests/test_subscription.py old mode 100644 new mode 100755 index cbf930fd..3c7651d4 --- a/components/pm-subscription-handler/tests/test_subscription.py +++ b/components/pm-subscription-handler/tests/test_subscription.py @@ -1,5 +1,5 @@ # ============LICENSE_START=================================================== -# Copyright (C) 2019 Nordix Foundation. +# Copyright (C) 2019-2020 Nordix Foundation. # ============================================================================ # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -18,26 +18,106 @@ import json import os import unittest +from test.support import EnvironmentVarGuard +from unittest import mock -from mod.subscription import Subscription, XnfFilter +from requests import Session +import mod.aai_client as aai_client +from mod import db, create_test_app +from mod.network_function import NetworkFunction +from mod.subscription import Subscription, NetworkFunctionFilter -class SubscriptionTestCase(unittest.TestCase): - def setUp(self): - with open(os.path.join(os.path.dirname(__file__), 'data/cbs_data.json'), 'r') as data: - self.cbs_data = json.load(data) - self.sub = Subscription(**self.cbs_data['policy']['subscription']) - self.xnf_filter = XnfFilter(**self.sub.nfFilter) +class SubscriptionTest(unittest.TestCase): + + @mock.patch.object(Session, 'put') + def setUp(self, mock_session): + with open(os.path.join(os.path.dirname(__file__), 'data/aai_xnfs.json'), 'r') as data: + self.aai_response_data = data.read() + mock_session.return_value.status_code = 200 + mock_session.return_value.text = self.aai_response_data + self.env = EnvironmentVarGuard() + self.env.set('AAI_SERVICE_HOST', '1.2.3.4') + self.env.set('AAI_SERVICE_PORT_AAI_SSL', '8443') + with open(os.path.join(os.path.dirname(__file__), 'data/cbs_data_1.json'), 'r') as data: + self.cbs_data_1 = json.load(data) + with open(os.path.join(os.path.dirname(__file__), + 'data/cbs_data_2.json'), 'r') as data: + self.cbs_data_2 = json.load(data) + self.sub_1, self.xnfs = aai_client.get_pmsh_subscription_data(self.cbs_data_1) + self.sub_2, self.xnfs = aai_client.get_pmsh_subscription_data(self.cbs_data_2) + self.nf_1 = NetworkFunction(nf_name='pnf_1', orchestration_status='Inventoried') + self.nf_2 = NetworkFunction(nf_name='pnf_2', orchestration_status='Active') + self.xnf_filter = NetworkFunctionFilter(**self.sub_1.nfFilter) + self.app = create_test_app() + self.app_context = self.app.app_context() + self.app_context.push() + db.create_all() + + def tearDown(self): + db.session.remove() + db.drop_all() + self.app_context.pop() def test_xnf_filter_true(self): - self.assertTrue(self.xnf_filter.is_xnf_in_filter('pnf1')) + self.assertTrue(self.xnf_filter.is_nf_in_filter('pnf1')) def test_xnf_filter_false(self): - self.assertFalse(self.xnf_filter.is_xnf_in_filter('PNF-33')) + self.assertFalse(self.xnf_filter.is_nf_in_filter('PNF-33')) def test_sub_measurement_group(self): - self.assertEqual(len(self.sub.measurementGroups), 2) + self.assertEqual(len(self.sub_1.measurementGroups), 2) def test_sub_file_location(self): - self.assertEqual(self.sub.fileLocation, '/pm/pm.xml') + self.assertEqual(self.sub_1.fileLocation, '/pm/pm.xml') + + def test_get_subscription(self): + sub_name = 'ExtraPM-All-gNB-R2B' + self.sub_1.create() + new_sub = Subscription.get(sub_name) + self.assertEqual(sub_name, new_sub.subscription_name) + + def test_get_subscription_no_match(self): + sub_name = 'sub2_does_not_exist' + sub = Subscription.get(sub_name) + self.assertEqual(sub, None) + + def test_get_subscriptions(self): + self.sub_1.create() + self.sub_2.create() + subs = self.sub_1.get_all() + + self.assertEqual(2, len(subs)) + + def test_create_existing_subscription(self): + sub1 = self.sub_1.create() + same_sub1 = self.sub_1.create() + self.assertEqual(sub1, same_sub1) + self.assertEqual(1, len(self.sub_1.get_all())) + + def test_get_nfs_per_subscription(self): + nf_array = [self.nf_1, self.nf_2] + self.sub_1.add_network_functions_to_subscription(nf_array) + nfs_for_sub_1 = Subscription.get_all_nfs_subscription_relations() + self.assertEqual(2, len(nfs_for_sub_1)) + + def test_add_network_functions_per_subscription(self): + nf_array = [self.nf_1, self.nf_2] + self.sub_1.add_network_functions_to_subscription(nf_array) + nfs_for_sub_1 = Subscription.get_all_nfs_subscription_relations() + self.assertEqual(2, len(nfs_for_sub_1)) + new_nf_array = [NetworkFunction(nf_name='vnf_3', orchestration_status='Inventoried')] + self.sub_1.add_network_functions_to_subscription(new_nf_array) + nf_subs = Subscription.get_all_nfs_subscription_relations() + print(nf_subs) + self.assertEqual(3, len(nf_subs)) + + def test_add_duplicate_network_functions_per_subscription(self): + nf_array = [self.nf_1] + self.sub_1.add_network_functions_to_subscription(nf_array) + nf_subs = Subscription.get_all_nfs_subscription_relations() + self.assertEqual(1, len(nf_subs)) + self.sub_1.add_network_functions_to_subscription(nf_array) + nf_subs = Subscription.get_all_nfs_subscription_relations() + self.assertEqual(1, len(nf_subs)) -- cgit 1.2.3-korg