# ============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 mod.api.db_models import SubscriptionModel
from mod.network_function import NetworkFunction
from mod.pmsh_utils import AppConfig
from mod.policy_response_handler import PolicyResponseHandler, policy_response_handle_functions
from mod.subscription import AdministrativeState, SubNfState


class PolicyResponseHandlerTest(TestCase):

    @patch('mod.pmsh_utils.AppConfig._get_pmsh_config')
    @patch('mod.create_app')
    @patch('mod.subscription.Subscription')
    @patch('mod.pmsh_utils._MrSub')
    def setUp(self, mock_mr_sub, mock_sub, mock_app, mock_get_app_conf):
        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
        mock_get_app_conf.return_value = self.cbs_data
        self.app_conf = AppConfig()
        self.sub = self.app_conf.subscription
        self.mock_app = mock_app
        self.nf = NetworkFunction(nf_name='nf1')
        self.policy_response_handler = PolicyResponseHandler(self.mock_policy_mr_sub,
                                                             self.app_conf,
                                                             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.app_conf.subscription.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.app_conf.subscription.subscriptionName,
                AdministrativeState.LOCKED.value,
                self.nf.nf_name, 'failed')
            mock_update_sub_nf.assert_called_with(
                subscription_name=self.app_conf.subscription.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.app_conf.subscription.subscriptionName,
                AdministrativeState.UNLOCKED.value,
                self.nf.nf_name, 'success')
            mock_update_sub_nf.assert_called_with(
                subscription_name=self.app_conf.subscription.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.app_conf.subscription.subscriptionName,
                AdministrativeState.UNLOCKED.value,
                self.nf.nf_name, 'failed')
            mock_update_sub_nf.assert_called_with(
                subscription_name=self.app_conf.subscription.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.app_conf.subscription.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.logger.error')
    @patch('mod.subscription.Subscription.get')
    def test_poll_policy_topic_exception(self, mock_get_sub, mock_logger):
        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()
        mock_logger.assert_called()