blob: 3d85123aef751fed941d04ba766e357bbda30d3a [file] [log] [blame]
# 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"]
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()
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()
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()
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()
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()