228 lines
8.0 KiB
Python
228 lines
8.0 KiB
Python
# Copyright (c) 2014 OpenStack Foundation.
|
|
# All Rights Reserved.
|
|
#
|
|
# 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 inspect
|
|
import os.path
|
|
import unittest2 as unittest
|
|
import yaml
|
|
|
|
import murano.policy.congress_rules as congress
|
|
|
|
|
|
class MockClassLoader(object):
|
|
def __init__(self, rules):
|
|
"""Create rules like this: ['child->parent', 'child->parent2']."""
|
|
|
|
self._rules_dict = {}
|
|
for rule in rules:
|
|
split = rule.split('->')
|
|
if split[0] in self._rules_dict:
|
|
self._rules_dict[split[0]].append(split[1])
|
|
else:
|
|
self._rules_dict[split[0]] = [split[1]]
|
|
|
|
def get_class(self, name):
|
|
if name not in self._rules_dict:
|
|
return None
|
|
parents = []
|
|
for parent_name in self._rules_dict[name]:
|
|
parents.append(MockClass({'name': parent_name}))
|
|
return MockClass({'parents': parents})
|
|
|
|
|
|
class MockClass(object):
|
|
def __init__(self, entries):
|
|
self.__dict__.update(entries)
|
|
|
|
|
|
class TestCongressRules(unittest.TestCase):
|
|
|
|
def _load_file(self, file_name):
|
|
model_file = os.path.join(
|
|
os.path.dirname(inspect.getfile(self.__class__)), file_name)
|
|
|
|
with open(model_file) as stream:
|
|
return yaml.load(stream)
|
|
|
|
def _create_rules_str(self, model_file, class_loader=None):
|
|
model = self._load_file(model_file)
|
|
|
|
congress_rules = congress.CongressRulesManager()
|
|
rules = congress_rules.convert(model, class_loader)
|
|
rules_str = ", \n".join(map(str, rules))
|
|
print rules_str
|
|
|
|
return rules_str
|
|
|
|
def test_empty_model(self):
|
|
congress_rules = congress.CongressRulesManager()
|
|
rules = congress_rules.convert(None)
|
|
self.assertTrue(len(rules) == 0)
|
|
|
|
def test_convert_simple_app(self):
|
|
rules_str = self._create_and_check_rules_str('model')
|
|
|
|
self.assertFalse("instance." in rules_str)
|
|
|
|
def test_convert_model_two_instances(self):
|
|
rules_str = self._create_and_check_rules_str('model_two_instances')
|
|
|
|
self.assertFalse("\"instances\"" in rules_str)
|
|
|
|
def test_convert_model_with_relations(self):
|
|
rules_str = self._create_rules_str('model_with_relations.yaml')
|
|
|
|
self.assertFalse(
|
|
'murano:properties+("50fa68ff-cd9a-4845-b573-2c80879d158d", '
|
|
'"server", "8ce94f23-f16a-40a1-9d9d-a877266c315d")' in rules_str)
|
|
|
|
self.assertTrue(
|
|
'murano:relationships+("50fa68ff-cd9a-4845-b573-2c80879d158d", '
|
|
'"8ce94f23-f16a-40a1-9d9d-a877266c315d", "server")' in rules_str)
|
|
|
|
self.assertTrue(
|
|
'murano:relationships+("0aafd67e-72e9-4ae0-bb62-fe724f77df2a", '
|
|
'"ed8df2b0-ddd2-4009-b3c9-2e7a368f3cb8", "instance")' in rules_str)
|
|
|
|
def test_convert_model_complex(self):
|
|
self._create_and_check_rules_str('model_complex')
|
|
|
|
def test_convert_renamed_app(self):
|
|
self._create_and_check_rules_str('model_renamed')
|
|
|
|
def test_parent_types(self):
|
|
|
|
# grand-parent
|
|
# / \
|
|
# parent1 parent2
|
|
# \ /
|
|
# io.murano.apps.linux.Git
|
|
|
|
class_loader = MockClassLoader([
|
|
'io.murano.apps.linux.Git->parent1',
|
|
'io.murano.apps.linux.Git->parent2',
|
|
'parent1->grand-parent',
|
|
'parent2->grand-parent'
|
|
])
|
|
|
|
rules_str = self._create_rules_str('model.yaml', class_loader)
|
|
|
|
self.assertTrue(
|
|
'murano:parent_types+("0c810278-7282-4e4a-9d69-7b4c36b6ce6f",'
|
|
' "parent1")' in rules_str)
|
|
|
|
self.assertTrue(
|
|
'murano:parent_types+("0c810278-7282-4e4a-9d69-7b4c36b6ce6f",'
|
|
' "parent2")' in rules_str)
|
|
|
|
self.assertTrue(
|
|
'murano:parent_types+("0c810278-7282-4e4a-9d69-7b4c36b6ce6f",'
|
|
' "grand-parent")' in rules_str)
|
|
|
|
self.assertTrue(
|
|
'murano:parent_types+("0c810278-7282-4e4a-9d69-7b4c36b6ce6f",'
|
|
' "io.murano.apps.linux.Git")' in rules_str)
|
|
|
|
def test_to_dictionary(self):
|
|
"""If model contains object entry (not dict)
|
|
we try to convert to dict using 'to_dictionary' method.
|
|
"""
|
|
|
|
class Struct(object):
|
|
def __init__(self, d):
|
|
self.__dict__ = d
|
|
|
|
def to_dictionary(self):
|
|
return self.__dict__
|
|
|
|
def __getitem__(self, item):
|
|
return self.__dict__[item]
|
|
|
|
d = {'?': {'id': '1', 'type': 't1'},
|
|
'apps': [Struct({'?': {'id': '2', 'type': 't2'},
|
|
'instances': [Struct(
|
|
{'?': {'id': '3', 'type': 't3'}})]})]
|
|
}
|
|
|
|
model = Struct(d)
|
|
|
|
congress_rules = congress.CongressRulesManager()
|
|
rules = congress_rules.convert(model)
|
|
rules_str = ", \n".join(map(str, rules))
|
|
print rules_str
|
|
|
|
self.assertTrue('murano:objects+("1", "1", "t1")' in rules_str)
|
|
self.assertTrue('murano:objects+("2", "1", "t2")' in rules_str)
|
|
self.assertTrue('murano:objects+("3", "1", "t3")' in rules_str)
|
|
|
|
def test_tenant_id(self):
|
|
model = self._load_file("model.yaml")
|
|
congress_rules = congress.CongressRulesManager()
|
|
rules = congress_rules.convert(model, tenant_id='tenant1')
|
|
rules_str = ", \n".join(map(str, rules))
|
|
self.assertTrue(
|
|
'murano:properties+("c86104748a0c4907b4c5981e6d3bce9f", '
|
|
'"tenant_id", "tenant1")' in rules_str)
|
|
|
|
def test_wordpress(self):
|
|
class_loader = MockClassLoader([
|
|
'io.murano.Environment->io.murano.Object',
|
|
'io.murano.resources.NeutronNetwork->io.murano.resources.Network',
|
|
'io.murano.resources.Network->io.murano.Object',
|
|
'io.murano.databases.MySql->io.murano.databases.SqlDatabase',
|
|
'io.murano.databases.MySql->io.murano.Application',
|
|
'io.murano.databases.SqlDatabase->io.murano.Object',
|
|
'io.murano.resources.LinuxInstance->io.murano.resources.Instance',
|
|
'io.murano.resources.Instance->io.murano.Object',
|
|
'io.murano.Application->io.murano.Object',
|
|
'io.murano.apps.apache.ApacheHttpServer->io.murano.Application',
|
|
'io.murano.apps.ZabbixServer->io.murano.Application',
|
|
'io.murano.apps.ZabbixAgent->io.murano.Application',
|
|
'io.murano.apps.WordPress->io.murano.Application',
|
|
|
|
'io.murano.resources.LinuxMuranoInstance->'
|
|
'io.murano.resources.LinuxInstance'
|
|
])
|
|
|
|
self._create_and_check_rules_str('wordpress', class_loader)
|
|
|
|
def _create_and_check_rules_str(self, model_name, class_loader=None):
|
|
rules_str = self._create_rules_str(
|
|
'{0}.yaml'.format(model_name), class_loader)
|
|
self._check_expected_rules(rules_str,
|
|
'expected_rules_{0}.txt'.format(model_name))
|
|
return rules_str
|
|
|
|
def _check_expected_rules(self, rules_str, expected_rules_file_name):
|
|
expected_rules_file = os.path.join(
|
|
os.path.dirname(inspect.getfile(self.__class__)),
|
|
expected_rules_file_name)
|
|
|
|
s = ''
|
|
with open(expected_rules_file) as f:
|
|
for line in f:
|
|
line = line.rstrip('\n')
|
|
if line not in rules_str:
|
|
s += 'Expected rule not found:\n\t' + line + '\n'
|
|
|
|
if len(s) > 0:
|
|
self.fail(s)
|
|
|
|
def test_state_rule(self):
|
|
rules_str = self._create_rules_str('model.yaml')
|
|
|
|
self.assertTrue(
|
|
'murano:states+("c86104748a0c4907b4c5981e6d3bce9f", "PENDING")'
|
|
in rules_str) |