summaryrefslogtreecommitdiffstats
path: root/components/pm-subscription-handler/tests
diff options
context:
space:
mode:
authoregernug <gerard.nugent@est.tech>2021-12-06 13:11:15 +0000
committeregernug <gerard.nugent@est.tech>2021-12-09 12:56:03 +0000
commit15dde94eccad074dda76c420f900c977c184b66f (patch)
tree263bc1ac72ca91311a0b77aeb5e49a8292d71965 /components/pm-subscription-handler/tests
parent10aa7f851d2256c5b158f8b4c5d3e2756890f7b7 (diff)
[PMSH] Add NF list to the Read by Subname call
Issue-ID: DCAEGEN2-2992 Add the list of NFs when a request is made for a subscription. This will assist with testing and visibility for the user. Change-Id: I6061f1dd1cbc6c579e98605c1d7bcbcea3c4e77b Signed-off-by: egernug <gerard.nugent@est.tech>
Diffstat (limited to 'components/pm-subscription-handler/tests')
-rwxr-xr-xcomponents/pm-subscription-handler/tests/base_setup.py5
-rw-r--r--components/pm-subscription-handler/tests/services/test_subscription_service.py1
2 files changed, 5 insertions, 1 deletions
diff --git a/components/pm-subscription-handler/tests/base_setup.py b/components/pm-subscription-handler/tests/base_setup.py
index be7d1b8f..33e2e91b 100755
--- a/components/pm-subscription-handler/tests/base_setup.py
+++ b/components/pm-subscription-handler/tests/base_setup.py
@@ -22,7 +22,7 @@ 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, NfSubRelationalModel
from mod.network_function import NetworkFunctionFilter
from mod.pmsh_utils import AppConfig
from mod.pmsh_config import AppConfig as NewAppConfig
@@ -66,6 +66,9 @@ def create_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
diff --git a/components/pm-subscription-handler/tests/services/test_subscription_service.py b/components/pm-subscription-handler/tests/services/test_subscription_service.py
index 8f3f0f4f..44fb4eff 100644
--- a/components/pm-subscription-handler/tests/services/test_subscription_service.py
+++ b/components/pm-subscription-handler/tests/services/test_subscription_service.py
@@ -370,6 +370,7 @@ class SubscriptionServiceTestCase(BaseClassSetup):
['measurementGroupName'], 'MG1')
self.assertEqual(len(subs[1]['subscription']['measurementGroups']), 2)
self.assertEqual(len(subs), 2)
+ self.assertEqual(len(subs[0]['subscription']['nfs']), 2)
@patch('mod.api.services.subscription_service.query_all_subscriptions',
MagicMock(return_value=[]))