Check if rally image already exist and don't create it again if so
[functest.git] / testcases / VIM / OpenStack / CI / libraries / run_rally.py
index 1f63b1b..a7f1db1 100644 (file)
@@ -8,36 +8,49 @@
 # which accompanies this distribution, and is available at
 # http://www.apache.org/licenses/LICENSE-2.0
 #
-import re, json, os, urllib2, argparse, logging, yaml
-
-with open('../functest.yaml') as f:
-    functest_yaml = yaml.safe_load(f)
-f.close()
+# 0.1 (05/2015) initial commit
+# 0.2 (28/09/2015) extract Tempest, format json result, add ceilometer suite
+# 0.3 (19/10/2015) remove Tempest from run_rally
+# and push result into test DB
+#
 
-HOME = os.environ['HOME']+"/"
-SCENARIOS_DIR = HOME + functest_yaml.get("general").get("directories").get("dir_rally_scn")
-RESULTS_DIR = HOME + functest_yaml.get("general").get("directories").get("dir_rally_res")
+import re
+import json
+import os
+import argparse
+import logging
+import yaml
+import requests
+import subprocess
+import sys
+from novaclient import client as novaclient
+from keystoneclient.v2_0 import client as keystoneclient
+from glanceclient import client as glanceclient
 
 """ tests configuration """
-tests = ['authenticate', 'glance', 'cinder', 'heat', 'keystone', 'neutron', 'nova', 'quotas', 'requests', 'tempest', 'vm', 'all', 'smoke']
+tests = ['authenticate', 'glance', 'cinder', 'ceilometer', 'heat', 'keystone',
+         'neutron', 'nova', 'quotas', 'requests', 'vm', 'all']
 parser = argparse.ArgumentParser()
-parser.add_argument("test_name", help="The name of the test you want to perform with rally. "
-                                      "Possible values are : "
-                                      "[ {d[0]} | {d[1]} | {d[2]} | {d[3]} | {d[4]} | {d[5]} | {d[6]} "
-                                      "| {d[7]} | {d[8]} | {d[9]} | {d[10]} | {d[11]} | {d[12]}]. The 'all' value performs all the tests scenarios "
-                                      "except 'tempest'".format(d=tests))
+parser.add_argument("test_name",
+                    help="Module name to be tested"
+                         "Possible values are : "
+                         "[ {d[0]} | {d[1]} | {d[2]} | {d[3]} | {d[4]} | "
+                         "{d[5]} | {d[6]} | {d[7]} | {d[8]} | {d[9]} | "
+                         "{d[10]} | {d[11]}]. The 'all' value "
+                         "performs all the  possible tests scenarios"
+                         .format(d=tests))
 
 parser.add_argument("-d", "--debug", help="Debug mode",  action="store_true")
-parser.add_argument("test_mode", help="Tempest test mode", nargs='?', default="smoke")
+parser.add_argument("-r", "--report",
+                    help="Create json result file",
+                    action="store_true")
+
 args = parser.parse_args()
-test_mode=args.test_mode
 
-if not args.test_name == "tempest":
-    if not args.test_mode == "smoke":
-        parser.error("test_mode is only used with tempest")
 
+FNULL = open(os.devnull, 'w')
 """ logging configuration """
-logger = logging.getLogger('run_rally')
+logger = logging.getLogger("run_rally")
 logger.setLevel(logging.DEBUG)
 
 ch = logging.StreamHandler()
@@ -46,25 +59,53 @@ if args.debug:
 else:
     ch.setLevel(logging.INFO)
 
-formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+formatter = logging.Formatter("%(asctime)s - %(name)s - "
+                              "%(levelname)s - %(message)s")
 ch.setFormatter(formatter)
 logger.addHandler(ch)
 
+REPO_PATH=os.environ['repos_dir']+'/functest/'
+if not os.path.exists(REPO_PATH):
+    logger.error("Functest repository directory not found '%s'" % REPO_PATH)
+    exit(-1)
+sys.path.append(REPO_PATH + "testcases/")
+import functest_utils
 
