144 lines
5.4 KiB
Python
144 lines
5.4 KiB
Python
# Copyright (c) 2014 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 tempest.api.data_processing import base as dp_base
|
|
from tempest.common.utils import data_utils
|
|
from tempest import test
|
|
|
|
|
|
class ClusterTemplateTest(dp_base.BaseDataProcessingTest):
|
|
"""Link to the API documentation is http://docs.openstack.org/developer/
|
|
sahara/restapi/rest_api_v1.0.html#cluster-templates
|
|
"""
|
|
@classmethod
|
|
def resource_setup(cls):
|
|
super(ClusterTemplateTest, cls).resource_setup()
|
|
# create node group template
|
|
node_group_template = {
|
|
'name': data_utils.rand_name('sahara-ng-template'),
|
|
'description': 'Test node group template',
|
|
'plugin_name': 'vanilla',
|
|
'hadoop_version': '1.2.1',
|
|
'node_processes': ['datanode'],
|
|
'flavor_id': cls.flavor_ref,
|
|
'node_configs': {
|
|
'HDFS': {
|
|
'Data Node Heap Size': 1024
|
|
}
|
|
}
|
|
}
|
|
resp_body = cls.create_node_group_template(**node_group_template)
|
|
node_group_template_id = resp_body['id']
|
|
|
|
cls.full_cluster_template = {
|
|
'description': 'Test cluster template',
|
|
'plugin_name': 'vanilla',
|
|
'hadoop_version': '1.2.1',
|
|
'cluster_configs': {
|
|
'HDFS': {
|
|
'dfs.replication': 2
|
|
},
|
|
'MapReduce': {
|
|
'mapred.map.tasks.speculative.execution': False,
|
|
'mapred.child.java.opts': '-Xmx500m'
|
|
},
|
|
'general': {
|
|
'Enable Swift': False
|
|
}
|
|
},
|
|
'node_groups': [
|
|
{
|
|
'name': 'master-node',
|
|
'flavor_id': cls.flavor_ref,
|
|
'node_processes': ['namenode'],
|
|
'count': 1
|
|
},
|
|
{
|
|
'name': 'worker-node',
|
|
'node_group_template_id': node_group_template_id,
|
|
'count': 3
|
|
}
|
|
]
|
|
}
|
|
# create cls.cluster_template variable to use for comparison to cluster
|
|
# template response body. The 'node_groups' field in the response body
|
|
# has some extra info that post body does not have. The 'node_groups'
|
|
# field in the response body is something like this
|
|
#
|
|
# 'node_groups': [
|
|
# {
|
|
# 'count': 3,
|
|
# 'name': 'worker-node',
|
|
# 'volume_mount_prefix': '/volumes/disk',
|
|
# 'created_at': '2014-05-21 14:31:37',
|
|
# 'updated_at': None,
|
|
# 'floating_ip_pool': None,
|
|
# ...
|
|
# },
|
|
# ...
|
|
# ]
|
|
cls.cluster_template = cls.full_cluster_template.copy()
|
|
del cls.cluster_template['node_groups']
|
|
|
|
def _create_cluster_template(self, template_name=None):
|
|
"""Creates Cluster Template with optional name specified.
|
|
|
|
It creates template, ensures template name and response body.
|
|
Returns id and name of created template.
|
|
"""
|
|
if not template_name:
|
|
# generate random name if it's not specified
|
|
template_name = data_utils.rand_name('sahara-cluster-template')
|
|
|
|
# create cluster template
|
|
resp_body = self.create_cluster_template(template_name,
|
|
**self.full_cluster_template)
|
|
|
|
# ensure that template created successfully
|
|
self.assertEqual(template_name, resp_body['name'])
|
|
self.assertDictContainsSubset(self.cluster_template, resp_body)
|
|
|
|
return resp_body['id'], template_name
|
|
|
|
@test.attr(type='smoke')
|
|
def test_cluster_template_create(self):
|
|
self._create_cluster_template()
|
|
|
|
@test.attr(type='smoke')
|
|
def test_cluster_template_list(self):
|
|
template_info = self._create_cluster_template()
|
|
|
|
# check for cluster template in list
|
|
_, templates = self.client.list_cluster_templates()
|
|
templates_info = [(template['id'], template['name'])
|
|
for template in templates]
|
|
self.assertIn(template_info, templates_info)
|
|
|
|
@test.attr(type='smoke')
|
|
def test_cluster_template_get(self):
|
|
template_id, template_name = self._create_cluster_template()
|
|
|
|
# check cluster template fetch by id
|
|
_, template = self.client.get_cluster_template(template_id)
|
|
self.assertEqual(template_name, template['name'])
|
|
self.assertDictContainsSubset(self.cluster_template, template)
|
|
|
|
@test.attr(type='smoke')
|
|
def test_cluster_template_delete(self):
|
|
template_id, _ = self._create_cluster_template()
|
|
|
|
# delete the cluster template by id
|
|
self.client.delete_cluster_template(template_id)
|
|
# TODO(ylobankov): check that cluster template is really deleted
|