
# Copyright 2017-present Open Networking 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.

# Tests for SimpleExampleServiceInstance model policies

import base64
import json
import os
import sys
import unittest
from mock import patch, PropertyMock, ANY, MagicMock
from unit_test_common import setup_sync_unit_test


class TestSimpleExampleServiceInstancePolicy(unittest.TestCase):

    def setUp(self):
        self.unittest_setup = setup_sync_unit_test(os.path.abspath(os.path.dirname(os.path.realpath(__file__))),
                                                   globals(),
                                                   [("simpleexampleservice", "simpleexampleservice.xproto"),
                                                    ("kubernetes-service", "kubernetes.xproto")] )

        self.MockObjectList = self.unittest_setup["MockObjectList"]
        self.model_accessor = self.unittest_setup["model_accessor"]

        sys.path.append(os.path.join(os.path.abspath(os.path.dirname(os.path.realpath(__file__))), "../model_policies"))

        from model_policy_simpleexampleserviceinstance import SimpleExampleServiceInstancePolicy
        self.policy_class = SimpleExampleServiceInstancePolicy

        self.service = SimpleExampleService(service_message="hello", service_secret="p@ssw0rd")
        self.k8s_service = KubernetesService(id=1111)
        self.k8s_service.get_service_instance_class=MagicMock(return_value=KubernetesServiceInstance)
        self.trust_domain = TrustDomain(owner=self.k8s_service, name="test-trust")
        self.image = Image(name="test-image", tag="1.2", kind="container")
        self.slice = Slice(trust_domain=self.trust_domain, service=self.service, default_image = self.image)
        self.service.slices = self.MockObjectList([self.slice])

    def tearDown(self):
        sys.path = self.unittest_setup["sys_path_save"]

    def test_policy_create(self):
        with patch.object(KubernetesService.objects, "get_items") as k8s_service_objects, \
                patch.object(Service.objects, "get_items") as service_objects, \
                patch.object(KubernetesServiceInstance, "save", autospec=True) as ksi_save, \
                patch.object(KubernetesConfigMap, "save", autospec=True) as kcfm_save, \
                patch.object(KubernetesConfigVolumeMount, "save", autospec=True) as kcfm_mnt_save, \
                patch.object(KubernetesSecret, "save", autospec=True) as ksec_save, \
                patch.object(KubernetesSecretVolumeMount, "save", autospec=True) as ksec_mnt_save:
            k8s_service_objects.return_value = [self.k8s_service]
            service_objects.return_value = [self.k8s_service, self.service]

            si = SimpleExampleServiceInstance(name="test-simple-instance",
                                              id=1112,
                                              owner=self.service, tenant_message="world", tenant_secret="l3tm31n")
            si.embedded_images = self.MockObjectList([])

            step = self.policy_class(model_accessor=self.model_accessor)

            desired_data = json.dumps({"index.html": step.render_index(si)})

            desired_secret_data = json.dumps({"service_secret.txt": base64.b64encode("p@ssw0rd"),
                                              "tenant_secret.txt": base64.b64encode("l3tm31n")})

            step.handle_create(si)

            # Saved twice, once with no_sync=True and once with no_sync=False
            self.assertEqual(ksi_save.call_count, 2)
            saved_ksi = ksi_save.call_args[0][0]
            self.assertEqual(saved_ksi.slice, self.slice)
            self.assertEqual(saved_ksi.owner, self.k8s_service)
            self.assertEqual(saved_ksi.image, self.image)
            self.assertEqual(saved_ksi.name, "simpleexampleserviceinstance-1112")

            # Config Map
            self.assertEqual(kcfm_save.call_count, 1)
            saved_cfm = kcfm_save.call_args[0][0]
            self.assertEqual(saved_cfm.name, "simpleexampleserviceinstance-map-1112")
            self.assertEqual(saved_cfm.trust_domain, self.trust_domain)
            self.assertEqual(saved_cfm.data, desired_data)

            # Mouhnt of Config Map to Service Instance
            self.assertEqual(kcfm_mnt_save.call_count, 1)
            saved_cfm_mnt = kcfm_mnt_save.call_args[0][0]
            self.assertEqual(saved_cfm_mnt.config, saved_cfm)
            self.assertEqual(saved_cfm_mnt.service_instance, saved_ksi)

            # Secret
            self.assertEqual(ksec_save.call_count, 1)
            saved_sec = ksec_save.call_args[0][0]
            self.assertEqual(saved_sec.name, "simpleexampleserviceinstance-secret-1112")
            self.assertEqual(saved_sec.trust_domain, self.trust_domain)
            self.assertEqual(saved_sec.data, desired_secret_data)

            # Mount of Secret to Service Instance
            self.assertEqual(ksec_mnt_save.call_count, 1)
            saved_sec_mnt = ksec_mnt_save.call_args[0][0]
            self.assertEqual(saved_sec_mnt.secret, saved_sec)
            self.assertEqual(saved_sec_mnt.service_instance, saved_ksi)

    def test_policy_update(self):
        with patch.object(KubernetesService.objects, "get_items") as k8s_service_objects, \
                patch.object(Service.objects, "get_items") as service_objects, \
                patch.object(KubernetesServiceInstance, "save", autospec=True) as ksi_save, \
                patch.object(KubernetesConfigMap, "save", autospec=True) as kcfm_save, \
                patch.object(KubernetesConfigVolumeMount, "save", autospec=True) as kcfm_mnt_save, \
                patch.object(KubernetesSecret, "save", autospec=True) as ksec_save, \
                patch.object(KubernetesSecretVolumeMount, "save", autospec=True) as ksec_mnt_save:
            k8s_service_objects.return_value = [self.k8s_service]
            service_objects.return_value = [self.k8s_service, self.service]

            si = SimpleExampleServiceInstance(name="test-simple-instance",
                                              id=1112,
                                              owner=self.service, tenant_message="world", tenant_secret="l3tm31n")
            si.embedded_images = self.MockObjectList([])

            ksi = KubernetesServiceInstance(owner=self.k8s_service, slice=self.slice, image=self.image,
                                            name="simpleexampleserviceinstance-1112")

            cfm = KubernetesConfigMap(trust_domain=self.trust_domain, name="simpleexampleserviceinstance-map-1112",
                                      data="junk")

            cfm_mnt = KubernetesConfigVolumeMount(config=cfm, service_instance=ksi)

            si.compute_instance = ksi
            ksi.kubernetes_config_volume_mounts = self.MockObjectList([cfm_mnt])

            step = self.policy_class(model_accessor=self.model_accessor)

            desired_data = json.dumps({"index.html": step.render_index(si)})

            step.handle_update(si)

            self.assertEqual(ksi_save.call_count, 1)

            # Config Map
            self.assertEqual(kcfm_save.call_count, 1)
            saved_cfm = kcfm_save.call_args[0][0]
            self.assertEqual(saved_cfm.name, "simpleexampleserviceinstance-map-1112")
            self.assertEqual(saved_cfm.trust_domain, self.trust_domain)
            self.assertEqual(saved_cfm.data, desired_data)

            self.assertEqual(kcfm_mnt_save.call_count, 0)
            self.assertEqual(ksec_save.call_count, 0)
            self.assertEqual(ksec_mnt_save.call_count, 0)

    def test_policy_update_no_difference(self):
        with patch.object(KubernetesService.objects, "get_items") as k8s_service_objects, \
                patch.object(Service.objects, "get_items") as service_objects, \
                patch.object(KubernetesServiceInstance, "save", autospec=True) as ksi_save, \
                patch.object(KubernetesConfigMap, "save", autospec=True) as kcfm_save, \
                patch.object(KubernetesConfigVolumeMount, "save", autospec=True) as kcfm_mnt_save, \
                patch.object(KubernetesSecret, "save", autospec=True) as ksec_save, \
                patch.object(KubernetesSecretVolumeMount, "save", autospec=True) as ksec_mnt_save:
            k8s_service_objects.return_value = [self.k8s_service]
            service_objects.return_value = [self.k8s_service, self.service]

            step = self.policy_class(model_accessor=self.model_accessor)

            si = SimpleExampleServiceInstance(name="test-simple-instance",
                                              id=1112,
                                              owner=self.service, tenant_message="world", tenant_secret="l3tm31n")
            si.embedded_images = self.MockObjectList([])

            desired_data = json.dumps({"index.html": step.render_index(si)})

            ksi = KubernetesServiceInstance(owner=self.k8s_service, slice=self.slice, image=self.image,
                                            name="simpleexampleserviceinstance-1112")

            cfm = KubernetesConfigMap(trust_domain=self.trust_domain, name="simpleexampleserviceinstance-map-1112",
                                      data=desired_data)

            cfm_mnt = KubernetesConfigVolumeMount(config=cfm, service_instance=ksi)

            si.compute_instance = ksi
            ksi.kubernetes_config_volume_mounts = self.MockObjectList([cfm_mnt])

            step.handle_update(si)

            self.assertEqual(ksi_save.call_count, 0)
            self.assertEqual(kcfm_save.call_count, 0)
            self.assertEqual(kcfm_mnt_save.call_count, 0)
            self.assertEqual(ksec_save.call_count, 0)
            self.assertEqual(ksec_mnt_save.call_count, 0)

    def test_policy_delete(self):
        with patch.object(KubernetesServiceInstance, "delete", autospec=True) as ksi_delete, \
                patch.object(SimpleExampleServiceInstance, "save", autospec=True) as sesi_save:
            si = SimpleExampleServiceInstance(name="test-simple-instance",
                                              id=1112,
                                              owner=self.service, tenant_message="world", tenant_secret="l3tm31n")

            ksi = KubernetesServiceInstance(owner=self.k8s_service, slice=self.slice, image=self.image,
                                            name="simpleexampleserviceinstance-1112")
            si.compute_instance = ksi

            step = self.policy_class(model_accessor=self.model_accessor)

            step.handle_delete(si)

            # The compute instance should have been deleted
            self.assertEqual(ksi_delete.call_count, 1)
            deleted_ksi = ksi_delete.call_args[0][0]
            self.assertEqual(deleted_ksi, ksi)

            # The serviceInstance should have had its compute_instance set to none
            self.assertEqual(si.compute_instance, None)

            # The SimpleExampleServiceInstance should have been saved
            self.assertEqual(sesi_save.call_count, 1)

if __name__ == '__main__':
    unittest.main()