diff options
author | raviteja.karumuri <raviteja.karumuri@est.tech> | 2021-11-24 13:10:09 +0000 |
---|---|---|
committer | raviteja.karumuri <raviteja.karumuri@est.tech> | 2021-11-29 12:39:30 +0000 |
commit | 7e3042157d736e1f81618b92afc3bab501755a31 (patch) | |
tree | 4301aece78d3c486e3b144e736dbf66fc5bd901a /components/pm-subscription-handler/tests/test_controller.py | |
parent | d7715808aaf9a4f891d2bd571af1a5acc86b4c7b (diff) |
[PMSH] Read all subscriptions API
Issue-ID: DCAEGEN2-2847
Signed-off-by: raviteja.karumuri <raviteja.karumuri@est.tech>
Change-Id: I6a2cbc127e12f2f6b051ed5f58fd2fa584a71908
Diffstat (limited to 'components/pm-subscription-handler/tests/test_controller.py')
-rwxr-xr-x | components/pm-subscription-handler/tests/test_controller.py | 47 |
1 files changed, 37 insertions, 10 deletions
diff --git a/components/pm-subscription-handler/tests/test_controller.py b/components/pm-subscription-handler/tests/test_controller.py index 7bd72a29..77ab889c 100755 --- a/components/pm-subscription-handler/tests/test_controller.py +++ b/components/pm-subscription-handler/tests/test_controller.py @@ -21,12 +21,13 @@ from unittest.mock import patch, MagicMock from http import HTTPStatus from mod import aai_client -from mod.api.controller import status, post_subscription, get_subscription_by_name +from mod.api.controller import status, post_subscription, get_subscription_by_name,\ + get_subscriptions from tests.base_setup import BaseClassSetup from mod.api.db_models import SubscriptionModel, NfMeasureGroupRelationalModel from mod.subscription import SubNfState from mod.network_function import NetworkFunctionFilter -from tests.base_setup import subscription_data +from tests.base_setup import create_subscription_data, create_multiple_subscription_data class ControllerTestCase(BaseClassSetup): @@ -111,11 +112,11 @@ class ControllerTestCase(BaseClassSetup): self.assertEqual(response[1], 400) self.assertEqual(response[0], 'No value provided in subscription name') - @patch('mod.api.services.subscription_service.get_subscription_by_name', - MagicMock(return_value=subscription_data('sub_demo'))) + @patch('mod.api.services.subscription_service.query_subscription_by_name', + MagicMock(return_value=create_subscription_data('sub_demo'))) def test_get_subscription_by_name_api(self): sub, status_code = get_subscription_by_name('sub_demo') - self.assertEqual(status_code, HTTPStatus.OK) + self.assertEqual(status_code, HTTPStatus.OK.value) self.assertEqual(sub['subscription']['subscriptionName'], 'sub_demo') self.assertEqual(sub['subscription']['nfFilter']['nfNames'], ['^pnf.*', '^vnf.*']) @@ -125,16 +126,42 @@ class ControllerTestCase(BaseClassSetup): self.assertEqual(sub['subscription']['operationalPolicyName'], 'pmsh_operational_policy') - @patch('mod.api.services.subscription_service.get_subscription_by_name', + @patch('mod.api.services.subscription_service.query_subscription_by_name', MagicMock(return_value=None)) - def test_get_subscription_by_name_api_error(self): + def test_get_subscription_by_name_api_none(self): sub, status_code = get_subscription_by_name('sub_demo') - self.assertEqual(status_code, HTTPStatus.NOT_FOUND) + self.assertEqual(status_code, HTTPStatus.NOT_FOUND.value) self.assertEqual(sub['error'], 'Subscription was not defined with the name : sub_demo') - @patch('mod.api.services.subscription_service.get_subscription_by_name', + @patch('mod.api.services.subscription_service.query_subscription_by_name', MagicMock(side_effect=Exception('something failed'))) def test_get_subscription_by_name_api_exception(self): sub, status_code = get_subscription_by_name('sub_demo') - self.assertEqual(status_code, HTTPStatus.INTERNAL_SERVER_ERROR) + self.assertEqual(status_code, HTTPStatus.INTERNAL_SERVER_ERROR.value) + + @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_api(self): + subs, status_code = get_subscriptions() + self.assertEqual(status_code, HTTPStatus.OK.value) + 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) + + @patch('mod.api.services.subscription_service.query_all_subscriptions', + MagicMock(return_value=None)) + def test_get_subscriptions_api_none(self): + subs, status_code = get_subscriptions() + self.assertEqual(status_code, HTTPStatus.OK.value) + self.assertEqual(subs, []) + + @patch('mod.api.services.subscription_service.query_all_subscriptions', + MagicMock(side_effect=Exception('something failed'))) + def test_get_subscriptions_api_exception(self): + subs, status_code = get_subscriptions() + self.assertEqual(status_code, HTTPStatus.INTERNAL_SERVER_ERROR.value) |