[PMSH] Refactor subscription processor and policy response handler

Signed-off-by: ERIMROB <robertas.rimkus@est.tech>
Change-Id: I91964848df8f7455169650b138b46d8dfc326b6f
Issue-ID: DCAEGEN2-1820

1 files changed, 133 insertions, 0 deletions

diff --git a/components/pm-subscription-handler/tests/test_policy_response_handler.py b/components/pm-subscription-handler/tests/test_policy_response_handler.py
new file mode 100644
index 00000000..1cf947fb
--- /dev/null
+++ b/components/pm-subscription-handler/tests/test_policy_response_handler.py
@@ -0,0 +1,133 @@
+# ============LICENSE_START===================================================
+#  Copyright (C) 2019-2020 Nordix Foundation.
+# ============================================================================
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# SPDX-License-Identifier: Apache-2.0
+# ============LICENSE_END=====================================================
+import json
+import os
+from unittest import TestCase
+from unittest.mock import patch
+
+from tenacity import stop_after_attempt
+
+from mod.db_models import SubscriptionModel
+from mod.network_function import NetworkFunction
+from mod.subscription import AdministrativeState, SubNfState
+from mod.policy_response_handler import PolicyResponseHandler, policy_response_handle_functions
+
+
+class PolicyResponseHandlerTest(TestCase):
+
+    @patch('mod.create_app')
+    @patch('mod.subscription.Subscription')
+    @patch('mod.pmsh_utils._MrSub')
+    def setUp(self, mock_mr_sub, mock_sub, mock_app):
+        with open(os.path.join(os.path.dirname(__file__), 'data/cbs_data_1.json'), 'r') as data:
+            self.cbs_data = json.load(data)
+        self.mock_policy_mr_sub = mock_mr_sub
+        self.mock_sub = mock_sub
+        self.mock_sub.subscriptionName = 'ExtraPM-All-gNB-R2B'
+        self.mock_app = mock_app
+        self.nf = NetworkFunction(nf_name='nf1')
+        self.policy_response_handler = PolicyResponseHandler(self.mock_policy_mr_sub,
+                                                             self.mock_sub.subscriptionName,
+                                                             self.mock_app)
+
+    @patch('mod.network_function.NetworkFunction.delete')
+    def test_handle_response_locked_success(self, mock_delete):
+        with patch.dict(policy_response_handle_functions,
+                        {AdministrativeState.LOCKED.value: {'success': mock_delete}}):
+            self.policy_response_handler._handle_response(self.mock_sub.subscriptionName,
+                                                          AdministrativeState.LOCKED.value,
+                                                          self.nf.nf_name, 'success')
+
+            mock_delete.assert_called()
+
+    @patch('mod.subscription.Subscription.update_sub_nf_status')
+    def test_handle_response_locked_failed(self, mock_update_sub_nf):
+        with patch.dict(policy_response_handle_functions,
+                        {AdministrativeState.LOCKED.value: {'failed': mock_update_sub_nf}}):
+            self.policy_response_handler._handle_response(self.mock_sub.subscriptionName,
+                                                          AdministrativeState.LOCKED.value,
+                                                          self.nf.nf_name, 'failed')
+            mock_update_sub_nf.assert_called_with(subscription_name=self.mock_sub.subscriptionName,
+                                                  status=SubNfState.DELETE_FAILED.value,
+                                                  nf_name=self.nf.nf_name)
+
+    @patch('mod.subscription.Subscription.update_sub_nf_status')
+    def test_handle_response_unlocked_success(self, mock_update_sub_nf):
+        with patch.dict(policy_response_handle_functions,
+                        {AdministrativeState.UNLOCKED.value: {'success': mock_update_sub_nf}}):
+            self.policy_response_handler._handle_response(self.mock_sub.subscriptionName,
+                                                          AdministrativeState.UNLOCKED.value,
+                                                          self.nf.nf_name, 'success')
+            mock_update_sub_nf.assert_called_with(subscription_name=self.mock_sub.subscriptionName,
+                                                  status=SubNfState.CREATED.value,
+                                                  nf_name=self.nf.nf_name)
+
+    @patch('mod.subscription.Subscription.update_sub_nf_status')
+    def test_handle_response_unlocked_failed(self, mock_update_sub_nf):
+        with patch.dict(policy_response_handle_functions,
+                        {AdministrativeState.UNLOCKED.value: {'failed': mock_update_sub_nf}}):
+            self.policy_response_handler._handle_response(self.mock_sub.subscriptionName,
+                                                          AdministrativeState.UNLOCKED.value,
+                                                          self.nf.nf_name, 'failed')
+            mock_update_sub_nf.assert_called_with(subscription_name=self.mock_sub.subscriptionName,
+                                                  status=SubNfState.CREATE_FAILED.value,
+                                                  nf_name=self.nf.nf_name)
+
+    def test_handle_response_exception(self):
+        self.assertRaises(Exception, self.policy_response_handler._handle_response, 'sub1',
+                          'wrong_state', 'nf1', 'wrong_message')
+
+    @patch('mod.policy_response_handler.PolicyResponseHandler._handle_response')
+    @patch('mod.subscription.Subscription.get')
+    def test_poll_policy_topic_calls_methods_correct_sub(self, mock_get_sub, mock_handle_response):
+        response_data = ['{"name": "ResponseEvent","status": { "subscriptionName": '
+                         '"ExtraPM-All-gNB-R2B", "nfName": "pnf300", "message": "success" } }']
+        self.mock_policy_mr_sub.get_from_topic.return_value = response_data
+        mock_get_sub.return_value = SubscriptionModel(subscription_name='ExtraPM-All-gNB-R2B',
+                                                      status=AdministrativeState.UNLOCKED.value)
+        self.policy_response_handler.poll_policy_topic()
+
+        self.mock_policy_mr_sub.get_from_topic.assert_called()
+
+        mock_handle_response.assert_called_with(self.mock_sub.subscriptionName,
+                                                AdministrativeState.UNLOCKED.value, 'pnf300',
+                                                'success')
+
+    @patch('mod.policy_response_handler.PolicyResponseHandler._handle_response')
+    @patch('mod.subscription.Subscription.get')
+    def test_poll_policy_topic_no_method_calls_incorrect_sub(self, mock_get_sub,
+                                                             mock_handle_response):
+        response_data = ['{"name": "ResponseEvent","status": { "subscriptionName": '
+                         '"Different_Subscription", "nfName": "pnf300", "message": "success" } }']
+        self.mock_policy_mr_sub.get_from_topic.return_value = response_data
+        mock_get_sub.return_value = SubscriptionModel(subscription_name='ExtraPM-All-gNB-R2B',
+                                                      status=AdministrativeState.UNLOCKED.value)
+        self.policy_response_handler.poll_policy_topic()
+
+        self.mock_policy_mr_sub.get_from_topic.assert_called()
+
+        mock_handle_response.assert_not_called()
+
+    @patch('mod.subscription.Subscription.get')
+    def test_poll_policy_topic_exception(self, mock_get_sub):
+        self.mock_policy_mr_sub.get_from_topic.return_value = 'wrong_return'
+        mock_get_sub.return_value = SubscriptionModel(subscription_name='ExtraPM-All-gNB-R2B',
+                                                      status=AdministrativeState.UNLOCKED.value)
+        self.policy_response_handler.poll_policy_topic.retry.stop = stop_after_attempt(1)
+
+        self.assertRaises(Exception, self.policy_response_handler.poll_policy_topic)