diff options
Diffstat (limited to 'components/pm-subscription-handler/tests/services/test_subscription_service.py')
| -rw-r--r-- | components/pm-subscription-handler/tests/services/test_subscription_service.py | 22 |
1 files changed, 21 insertions, 1 deletions
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 a0e4d541..44fb4eff 100644 --- a/components/pm-subscription-handler/tests/services/test_subscription_service.py +++ b/components/pm-subscription-handler/tests/services/test_subscription_service.py @@ -29,6 +29,7 @@ from mod.api.custom_exception import DuplicateDataException, InvalidDataExceptio from mod.pmsh_config import AppConfig from tests.base_setup import BaseClassSetup from mod.api.services import subscription_service, nf_service, measurement_group_service +from tests.base_setup import create_multiple_subscription_data class SubscriptionServiceTestCase(BaseClassSetup): @@ -170,7 +171,7 @@ class SubscriptionServiceTestCase(BaseClassSetup): @patch.object(aai_client, '_get_all_aai_nf_data') @patch.object(aai_client, 'get_aai_model_data') - @patch.object(measurement_group_service, 'apply_nf_to_measgroup') + @patch.object(measurement_group_service, 'apply_nf_status_to_measurement_group') @patch.object(NetworkFunctionFilter, 'get_network_function_filter') def test_apply_measurement_grp_to_nfs(self, mock_filter_call, mock_apply_nf, mock_model_aai, mock_aai): @@ -357,3 +358,22 @@ class SubscriptionServiceTestCase(BaseClassSetup): db_string = '{}' db_array = convert_db_string_to_list(db_string) self.assertEqual(len(db_array), 0) + + @patch('mod.api.services.subscription_service.query_all_subscriptions', + MagicMock(return_value=create_multiple_subscription_data( + ['sub_demo_one', 'sub_demo_two']))) + def test_get_subscriptions_list(self): + subs = subscription_service.get_subscriptions_list() + self.assertEqual(subs[0]['subscription']['subscriptionName'], 'sub_demo_one') + self.assertEqual(subs[1]['subscription']['subscriptionName'], 'sub_demo_two') + self.assertEqual(subs[1]['subscription']['measurementGroups'][0]['measurementGroup'] + ['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=[])) + def test_get_subscriptions_list_empty(self): + subs = subscription_service.get_subscriptions_list() + self.assertEqual(subs, []) |