# Copyright 2019 ZTE Corporation.
#
# 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.

import json
import uuid

import mock
from django.test import TestCase
from rest_framework import status
from rest_framework.test import APIClient
from lcm.pub.database.models import NSInstModel
from lcm.pub.utils import restcall


class TestNsInstanceApi(TestCase):

    def setUp(self):
        self.apiClient = APIClient()
        self.format = 'json'
        self.ns_instances_url = '/api/nslcm/v1/ns_instances'
        self.nsd_id = str(uuid.uuid4())
        self.ns_package_id = str(uuid.uuid4())

    def tearDown(self):
        NSInstModel.objects.all().delete()

    @mock.patch.object(restcall, 'call_req')
    def test_create_ns(self, mock_call_req):
        nspackage_info = {
            "csarId": self.ns_package_id,
            "packageInfo": {
                "nsPackageId": self.ns_package_id,
                "nsdId": self.nsd_id
            }
        }
        r1_query_nspackage_from_catalog = [0, json.JSONEncoder().encode(nspackage_info), '201']
        r2_create_ns_to_aai = [0, json.JSONEncoder().encode({}), '201']
        mock_call_req.side_effect = [r1_query_nspackage_from_catalog, r2_create_ns_to_aai]

        header = {
            'HTTP_GLOBALCUSTOMERID': 'global-customer-id-test1',
            'HTTP_SERVICETYPE': 'service-type-test1'
        }

        data = {
            "nsdId": self.nsd_id,
            "nsName": "ns",
            "nsDescription": "description"
        }
        response = self.apiClient.post(self.ns_instances_url, data=data, format=self.format, **header)
        self.failUnlessEqual(status.HTTP_201_CREATED, response.status_code)
        return response.data['id']

    def test_ns_instances_method_not_allowed(self):
        header = {
            'HTTP_GLOBALCUSTOMERID': 'global-customer-id-test1',
            'HTTP_SERVICETYPE': 'service-type-test1'
        }

        data = {
            "nsdId": self.nsd_id,
            "nsName": "ns",
            "nsDescription": "description"
        }
        response = self.apiClient.delete(self.ns_instances_url, data=data, format=self.format, **header)
        self.failUnlessEqual(status.HTTP_405_METHOD_NOT_ALLOWED, response.status_code)
        response = self.apiClient.put(self.ns_instances_url, data=data, format=self.format, **header)
        self.failUnlessEqual(status.HTTP_405_METHOD_NOT_ALLOWED, response.status_code)
        response = self.apiClient.patch(self.ns_instances_url, data=data, format=self.format, **header)
        self.failUnlessEqual(status.HTTP_405_METHOD_NOT_ALLOWED, response.status_code)

    def test_invidual_ns_instance_method_not_allowed(self):
        header = {
            'HTTP_GLOBALCUSTOMERID': 'global-customer-id-test1',
            'HTTP_SERVICETYPE': 'service-type-test1'
        }

        data = {
            "nsdId": self.nsd_id,
            "nsName": "ns",
            "nsDescription": "description"
        }
        url = self.ns_instances_url + '/1'
        response = self.apiClient.post(url, data=data, format=self.format, **header)
        self.failUnlessEqual(status.HTTP_405_METHOD_NOT_ALLOWED, response.status_code)
        response = self.apiClient.put(url, data=data, format=self.format, **header)
        self.failUnlessEqual(status.HTTP_405_METHOD_NOT_ALLOWED, response.status_code)
        response = self.apiClient.patch(url, data=data, format=self.format, **header)
        self.failUnlessEqual(status.HTTP_405_METHOD_NOT_ALLOWED, response.status_code)

    def test_query_ns(self):
        NSInstModel.objects.all().delete()
        self.test_create_ns()
        response = self.apiClient.get(self.ns_instances_url)
        self.failUnlessEqual(status.HTTP_200_OK, response.status_code, response.data)
        self.assertIsNotNone(response.data)
        self.assertEqual(1, len(response.data))
        self.assertEquals(self.nsd_id, response.data[0]['nsdId'])
        self.assertEquals('ns', response.data[0]['nsInstanceName'])
        self.assertEquals('NOT_INSTANTIATED', response.data[0]['nsState'])

    def test_query_one_ns(self):
        NSInstModel.objects.all().delete()
        id = self.test_create_ns()
        url = self.ns_instances_url + '/' + id
        response = self.apiClient.get(url)
        self.failUnlessEqual(status.HTTP_200_OK, response.status_code, response.data)
        self.assertIsNotNone(response.data)
        self.assertEquals(self.nsd_id, response.data['nsdId'])
        self.assertEquals('ns', response.data['nsInstanceName'])
        self.assertEquals('NOT_INSTANTIATED', response.data['nsState'])

    @mock.patch.object(restcall, 'call_req')
    def test_delete_ns(self, mock_call_req):
        NSInstModel(id="1", nspackage_id="7", nsd_id="2").save()
        ns_info = {
            "service-instance-id": "service-instance-id-9b9348f2-f75d-4559-823d-db7ac138ed34",
            "service-instance-name": "service-instance-name-9b9348f2-f75d-4559-823d-db7ac138ed34",
            "service-type": "service-type-9b9348f2-f75d-4559-823d-db7ac138ed34",
            "service-role": "service-role-9b9348f2-f75d-4559-823d-db7ac138ed34",
            "resource-version": "1505350720009"
        }
        r1_query_ns_to_aai = [0, json.JSONEncoder().encode(ns_info), '200']
        r2_delete_ns_to_aai = [0, json.JSONEncoder().encode({}), '200']
        mock_call_req.side_effect = [r1_query_ns_to_aai, r2_delete_ns_to_aai]
        url = self.ns_instances_url + '/1'
        response = self.apiClient.delete(url)
        self.failUnlessEqual(status.HTTP_204_NO_CONTENT, response.status_code)