#!/usr/bin/python -u # # Author: Jose Lausuch (jose.lausuch@ericsson.com) # # All rights reserved. This program and the accompanying materials # are made available under the terms of the Apache License, Version 2.0 # which accompanies this distribution, and is available at # http://www.apache.org/licenses/LICENSE-2.0 # import datetime import importlib import os import re import sys import argparse import functest.ci.generate_report as generate_report import functest.ci.tier_builder as tb import functest.core.testcase_base as testcase_base import functest.utils.functest_logger as ft_logger import functest.utils.functest_utils as ft_utils import functest.utils.functest_constants as ft_constants import functest.utils.openstack_clean as os_clean import functest.utils.openstack_snapshot as os_snapshot import functest.utils.openstack_utils as os_utils parser = argparse.ArgumentParser() parser.add_argument("-t", "--test", dest="test", action='store', help="Test case or tier (group of tests) to be executed. " "It will run all the test if not specified.") parser.add_argument("-n", "--noclean", help="Do not clean OpenStack resources" " after running each test (default=false).", action="store_true") parser.add_argument("-r", "--report", help="Push results to database " "(default=false).", action="store_true") args = parser.parse_args() """ logging configuration """ logger = ft_logger.Logger("run_tests").getLogger() """ global variables """ EXEC_SCRIPT = ("%s/functest/ci/exec_test.sh" % ft_constants.FUNCTEST_REPO_DIR) # This will be the return code of this script. If any of the tests fails, # this variable will change to -1 class GlobalVariables: EXECUTED_TEST_CASES = [] OVERALL_RESULT = 0 CLEAN_FLAG = True REPORT_FLAG = False def print_separator(str, count=45): line = "" for i in range(0, count - 1): line += str logger.info("%s" % line) def source_rc_file(): rc_file = ft_constants.OPENSTACK_CREDS if not os.path.isfile(rc_file): logger.error("RC file %s does not exist..." % rc_file) sys.exit(1) logger.debug("Sourcing the OpenStack RC file...") creds = os_utils.source_credentials(rc_file) for key, value in creds.iteritems(): if re.search("OS_", key): if key == 'OS_AUTH_URL': ft_constants.OS_AUTH_URL = value elif key == 'OS_USERNAME': ft_constants.OS_USERNAME = value elif key == 'OS_TENANT_NAME': ft_constants.OS_TENANT_NAME = value elif key == 'OS_PASSWORD': ft_constants.OS_PASSWORD = value logger.debug("OS_AUTH_URL:%s" % ft_constants.OS_AUTH_URL) logger.debug("OS_USERNAME:%s" % ft_constants.OS_USERNAME) logger.debug("OS_TENANT_NAME:%s" % ft_constants.OS_TENANT_NAME) logger.debug("OS_PASSWORD:%s" % ft_constants.OS_PASSWORD) def generate_os_snapshot(): os_snapshot.main() def cleanup(): os_clean.main() def update_test_info(test_name, result, duration): for test in GlobalVariables.EXECUTED_TEST_CASES: if test['test_name'] == test_name: test.update({"result": result, "duration": duration}) def get_run_dict_if_defined(testname): try: dict = ft_utils.get_dict_by_test(testname) if not dict: logger.error("Cannot get {}'s config options".format(testname)) elif 'run' in dict: return dict['run'] return None except Exception: logger.exception("Cannot get {}'s config options".format(testname)) return None def run_test(test, tier_name): result_str = "PASS" start = datetime.datetime.now() test_name = test.get_name() logger.info("\n") # blank line print_separator("=") logger.info("Running test case '%s'..." % test_name) print_separator("=") logger.debug("\n%s" % test) source_rc_file() if GlobalVariables.CLEAN_FLAG: generate_os_snapshot() flags = (" -t %s" % (test_name)) if GlobalVariables.REPORT_FLAG: flags += " -r" result = testcase_base.TestcaseBase.EX_RUN_ERROR run_dict = get_run_dict_if_defined(test_name) if run_dict: try: module = importlib.import_module(run_dict['module']) cls = getattr(module, run_dict['class']) test_case = cls() result = test_case.run() if result == testcase_base.TestcaseBase.EX_OK: if GlobalVariables.REPORT_FLAG: test_case.push_to_db() result = test_case.check_criteria() except ImportError: logger.exception("Cannot import module {}".format( run_dict['module'])) except AttributeError: logger.exception("Cannot get class {}".format( run_dict['class'])) else: cmd = ("%s%s" % (EXEC_SCRIPT, flags)) logger.info("Executing command {} because {} " "doesn't implement the new framework".format( cmd, test_name)) result = ft_utils.execute_command(cmd) if GlobalVariables.CLEAN_FLAG: cleanup() end = datetime.datetime.now() duration = (end - start).seconds duration_str = ("%02d:%02d" % divmod(duration, 60)) logger.info("Test execution time: %s" % duration_str) if result != 0: logger.error("The test case '%s' failed. " % test_name) OVERALL_RESULT = -1 result_str = "FAIL" if test.is_blocking(): if not args.test or args.test == "all":