-def get_tempest_id(cmd_raw):
-    """
-    get task id from command rally result
-    :param cmd_raw:
-    :return: task_id as string
-    """
-    taskid_re = re.compile('^Verification UUID: (.*)$')
-    for line in cmd_raw.splitlines(True):
-        line = line.strip()
-    match = taskid_re.match(line)
+with open("/home/opnfv/functest/conf/config_functest.yaml") as f:
+    functest_yaml = yaml.safe_load(f)
+f.close()
+
+HOME = os.environ['HOME']+"/"
+SCENARIOS_DIR = REPO_PATH + functest_yaml.get("general"). \
+    get("directories").get("dir_rally_scn")
+RESULTS_DIR = functest_yaml.get("general").get("directories"). \
+    get("dir_rally_res")
+TEST_DB = functest_yaml.get("results").get("test_db_url")
+
+GLANCE_IMAGE_NAME = "functest-img-rally"
+GLANCE_IMAGE_FILENAME = functest_yaml.get("general"). \
+    get("openstack").get("image_file_name")
+GLANCE_IMAGE_FORMAT = functest_yaml.get("general"). \
+    get("openstack").get("image_disk_format")
+GLANCE_IMAGE_PATH = functest_yaml.get("general"). \
+    get("directories").get("dir_functest_data") + "/" + GLANCE_IMAGE_FILENAME
+
+
+def push_results_to_db(payload):
+
+    url = TEST_DB + "/results"
+    installer = functest_utils.get_installer_type(logger)
+    git_version = functest_utils.get_git_branch(REPO_PATH)
+    pod_name = functest_utils.get_pod_name(logger)
+    # TODO pod_name hardcoded, info shall come from Jenkins
+    params = {"project_name": "functest", "case_name": "Rally",
+              "pod_name": pod_name, "installer": installer,
+              "version": git_version, "details": payload}
+
+    headers = {'Content-Type': 'application/json'}
+    r = requests.post(url, data=json.dumps(params), headers=headers)
+    logger.debug(r)
 
