1 # Copyright (c) 2015 Intel Research and Development Ireland Ltd.
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
15 __author__ = 'vmriccox'
18 from keystoneclient.v2_0 import client as keystoneClient
19 from heatclient import client as heatClient
20 from heatclient.common import template_utils
22 from experimental_framework import common
27 def __init__(self, credentials):
28 self.ip_controller = credentials['ip_controller']
29 self.heat_url = credentials['heat_url']
30 self.user = credentials['user']
31 self.password = credentials['password']
32 self.auth_uri = credentials['auth_uri']
33 self.project_id = credentials['project']
36 # TODO: verify that init_heat is useless in the constructor
40 keystone = keystoneClient.Client(username=self.user,
41 password=self.password,
42 tenant_name=self.project_id,
43 auth_url=self.auth_uri)
44 auth_token = keystone.auth_token
45 self.heat = heatClient.Client('1', endpoint=self.heat_url,
48 def print_stacks(self, name=None):
49 for stack in self.heat.stacks.list():
50 if (name and stack.stack_name == name) or not name:
51 common.LOG.info("Stack Name: " + stack.stack_name)
52 common.LOG.info("Stack Status: " + stack.stack_status)
54 def create_stack(self, template_file, stack_name, parameters):
57 tpl_files, template = \
58 template_utils.get_template_contents(template_file)
62 'files': dict(list(tpl_files.items()))
64 self.heat.stacks.create(stack_name=stack_name, files=fields['files'],
65 template=template, parameters=parameters)
66 self.print_stacks(stack_name)
68 def is_stack_deployed(self, stack_name):
70 if stack_name in self.heat.stacks.list():
74 def check_stack_status(self, stack_name):
76 Returns a string representing the status of a stack from Heat
78 :param stack_name: Name of the stack to be checked (type: str)
81 for stack in self.heat.stacks.list():
82 if stack.stack_name == stack_name:
83 return stack.stack_status
86 def validate_heat_template(self, heat_template_file):
88 if not self.heat.stacks.validate(template=open(heat_template_file,
90 raise ValueError('The provided heat template "' +
92 '" is not in the correct format')
94 def delete_stack(self, stack_name):
97 for stack in self.heat.stacks.list():
98 if stack.stack_name == stack_name:
99 self.heat.stacks.delete(stack.id)