264 lines
8.9 KiB
Python
264 lines
8.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2016 Mirantis, Inc.
|
|
#
|
|
# 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.
|
|
|
|
from nailgun.objects import deployment_graph
|
|
from nailgun.test import base
|
|
|
|
|
|
JSON_TASKS = [
|
|
{
|
|
'id': 'post_deployment_end',
|
|
'type': 'stage',
|
|
'requires': ['post_deployment_start']
|
|
},
|
|
{
|
|
'id': 'primary-controller',
|
|
'parameters': {'strategy': {'type': 'one_by_one'}},
|
|
'required_for': ['deploy_end'],
|
|
'requires': ['deploy_start'],
|
|
'role': ['primary-controller'],
|
|
'type': 'group'
|
|
},
|
|
{
|
|
'id': 'test-roles',
|
|
'role': '*',
|
|
'type': 'group'
|
|
},
|
|
{
|
|
'id': 'cross-dep-test',
|
|
'type': 'puppet',
|
|
'cross-depended-by': ['a', 'b'],
|
|
'cross-depends': ['c', 'd'],
|
|
},
|
|
{
|
|
'id': 'custom-fields-test',
|
|
'type': 'puppet',
|
|
'CUSTOM_FIELD1': 'custom',
|
|
'CUSTOM_FIELD2': ['custom'],
|
|
'CUSTOM_FIELD3': {'custom': 'custom'},
|
|
},
|
|
{
|
|
'id': 'ssl-keys-saving',
|
|
'type': 'puppet',
|
|
'version': '2.0.0',
|
|
'groups': ['primary-controller', 'controller', 'compute',
|
|
'cinder', 'primary-mongo', 'mongo', 'ceph-osd', 'virt'],
|
|
'requires': ['firewall'],
|
|
'condition': "(settings:public_ssl.horizon.value == true or settings:"
|
|
"public_ssl.services.value == true) and settings:"
|
|
"public_ssl.cert_source.value == 'user_uploaded'",
|
|
'required_for': ['deploy_end'],
|
|
'parameters': {
|
|
'puppet_manifest': '/etc/puppet/modules/osnailyfacter/'
|
|
'modular/ssl/ssl_keys_saving.pp',
|
|
'puppet_modules': '/etc/puppet/modules',
|
|
'timeout': 3600
|
|
},
|
|
'test_pre': {
|
|
'cmd': 'ruby /etc/puppet/modules/osnailyfacter/'
|
|
'modular/ssl/ssl_keys_saving_pre.rb'
|
|
}
|
|
}
|
|
]
|
|
|
|
EXPECTED_TASKS = [
|
|
{
|
|
'task_name': 'cross-dep-test',
|
|
'id': 'cross-dep-test', # legacy
|
|
'type': 'puppet',
|
|
'version': '1.0.0',
|
|
'cross_depended_by': ['a', 'b'],
|
|
'cross_depends': ['c', 'd'],
|
|
'cross-depended-by': ['a', 'b'], # legacy
|
|
'cross-depends': ['c', 'd'], # legacy
|
|
}, {
|
|
'task_name': 'post_deployment_end',
|
|
'id': 'post_deployment_end', # legacy
|
|
'type': 'stage',
|
|
'version': '1.0.0',
|
|
'requires': ['post_deployment_start'],
|
|
}, {
|
|
'task_name': 'test-roles',
|
|
'id': 'test-roles',
|
|
'role': '*',
|
|
'roles': '*',
|
|
'version': '1.0.0',
|
|
'type': 'group'
|
|
}, {
|
|
'task_name': 'primary-controller',
|
|
'id': 'primary-controller', # legacy
|
|
'type': 'group',
|
|
'version': '1.0.0',
|
|
'required_for': ['deploy_end'],
|
|
'requires': ['deploy_start'],
|
|
'roles': ['primary-controller'],
|
|
'role': ['primary-controller'], # legacy
|
|
'parameters': {'strategy': {'type': 'one_by_one'}},
|
|
},
|
|
{
|
|
'id': 'custom-fields-test',
|
|
'task_name': 'custom-fields-test',
|
|
'type': 'puppet',
|
|
'version': '1.0.0',
|
|
'CUSTOM_FIELD1': 'custom',
|
|
'CUSTOM_FIELD2': ['custom'],
|
|
'CUSTOM_FIELD3': {'custom': 'custom'},
|
|
},
|
|
{
|
|
'id': 'ssl-keys-saving',
|
|
'task_name': 'ssl-keys-saving',
|
|
'type': 'puppet',
|
|
'version': '2.0.0',
|
|
'groups': ['primary-controller', 'controller', 'compute',
|
|
'cinder', 'primary-mongo', 'mongo', 'ceph-osd', 'virt'],
|
|
'requires': ['firewall'],
|
|
'condition': "(settings:public_ssl.horizon.value == true or "
|
|
"settings:public_ssl.services.value == true) "
|
|
"and settings:public_ssl.cert_source.value == "
|
|
"'user_uploaded'",
|
|
'required_for': ['deploy_end'],
|
|
'parameters': {
|
|
'puppet_manifest': '/etc/puppet/modules/osnailyfacter/'
|
|
'modular/ssl/ssl_keys_saving.pp',
|
|
'puppet_modules': '/etc/puppet/modules',
|
|
'timeout': 3600
|
|
},
|
|
'test_pre': {
|
|
'cmd': 'ruby /etc/puppet/modules/osnailyfacter/'
|
|
'modular/ssl/ssl_keys_saving_pre.rb'
|
|
}
|
|
}
|
|
]
|
|
|
|
|
|
class TestDeploymentGraphModel(
|
|
base.BaseTestCase,
|
|
base.DeploymentTasksTestMixin
|
|
):
|
|
|
|
def test_deployment_graph_creation(self):
|
|
dg = deployment_graph.DeploymentGraph.create({
|
|
'tasks': JSON_TASKS,
|
|
'name': 'test_graph',
|
|
'node_filter': '$.pending_deletion',
|
|
'on_success': {'node_attributes': {'status': 'discover'}},
|
|
'on_error': {'node_attributes': {'status': 'error'}},
|
|
'on_stop': {}
|
|
})
|
|
serialized = deployment_graph.DeploymentGraph.to_dict(dg)
|
|
self.assertEqual(serialized['name'], 'test_graph')
|
|
self.assertEqual('$.pending_deletion', serialized['node_filter'])
|
|
self.assertEqual(
|
|
{'status': 'discover'}, serialized['on_success']['node_attributes']
|
|
)
|
|
self.assertEqual(
|
|
{'status': 'error'}, serialized['on_error']['node_attributes']
|
|
)
|
|
self.assertEqual(
|
|
{}, serialized['on_stop']
|
|
)
|
|
self.assertItemsEqual(serialized['tasks'], EXPECTED_TASKS)
|
|
|
|
def test_create_deployment_graph_with_default_handlers(self):
|
|
dg = deployment_graph.DeploymentGraph.create({
|
|
'tasks': JSON_TASKS, 'name': 'test_graph'
|
|
})
|
|
serialized = deployment_graph.DeploymentGraph.to_dict(dg)
|
|
self.assertEqual(serialized['name'], 'test_graph')
|
|
self.assertItemsEqual(serialized['tasks'], EXPECTED_TASKS)
|
|
|
|
def test_deployment_graph_update(self):
|
|
updated_tasks = [
|
|
{
|
|
'task_name': 'updated',
|
|
'type': 'puppet'
|
|
}
|
|
]
|
|
expected_updated_tasks = [
|
|
{
|
|
'task_name': 'updated',
|
|
'type': 'puppet'
|
|
}
|
|
]
|
|
|
|
dg = deployment_graph.DeploymentGraph.create(
|
|
{'tasks': JSON_TASKS, 'name': 'test_graph'})
|
|
deployment_graph.DeploymentGraph.update(dg, {'tasks': updated_tasks})
|
|
serialized = deployment_graph.DeploymentGraph.to_dict(dg)
|
|
self.assertEqual(serialized['name'], 'test_graph')
|
|
self._compare_tasks(
|
|
expected_updated_tasks, serialized['tasks'])
|
|
|
|
def test_deployment_graph_delete(self):
|
|
cluster = self.env.create()
|
|
deployment_graph.DeploymentGraph.create_for_model(
|
|
{'tasks': JSON_TASKS, 'name': 'test_graph'}, cluster, 'test_graph'
|
|
)
|
|
dg = deployment_graph.DeploymentGraph.get_for_model(
|
|
cluster, 'test_graph'
|
|
)
|
|
self.assertIsNotNone(dg)
|
|
deployment_graph.DeploymentGraph.delete(dg)
|
|
dg = deployment_graph.DeploymentGraph.get_for_model(
|
|
cluster, 'test_graph'
|
|
)
|
|
self.assertIsNone(dg)
|
|
|
|
def test_deployment_graph_create_for_model(self):
|
|
cluster = self.env.create()
|
|
deployment_graph.DeploymentGraph.create_for_model(
|
|
{'tasks': JSON_TASKS, 'name': 'test_graph'}, cluster, 'test_graph'
|
|
)
|
|
dg = deployment_graph.DeploymentGraph.get_for_model(
|
|
cluster, 'test_graph'
|
|
)
|
|
self._compare_tasks(
|
|
EXPECTED_TASKS, deployment_graph.DeploymentGraph.get_tasks(dg)
|
|
)
|
|
|
|
def test_create_tasks_with_yaql_condition(self):
|
|
task = {
|
|
'task_name': 'test',
|
|
'type': 'puppet',
|
|
'condition': {'yaql_exp': 'changed($.nodes)'},
|
|
'roles': ['/.*/'],
|
|
'requires': ['other_task'],
|
|
}
|
|
|
|
graph = deployment_graph.DeploymentGraph.create(
|
|
{'name': 'test_graph', 'tasks': [task]}
|
|
)
|
|
self._compare_tasks(
|
|
[task], deployment_graph.DeploymentGraph.get_tasks(graph)
|
|
)
|
|
|
|
def test_get_metadata(self):
|
|
graph = deployment_graph.DeploymentGraph.create({
|
|
'name': 'test_graph', 'tasks': [],
|
|
'node_filter': '$.deleted',
|
|
'on_success': {},
|
|
'on_error': {'node_attributes': {'error_type': 'deploy'}}
|
|
})
|
|
metadata = deployment_graph.DeploymentGraph.get_metadata(graph)
|
|
self.assertEqual(
|
|
{
|
|
'node_filter': '$.deleted',
|
|
'on_success': {},
|
|
'on_error': {'node_attributes': {'error_type': 'deploy'}}
|
|
},
|
|
metadata
|
|
)
|