[PMSH] Read all subscriptions API

Issue-ID: DCAEGEN2-2847
Signed-off-by: raviteja.karumuri <raviteja.karumuri@est.tech>
Change-Id: I6a2cbc127e12f2f6b051ed5f58fd2fa584a71908

1 files changed, 35 insertions, 1 deletions

diff --git a/components/pm-subscription-handler/tests/base_setup.py b/components/pm-subscription-handler/tests/base_setup.py
index 4328f59c..be7d1b8f 100755
--- a/components/pm-subscription-handler/tests/base_setup.py
+++ b/components/pm-subscription-handler/tests/base_setup.py
@@ -29,11 +29,29 @@ 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',
@@ -51,6 +69,22 @@ def subscription_data(subscription_name):
     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
+
+
 class BaseClassSetup(TestCase):
     app = None
     app_context = None