yardstick/vTC/apexlake/experimental_framework/deployment_unit.py

   1 # Copyright (c) 2015 Intel Research and Development Ireland Ltd.
   2 #
   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
   6 #
   7 #      http://www.apache.org/licenses/LICENSE-2.0
   8 #
   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.
  14
  15 import os
  16 import time
  17
  18 from experimental_framework import heat_manager
  19 from experimental_framework import common
  20
  21 MAX_RETRY = 3
  22
  23
  24 class DeploymentUnit:
  25     """
  26     This unit is in charge to manage the deployment of the workloads under
  27     test and any other workloads necessary to
  28     the benchmark
  29     """
  30
  31     def __init__(self, openstack_credentials):
  32         self.heat_manager = heat_manager.HeatManager(openstack_credentials)
  33         self.deployed_stacks = list()
  34
  35     def destroy_heat_template(self, stack_name):
  36         """
  37         Destroys a stack
  38         :param stack_name: Stack of the name to be destroyed (sting)
  39         :return: None
  40         """
  41         try:
  42             if self.heat_manager.check_stack_status(stack_name):
  43                 if stack_name in self.deployed_stacks:
  44                     self.deployed_stacks.remove(stack_name)
  45                 self.heat_manager.delete_stack(stack_name)
  46
  47             status = self.heat_manager.check_stack_status(stack_name)
  48             while status and 'DELETE_IN_PROGRESS' in status:
  49                 common.LOG.info(status)
  50                 time.sleep(5)
  51                 status = self.heat_manager.check_stack_status(stack_name)
  52             return True
  53         except:
  54             return False
  55
  56     def destroy_all_deployed_stacks(self):
  57         """
  58         Destroys all the stacks currently deployed
  59         :return: None
  60         """
  61         for stack in self.deployed_stacks:
  62             if self.heat_manager.is_stack_deployed(stack):
  63                 self.destroy_heat_template(stack)
  64
  65     def deploy_heat_template(self, template_file, stack_name, parameters,
  66                              attempt=0):
  67         """
  68         Deploys a heat template and in case of failure retries 3 times
  69         :param template_file: full path file name of the heat template
  70         :param stack_name: name of the stack to deploy
  71         :param parameters: parameters to be given to the heat template
  72         :param attempt: number of current attempt
  73         :return: returns True in case the creation is completed
  74                  returns False in case the creation is failed
  75         """
  76         if not os.path.isfile(template_file):
  77             raise ValueError('The specified file does not exist ("' +
  78                              template_file + '")')
  79         self.heat_manager.validate_heat_template(template_file)
  80         try:
  81             self.heat_manager.create_stack(template_file, stack_name,
  82                                            parameters)
  83             deployed = True
  84         except:
  85             deployed = False
  86
  87         if not deployed and 'COMPLETE' in \
  88                 self.heat_manager.check_stack_status(stack_name):
  89             try:
  90                 self.destroy_heat_template(stack_name)
  91             except:
  92                 pass
  93
  94         status = self.heat_manager.check_stack_status(stack_name)
  95         while status and 'CREATE_IN_PROGRESS' in status:
  96             time.sleep(5)
  97             status = self.heat_manager.check_stack_status(stack_name)
  98         if status and ('FAILED' in status or 'NOT_FOUND' in status):
  99             if attempt < MAX_RETRY:
 100                 attempt += 1
 101                 try:
 102                     self.destroy_heat_template(stack_name)
 103                 except Exception as e:
 104                     common.LOG.debug(e.message)
 105                     pass
 106                 return self.deploy_heat_template(template_file, stack_name,
 107                                                  parameters, attempt)
 108             else:
 109                 try:
 110                     self.destroy_heat_template(stack_name)
 111                 except Exception as e:
 112                     common.LOG.debug(e.message)
 113                 finally:
 114                     return False
 115         if self.heat_manager.check_stack_status(stack_name) and \
 116             'COMPLETE' in self.heat_manager.\
 117                 check_stack_status(stack_name):
 118             self.deployed_stacks.append(stack_name)
 119             return True