diff options
Diffstat (limited to 'components/pm-subscription-handler/tests/test_subscription_handler.py')
-rw-r--r-- | components/pm-subscription-handler/tests/test_subscription_handler.py | 76 |
1 files changed, 36 insertions, 40 deletions
diff --git a/components/pm-subscription-handler/tests/test_subscription_handler.py b/components/pm-subscription-handler/tests/test_subscription_handler.py index 3eb12bca..a277da66 100644 --- a/components/pm-subscription-handler/tests/test_subscription_handler.py +++ b/components/pm-subscription-handler/tests/test_subscription_handler.py @@ -21,82 +21,78 @@ from unittest import TestCase from unittest.mock import patch from mod.network_function import NetworkFunction +from mod.pmsh_utils import AppConfig from mod.subscription import AdministrativeState from mod.subscription_handler import SubscriptionHandler class SubscriptionHandlerTest(TestCase): + @patch('mod.pmsh_utils.AppConfig._get_pmsh_config') @patch('mod.create_app') - @patch('mod.subscription.Subscription') @patch('mod.pmsh_utils._MrPub') - @patch('mod.pmsh_utils.AppConfig') @patch('mod.pmsh_utils.PeriodicTask') - def setUp(self, mock_aai_event_thread, mock_app_conf, mock_mr_pub, - mock_sub, mock_app): + def setUp(self, mock_aai_event_thread, mock_mr_pub, mock_app, mock_get_pmsh_config): with open(os.path.join(os.path.dirname(__file__), 'data/cbs_data_1.json'), 'r') as data: - self.cbs_data_1 = json.load(data) + self.cbs_data = json.load(data) + mock_get_pmsh_config.return_value = self.cbs_data + self.app_conf = AppConfig() self.mock_app = mock_app - self.mock_sub = mock_sub self.mock_mr_pub = mock_mr_pub - self.mock_app_conf = mock_app_conf self.mock_aai_event_thread = mock_aai_event_thread self.nf_1 = NetworkFunction(nf_name='pnf_1') self.nf_2 = NetworkFunction(nf_name='pnf_2') self.nfs = [self.nf_1, self.nf_2] + def tearDown(self): + pass + @patch('mod.logger.info') - @patch('mod.aai_client.get_pmsh_subscription_data') + @patch('mod.aai_client.get_pmsh_nfs_from_aai') def test_execute_no_change_of_state(self, mock_get_aai, mock_logger): - mock_get_aai.return_value = self.mock_sub, self.nfs + mock_get_aai.return_value = self.nfs sub_handler = SubscriptionHandler(AdministrativeState.UNLOCKED.value, self.mock_mr_pub, - self.mock_app, self.mock_app_conf, + self.mock_app, self.app_conf, self.mock_aai_event_thread) - with patch('mod.pmsh_utils.ConfigHandler.get_pmsh_config', return_value=self.cbs_data_1): - sub_handler.execute() + sub_handler.execute() mock_logger.assert_called_with('Administrative State did not change in the Config') - @patch('mod.aai_client.get_pmsh_subscription_data') - def test_execute_change_of_state_unlocked(self, mock_get_aai): - mock_get_aai.return_value = self.mock_sub, self.nfs + @patch('mod.subscription.Subscription.process_subscription') + @patch('mod.aai_client.get_pmsh_nfs_from_aai') + def test_execute_change_of_state_unlocked(self, mock_get_aai, mock_process_sub): + mock_get_aai.return_value = self.nfs self.mock_aai_event_thread.return_value.start.return_value = 'start_method' sub_handler = SubscriptionHandler(AdministrativeState.LOCKED.value, self.mock_mr_pub, - self.mock_app, self.mock_app_conf, + self.mock_app, self.app_conf, self.mock_aai_event_thread.return_value) - with patch('mod.pmsh_utils.ConfigHandler.get_pmsh_config', return_value=self.cbs_data_1): - sub_handler.execute() - + sub_handler.execute() self.assertEqual(AdministrativeState.UNLOCKED.value, sub_handler.administrative_state) - self.mock_sub.process_subscription.assert_called_with(self.nfs, self.mock_mr_pub, - self.mock_app_conf) + mock_process_sub.assert_called_with(self.nfs, self.mock_mr_pub, self.app_conf) self.mock_aai_event_thread.return_value.start.assert_called() - @patch('mod.aai_client.get_pmsh_subscription_data') - def test_execute_change_of_state_locked(self, mock_get_aai): - mock_get_aai.return_value = self.mock_sub, self.nfs + @patch('mod.subscription.Subscription.process_subscription') + @patch('mod.aai_client.get_pmsh_nfs_from_aai') + def test_execute_change_of_state_locked(self, mock_get_aai, mock_process_sub): + mock_get_aai.return_value = self.nfs self.mock_aai_event_thread.return_value.cancel.return_value = 'cancel_method' - self.cbs_data_1['policy']['subscription']['administrativeState'] = \ - AdministrativeState.LOCKED.value + self.app_conf.subscription.administrativeState = AdministrativeState.LOCKED.value sub_handler = SubscriptionHandler(AdministrativeState.UNLOCKED.value, self.mock_mr_pub, - self.mock_app, self.mock_app_conf, + self.mock_app, self.app_conf, self.mock_aai_event_thread.return_value) - with patch('mod.pmsh_utils.ConfigHandler.get_pmsh_config', return_value=self.cbs_data_1): - sub_handler.execute() - + sub_handler.execute() self.assertEqual(AdministrativeState.LOCKED.value, sub_handler.administrative_state) - self.mock_sub.process_subscription.assert_called_with(self.nfs, self.mock_mr_pub, - self.mock_app_conf) + mock_process_sub.assert_called_with(self.nfs, self.mock_mr_pub, self.app_conf) self.mock_aai_event_thread.return_value.cancel.assert_called() + self.app_conf.subscription.administrativeState = AdministrativeState.UNLOCKED.value + @patch('mod.subscription.Subscription.process_subscription') @patch('mod.logger.error') - @patch('mod.aai_client.get_pmsh_subscription_data') - def test_execute_exception(self, mock_get_aai, mock_logger): - mock_get_aai.return_value = self.mock_sub, self.nfs - self.mock_sub.process_subscription.side_effect = Exception + @patch('mod.aai_client.get_pmsh_nfs_from_aai') + def test_execute_exception(self, mock_get_aai, mock_logger, mock_process_sub): + mock_get_aai.return_value = self.nfs + mock_process_sub.side_effect = Exception sub_handler = SubscriptionHandler(AdministrativeState.LOCKED.value, self.mock_mr_pub, - self.mock_app, self.mock_app_conf, + self.mock_app, self.app_conf, self.mock_aai_event_thread) - with patch('mod.pmsh_utils.ConfigHandler.get_pmsh_config', return_value=self.cbs_data_1): - sub_handler.execute() - + sub_handler.execute() mock_logger.assert_called_with('Error occurred during the activation/deactivation process ') |