testcases/vIMS/CI/vIMS.py

   1 #!/usr/bin/python
   2 # coding: utf8
   3 #######################################################################
   4 #
   5 #   Copyright (c) 2015 Orange
   6 #   valentin.boucher@orange.com
   7 #
   8 # All rights reserved. This program and the accompanying materials
   9 # are made available under the terms of the Apache License, Version 2.0
  10 # which accompanies this distribution, and is available at
  11 # http://www.apache.org/licenses/LICENSE-2.0
  12 ########################################################################
  13
  14 import os, time, subprocess, logging, argparse, yaml, pprint, sys, shutil
  15 from git import Repo
  16 import keystoneclient.v2_0.client as ksclient
  17 import glanceclient.client as glclient
  18 import novaclient.client as nvclient
  19 from neutronclient.v2_0 import client as ntclient
  20
  21 import urllib
  22 pp = pprint.PrettyPrinter(indent=4)
  23
  24
  25 parser = argparse.ArgumentParser()
  26 parser.add_argument("repo_path", help="Path to the repository")
  27 parser.add_argument("-d", "--debug", help="Debug mode",  action="store_true")
  28 args = parser.parse_args()
  29
  30 sys.path.append(args.repo_path + "testcases/")
  31
  32 import functest_utils
  33
  34 """ logging configuration """
  35 logger = logging.getLogger('vIMS')
  36 logger.setLevel(logging.DEBUG)
  37
  38 ch = logging.StreamHandler()
  39 if args.debug:
  40     ch.setLevel(logging.DEBUG)
  41 else:
  42     ch.setLevel(logging.INFO)
  43 formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
  44 ch.setFormatter(formatter)
  45 logger.addHandler(ch)
  46
  47
  48 HOME = os.environ['HOME']+"/"
  49 # with open(args.repo_path+"config_functest.yaml") as f:
  50 with open(args.repo_path + "testcases/config_functest.yaml") as f:
  51     functest_yaml = yaml.safe_load(f)
  52 f.close()
  53
  54 # Cloudify parameters
  55 REPO_PATH = args.repo_path
  56 HOME = os.environ['HOME']+"/"
  57 VIMS_DIR = REPO_PATH + functest_yaml.get("general").get("directories").get("dir_vIMS")
  58 VIMS_DATA_DIR = HOME + functest_yaml.get("general").get("directories").get("dir_vIMS_data")
  59
  60 TENANT_NAME = functest_yaml.get("vIMS").get("general").get("tenant_name")
  61 TENANT_DESCRIPTION = functest_yaml.get("vIMS").get("general").get("tenant_description")
  62 BASE_IMAGE_URL = functest_yaml.get("vIMS").get("general").get("base_image_url")
  63 BASE_IMAGE_NAME = functest_yaml.get("vIMS").get("general").get("base_image_name")
  64 GLANCE_IMAGE_NAME = functest_yaml.get("vIMS").get("cloudify").get("inputs").get("image_id")
  65
  66 CFY_MANAGER_BLUEPRINT = functest_yaml.get("vIMS").get("cloudify").get("blueprint")
  67 CFY_INPUTS =  functest_yaml.get("vIMS").get("cloudify").get("inputs")
  68 CFY_INPUTS_PATH =  functest_yaml.get("vIMS").get("cloudify").get("inputs_path")
  69
  70 CW_BLUEPRINT = functest_yaml.get("vIMS").get("clearwater").get("blueprint")
  71 CW_DEPLOYMENT_NAME = functest_yaml.get("vIMS").get("clearwater").get("deployment-name")
  72 CW_INPUTS =  functest_yaml.get("vIMS").get("clearwater").get("inputs")
  73 CW_DOMAIN_NAME =  functest_yaml.get("vIMS").get("clearwater").get("inputs").get("public_domain")
  74
  75
  76 def pMsg(value):
  77     """pretty printing"""
  78     pp.pprint(value)
  79
  80 def download_and_add_image_on_glance(glance, image_name, image_url):
  81     dest_path = VIMS_DATA_DIR + "tmp/"
  82     file_name = image_url.rsplit('/')[-1]
  83     if not functest_utils.download_url(image_url, dest_path):
  84         logger.error("Failed to download image %s" %file_name)
  85         return False
  86
  87     image = functest_utils.create_glance_image(glance, image_name, dest_path + file_name)
  88     if not image:
  89         logger.error("Failed to upload image on glance")
  90         return False
  91
  92     return image
  93
  94 def download_blueprints(blueprint_url, branch, dest_path):
  95     if os.path.exists(dest_path):
  96         shutil.rmtree(dest_path)
  97     try:
  98         Repo.clone_from(blueprint_url, dest_path, branch=branch)
  99         return True
 100     except:
 101         return False
 102
 103 def initialize_deployments():
 104     if not os.path.exists(VIMS_DATA_DIR):
 105         os.makedirs(VIMS_DATA_DIR)
 106
 107     ks_creds = functest_utils.get_credentials("keystone")
 108     nv_creds = functest_utils.get_credentials("nova")
 109     nt_creds = functest_utils.get_credentials("neutron")
 110
 111     logger.info("Prepare OpenStack plateform (create tenant and user)")
 112     keystone = ksclient.Client(**ks_creds)
 113
 114     user_id = functest_utils.get_user_id(keystone, ks_creds['username'])
 115     if user_id == '':
 116         logger.error("Error : Failed to get id of %s user" %ks_creds['username'])
 117         exit(-1)
 118
 119     tenant_id = functest_utils.create_tenant(keystone, TENANT_NAME, TENANT_DESCRIPTION)
 120     if tenant_id == '':
 121         logger.error("Error : Failed to create %s tenant" %TENANT_NAME)
 122         exit(-1)
 123
 124     role_name = "admin"
 125     role_id = functest_utils.get_role_id(keystone, role_name)
 126     if role_id == '':
 127         logger.error("Error : Failed to get id for %s role" %role_name)
 128
 129     if not functest_utils.add_role_user(keystone, user_id, role_id, tenant_id):
 130         logger.error("Error : Failed to add %s on tenant" %ks_creds['username'])
 131
 132     user_id = functest_utils.create_user(keystone, TENANT_NAME, TENANT_NAME, None, tenant_id)
 133     if user_id == '':
 134         logger.error("Error : Failed to create %s user" %TENANT_NAME)
 135
 136     logger.info("Update OpenStack creds informations")
 137     ks_creds.update({
 138         "username": TENANT_NAME,
 139         "password": TENANT_NAME,
 140         "tenant_name": TENANT_NAME,
 141         })
 142
 143     nt_creds.update({
 144         "tenant_name": TENANT_NAME,
 145         })
 146
 147     nv_creds.update({
 148         "project_id": TENANT_NAME,
 149         })
 150
 151     logger.info("Upload ubuntu image if it doesn't exist")
 152     glance_endpoint = keystone.service_catalog.url_for(service_type='image',
 153                                                    endpoint_type='publicURL')
 154     glance = glclient.Client(1, glance_endpoint, token=keystone.auth_token)
 155
 156     image_id = functest_utils.get_image_id(glance, BASE_IMAGE_NAME)
 157     if image_id == '':
 158         logger.info("""%s image doesn't exist on glance repository.
 159                         Try downloading this image and upload on glance !""" %BASE_IMAGE_NAME)
 160         image_id = download_and_add_image_on_glance(glance, BASE_IMAGE_NAME, BASE_IMAGE_URL)
 161
 162     if image_id == '':
 163         logger.error("Error : Failed to find or upload required OS image for this deployment" %flavor_name)
 164         exit(-1)
 165
 166     logger.info("Collect flavor id for cloudify and clearwater VMs")
 167     nova = nvclient.Client("2", **nv_creds)
 168
 169     flavor_name = "m1.small"
 170     flavor_id = functest_utils.get_flavor_id(nova, flavor_name)
 171     if flavor_id == '':
 172         logger.error("Failed to find %s flavor. Try with ram range requirement !" %flavor_name)
 173         flavor_id = get_flavor_id_by_ram_range(nova, 1792, 2048)
 174
 175     if flavor_id == '':
 176         logger.error("Failed to find required flavor for this deployment" %flavor_name)
 177         exit(-1)
 178
 179     logger.info("Update security group quota for this tenant")
 180     neutron = ntclient.Client(**nt_creds)
 181     if not functest_utils.update_sg_quota(neutron, tenant_id, 50, 100):
 182         logger.error("Failed to update security group quota for tenant %s" %TENANT_NAME)
 183         exit(-1)
 184
 185     ext_net = functest_utils.get_external_net(neutron)
 186     if not ext_net:
 187         logger.error("Failed to get external network")
 188         exit(-1)
 189
 190     logger.info("Update inputs informations")
 191     CFY_INPUTS['image_id'] = image_id
 192     CFY_INPUTS['flavor_id'] = flavor_id
 193     CFY_INPUTS['external_network_name'] = ext_net
 194
 195     CW_INPUTS['image_id'] = image_id
 196     CW_INPUTS['flavor_id'] = flavor_id
 197     CW_INPUTS['external_network_name'] = ext_net
 198
 199     CFY_INPUTS['keystone_username'] = ks_creds['username']
 200     CFY_INPUTS['keystone_password'] = ks_creds['password']
 201     CFY_INPUTS['keystone_url'] = ks_creds['auth_url']
 202     CFY_INPUTS['keystone_tenant_name'] = ks_creds['tenant_name']
 203
 204     logger.info("Prepare virtualenv for cloudify-cli")
 205     cmd = "chmod +x " + VIMS_DIR + "create_venv.sh"
 206     functest_utils.execute_command(cmd,logger)
 207     cmd = VIMS_DIR + "create_venv.sh " + VIMS_DATA_DIR
 208     functest_utils.execute_command(cmd,logger)
 209
 210 def cleanup_deployments():
 211     ks_creds = functest_utils.get_credentials("keystone")
 212
 213     keystone = ksclient.Client(**ks_creds)
 214
 215     logger.info("Removing %s tenant .." %CFY_INPUTS['keystone_tenant_name'])
 216     tenant_id = functest_utils.get_tenant_id(keystone, CFY_INPUTS['keystone_tenant_name'])
 217     if tenant_id == '':
 218         logger.error("Error : Failed to get id of %s tenant" %CFY_INPUTS['keystone_tenant_name'])
 219     else:
 220         if not functest_utils.delete_tenant(keystone, tenant_id):
 221             logger.error("Error : Failed to remove %s tenant" %CFY_INPUTS['keystone_tenant_name'])
 222
 223     logger.info("Removing %s user .." %CFY_INPUTS['keystone_username'])
 224     user_id = functest_utils.get_user_id(keystone, CFY_INPUTS['keystone_username'])
 225     if user_id == '':
 226         logger.error("Error : Failed to get id of %s user" %CFY_INPUTS['keystone_username'])
 227     else:
 228         if not functest_utils.delete_user(keystone, user_id):
 229             logger.error("Error : Failed to remove %s user" %CFY_INPUTS['keystone_username'])
 230
 231 def deploy_cloudify_manager():
 232
 233     logger.info("Downloading the cloudify manager server blueprint")
 234     download_result = download_blueprints(CFY_MANAGER_BLUEPRINT['url'],
 235                                                         CFY_MANAGER_BLUEPRINT['branch'],
 236                                                         VIMS_DATA_DIR + 'cloudify-manager-blueprint/')
 237
 238     if not download_result:
 239         logger.error("Failed to download manager blueprint")
 240         exit(-1)
 241
 242     logger.info("Writing the inputs file")
 243     with open( VIMS_DATA_DIR + 'cloudify-manager-blueprint/' + CFY_INPUTS_PATH, "w") as f:
 244         f.write(yaml.dump(CFY_INPUTS, default_style='"') )
 245     f.close()
 246
 247     logger.info("Launching the cloudify-manager deployment")
 248     script = "source " + VIMS_DATA_DIR + "venv_cloudify/bin/activate; "
 249     script += "cd " + VIMS_DATA_DIR + "; "
 250     script += "cfy init -r; "
 251     script += "cd cloudify-manager-blueprint/openstack; "
 252     script += "cfy local create-requirements -o requirements.txt -p openstack-manager-blueprint.yaml; "
 253     script += "pip install -r requirements.txt; "
 254     script += "cfy bootstrap --install-plugins -p openstack-manager-blueprint.yaml -i inputs.yaml; "
 255     cmd = "/bin/bash -c '" + script + "'"
 256     functest_utils.execute_command(cmd, logger)
 257
 258     logger.info("Cloudify-manager server is UP !")
 259
 260 def undeploy_cloudify_manager():
 261
 262     logger.info("Launching the cloudify-manager undeployment")
 263     script = "source " + VIMS_DATA_DIR + "venv_cloudify/bin/activate; "
 264     script += "cd " + VIMS_DATA_DIR + "; "
 265     script += "cfy teardown -f; "
 266     cmd = "/bin/bash -c '" + script + "'"
 267     functest_utils.execute_command(cmd, logger)
 268
 269     logger.info("Cloudify-manager server has been successfully removed!")
 270
 271 def deploy_clearwater():
 272
 273     logger.info("Downloading the {0} blueprint".format(CW_BLUEPRINT['file_name']))
 274     download_result = download_blueprints(CW_BLUEPRINT['url'], CW_BLUEPRINT['branch'],
 275                                               VIMS_DATA_DIR + CW_BLUEPRINT['destination_folder'])
 276
 277     if not download_result:
 278         logger.error("Failed to download blueprint {0}".format(CW_BLUEPRINT['file_name']))
 279         exit(-1)
 280
 281     logger.info("Writing the inputs file")
 282     with open(VIMS_DATA_DIR + CW_BLUEPRINT['destination_folder'] + "/inputs.yaml", "w") as f:
 283         f.write(yaml.dump(CW_INPUTS, default_style='"') )
 284     f.close()
 285
 286     logger.info("Launching the {0} deployment".format(CW_BLUEPRINT['name']))
 287     script = "source " + VIMS_DATA_DIR + "venv_cloudify/bin/activate; "
 288     script += "cd " + VIMS_DATA_DIR + CW_BLUEPRINT['destination_folder'] + "; "
 289     script += "cfy blueprints upload -b " + CW_BLUEPRINT['name'] + " -p openstack-blueprint.yaml; "
 290     script += "cfy deployments create -b " + CW_BLUEPRINT['name'] + " -d " + CW_DEPLOYMENT_NAME + " --inputs inputs.yaml; "
 291     script += "cfy executions start -w install -d " + CW_DEPLOYMENT_NAME + "; "
 292
 293     cmd = "/bin/bash -c '" + script + "'"
 294     functest_utils.execute_command(cmd, logger)
 295
 296     logger.info("Clearwater vIMS is UP !")
 297
 298 def undeploy_clearwater():
 299
 300     logger.info("Launching the {0} undeployment".format(CW_BLUEPRINT['name']))
 301     script = "source " + VIMS_DATA_DIR + "venv_cloudify/bin/activate; "
 302     script += "cd " + VIMS_DATA_DIR + "; "
 303     script += "cfy executions start -w uninstall -d " + CW_DEPLOYMENT_NAME + "; "
 304     script += "cfy deployments delete -d " + CW_DEPLOYMENT_NAME + "; "
 305
 306     cmd = "/bin/bash -c '" + script + "'"
 307     functest_utils.execute_command(cmd, logger)
 308
 309 def test_clearwater():
 310
 311     script = "source " + VIMS_DATA_DIR + "venv_cloudify/bin/activate; "
 312     script += "cd " + VIMS_DATA_DIR + "; "
 313     script += "cfy deployments outputs -d clearwater-opnfv | grep Value: | sed \"s/ *Value: //g\";"
 314     cmd = "/bin/bash -c '" + script + "'"
 315     dns_ip = os.popen(cmd).read()
 316     dns_ip = dns_ip.splitlines()[0]
 317
 318     # Coming soon
 319
 320 def main():
 321     initialize_deployments()
 322     deploy_cloudify_manager()
 323     deploy_clearwater()
 324
 325     #test_clearwater()
 326
 327     undeploy_clearwater()
 328     undeploy_cloudify_manager()
 329     cleanup_deployments()
 330
 331 if __name__ == '__main__':
 332     main()