1 files changed, 60 insertions, 2 deletions
diff --git a/components/pm-subscription-handler/tests/base_setup.py b/components/pm-subscription-handler/tests/base_setup.py
index 4328f59c..560eaeb8 100755
--- a/components/pm-subscription-handler/tests/base_setup.py
+++ b/components/pm-subscription-handler/tests/base_setup.py
@@ -22,18 +22,37 @@ from unittest import TestCase
 from unittest.mock import patch, MagicMock
 
 from mod import create_app, db
-from mod.api.db_models import NetworkFunctionFilterModel, MeasurementGroupModel, SubscriptionModel
+from mod.api.db_models import NetworkFunctionFilterModel, MeasurementGroupModel, \
+    SubscriptionModel, NetworkFunctionModel, NfSubRelationalModel
 from mod.network_function import NetworkFunctionFilter
 from mod.pmsh_utils import AppConfig
 from mod.pmsh_config import AppConfig as NewAppConfig
 
 
 def get_pmsh_config(file_path='data/cbs_data_1.json'):
+    """
+    Gets PMSH config from the JSON file
+
+    Args:
+        file_path (String): Name of the file with path
+
+    Returns
+        dict: Dictionary representation of the the service configuration
+    """
     with open(os.path.join(os.path.dirname(__file__), file_path), 'r') as data:
         return json.load(data)
 
 
-def subscription_data(subscription_name):
+def create_subscription_data(subscription_name):
+    """
+    Creates subscription model object
+
+    Args:
+        subscription_name (String): Name of the Subscription
+
+    Returns
+        SubscriptionModel: single subscription model object
+    """
     nf_filter = NetworkFunctionFilterModel(subscription_name, '{^pnf.*,^vnf.*}',
                                            '{}', '{}', '{}')
     mg_first = MeasurementGroupModel(subscription_name, 'MG1', 'UNLOCKED', 15, '/pm/pm.xml',
@@ -48,9 +67,48 @@ def subscription_data(subscription_name):
                                            'pmsh_control_loop_name', 'LOCKED')
     subscription_model.network_filter = nf_filter
     subscription_model.measurement_groups = mg_list
+    nf1 = NfSubRelationalModel(subscription_name, "pnf_101", "LOCKED")
+    nf2 = NfSubRelationalModel(subscription_name, "pnf_102", "LOCKED")
+    subscription_model.nfs = [nf1, nf2]
     return subscription_model
 
 
+def create_multiple_subscription_data(subscription_names):
+    """
+    Creates a list of subscription model objects
+
+    Args:
+        subscription_names (List): Name of the Subscriptions
+
+    Returns
+        list (SubscriptionModel): of subscription model objects
+    """
+    subscriptions = []
+    for subscription_name in subscription_names:
+        subscriptions.append(create_subscription_data(subscription_name))
+    return subscriptions
+
+
+def create_multiple_network_function_data(nf_name_list):
+    """
+    Creates list of network function model objects
+
+    Args:
+        nf_name_list (list): Network function names
+
+    Returns
+        list: of network function model objects
+    """
+    nf_list = []
+    for nf_name in nf_name_list:
+        nf = NetworkFunctionModel(nf_name, '10.10.10.32', '2001:0db8:0:0:0:0:1428:57ab',
+                                  '687kj45-d396-4efb-af02-6b83499b12f8',
+                                  'e80a6ae3-cafd-4d24-850d-e14c084a5ca9',
+                                  'model_name', 'pm_control', '1.0.2')
+        nf_list.append(nf)
+    return nf_list
+
+
 class BaseClassSetup(TestCase):
     app = None
     app_context = None