-    if match:
-        return match.group(1)
-    return None
 
 def get_task_id(cmd_raw):
     """
@@ -100,92 +141,73 @@ def task_succeed(json_raw):
 
     return True
 
-def run_tempest():
-    """
-    the function dedicated to Tempest (functional tests for OpenStack)
-    :param test_mode: Tempest mode smoke (default), full, ..
-    :return: void
-    """
-    logger.info('starting {} Tempest ...'.format(test_mode))
-
-    """ get the date """
-    cmd = os.popen("date '+%d%m%Y_%H%M'")
-    test_date = cmd.read().rstrip()
-
-    cmd_line = "rally verify start {}".format(test_mode)
-    logger.debug('running command line : {}'.format(cmd_line))
-    cmd = os.popen(cmd_line)
-    task_id = get_tempest_id(cmd.read())
-    logger.debug('task_id : {}'.format(task_id))
-
-    if task_id is None:
-        logger.error("failed to retrieve task_id")
-    exit(-1)
-
-    """ check for result directory and create it otherwise """
-    if not os.path.exists(RESULTS_DIR):
-        logger.debug('does not exists, we create it'.format(RESULTS_DIR))
-        os.makedirs(RESULTS_DIR)
-
-    """ write log report file """
-    report_file_name = '{}opnfv-tempest-{}.log'.format(RESULTS_DIR, test_date)
-    cmd_line = "rally verify detailed {} > {} ".format(task_id, report_file_name)
-    logger.debug('running command line : {}'.format(cmd_line))
-    os.popen(cmd_line)
-
 
 def run_task(test_name):
-    """
-    the "main" function of the script who lunch rally for a task
-    :param test_name: name for the rally test
-    :return: void
-    """
-    logger.info('starting {} test ...'.format(test_name))
+    #
+    # the "main" function of the script who lunch rally for a task
+    # :param test_name: name for the rally test
+    # :return: void
+    #
 
-    """ get the date """
-    cmd = os.popen("date '+%d%m%Y_%H%M'")
-    test_date = cmd.read().rstrip()
+    logger.info('starting {} test ...'.format(test_name))
 
-    """ check directory for scenarios test files or retrieve from git otherwise"""
+    # check directory for scenarios test files or retrieve from git otherwise
     proceed_test = True
     test_file_name = '{}opnfv-{}.json'.format(SCENARIOS_DIR, test_name)
+
     if not os.path.exists(test_file_name):
-        logger.debug('{} does not exists'.format(test_file_name))
-        proceed_test = retrieve_test_cases_file(test_name, SCENARIOS_DIR)
+        logger.error("The scenario '%s' does not exist." % test_file_name)
+        exit(-1)
 
-    """ we do the test only if we have a scenario test file """
+    # we do the test only if we have a scenario test file
     if proceed_test:
         logger.debug('Scenario fetched from : {}'.format(test_file_name))
-        cmd_line = "rally task start --abort-on-sla-failure %s" % test_file_name
+        cmd_line = "rally task start --abort-on-sla-failure {}".format(test_file_name)
         logger.debug('running command line : {}'.format(cmd_line))
-        cmd = os.popen(cmd_line)
-        task_id = get_task_id(cmd.read())
+        p = subprocess.Popen(cmd_line, stdout=subprocess.PIPE, stderr=FNULL, shell=True)
+        result = ""
+        while p.poll() is None:
+            l = p.stdout.readline()
+            print l.replace('\n', '')
+            result += l
+
+        task_id = get_task_id(result)
         logger.debug('task_id : {}'.format(task_id))
 
         if task_id is None:
             logger.error("failed to retrieve task_id")
             exit(-1)
 
-        """ check for result directory and create it otherwise """
+        # check for result directory and create it otherwise
         if not os.path.exists(RESULTS_DIR):
             logger.debug('does not exists, we create it'.format(RESULTS_DIR))
             os.makedirs(RESULTS_DIR)
 
-        """ write html report file """
-        report_file_name = '{}opnfv-{}-{}.html'.format(RESULTS_DIR, test_name, test_date)
-        cmd_line = "rally task report %s --out %s" % (task_id, report_file_name)
+        # write html report file
+        report_file_name = '{}opnfv-{}.html'.format(RESULTS_DIR, test_name)
+        cmd_line = "rally task report {} --out {}".format(task_id,
+                                                          report_file_name)
+
         logger.debug('running command line : {}'.format(cmd_line))
         os.popen(cmd_line)
 
-        """ get and save rally operation JSON result """
+        # get and save rally operation JSON result
         cmd_line = "rally task results %s" % task_id
         logger.debug('running command line : {}'.format(cmd_line))
         cmd = os.popen(cmd_line)
         json_results = cmd.read()
-        with open('{}opnfv-{}-{}.json'.format(RESULTS_DIR, test_name, test_date), 'w') as f:
+        with open('{}opnfv-{}.json'.format(RESULTS_DIR, test_name), 'w') as f:
             logger.debug('saving json file')
             f.write(json_results)
-            logger.debug('saving json file2')
+
+        with open('{}opnfv-{}.json'
+                  .format(RESULTS_DIR, test_name)) as json_file:
+            json_data = json.load(json_file)
+
+        # Push results in payload of testcase
+        if args.report:
+            logger.debug("Push result into DB")
+            push_results_to_db(json_data)
 
         """ parse JSON operation result """
         if task_succeed(json_results):
@@ -193,53 +215,61 @@ def run_task(test_name):
         else:
             print 'Test KO'
     else:
-        logger.error('{} test failed, unable to fetch a scenario test file'.format(test_name))
-
-
-def retrieve_test_cases_file(test_name, tests_path):
-    """
-    Retrieve from github the sample test files
-    :return: Boolean that indicates the retrieval status
-    """
-
-    """ do not add the "/" at the end """
-    url_base = "https://git.opnfv.org/cgit/functest/plain/testcases/VIM/OpenStack/CI/suites"
-
-    test_file_name = 'opnfv-{}.json'.format(test_name)
-    logger.info('fetching {}/{} ...'.format(url_base, test_file_name))
-
-    try:
-        response = urllib2.urlopen('{}/{}'.format(url_base, test_file_name))
-    except (urllib2.HTTPError, urllib2.URLError):
-        return False
-    file_raw = response.read()
-
-    """ check if the test path exist otherwise we create it """
-    if not os.path.exists(tests_path):
-        os.makedirs(tests_path)
-
-    with open('{}/{}'.format(tests_path, test_file_name), 'w') as f:
-        f.write(file_raw)
-    return True
+        logger.error('{} test failed, unable to fetch a scenario test file'
+                     .format(test_name))
 
 
 def main():
-    """ configure script """
+    # configure script
     if not (args.test_name in tests):
         logger.error('argument not valid')
         exit(-1)
 
+    creds_nova = functest_utils.get_credentials("nova")
+    nova_client = novaclient.Client('2',**creds_nova)
+    creds_keystone = functest_utils.get_credentials("keystone")
+    keystone_client = keystoneclient.Client(**creds_keystone)
+    glance_endpoint = keystone_client.service_catalog.url_for(service_type='image',
+                                                   endpoint_type='publicURL')
+    glance_client = glanceclient.Client(1, glance_endpoint,
+                                        token=keystone_client.auth_token)
+
+
+    image_id = functest_utils.get_image_id(glance_client, GLANCE_IMAGE_NAME)
+
+    if image_id == '':
+        logger.debug("Creating image '%s' from '%s'..." % (GLANCE_IMAGE_NAME, \
+                                                           GLANCE_IMAGE_PATH))
+        image_id = functest_utils.create_glance_image(glance_client,\
+                                                GLANCE_IMAGE_NAME,GLANCE_IMAGE_PATH)
+        if not image_id:
+            logger.error("Failed to create the Glance image...")
+            exit(-1)
+        else:
+            logger.debug("Image '%s' with ID '%s' created succesfully ." \
+                         % (GLANCE_IMAGE_NAME, image_id))
+    else:
+        logger.debug("Using existing image '%s' with ID '%s'..." \
+                     % (GLANCE_IMAGE_NAME,image_id))
+
+
     if args.test_name == "all":
         for test_name in tests:
-            if not (test_name == 'all' or test_name == 'tempest'):
+            if not (test_name == 'all' or
+                    test_name == 'heat' or
+                    test_name == 'ceilometer' or
+                    test_name == 'smoke' or
+                    test_name == 'vm'):
                 print(test_name)
                 run_task(test_name)
     else:
         print(args.test_name)
-        if args.test_name == 'tempest':
-            run_tempest()
-        else:
-            run_task(args.test_name)
+        run_task(args.test_name)
+
+    logger.debug("Deleting image '%s' with ID '%s'..." \
+                         % (GLANCE_IMAGE_NAME, image_id))
+    if not functest_utils.delete_glance_image(nova_client, image_id):
+        logger.error("Error deleting the glance image")
 
 if __name__ == '__main__':
     main()