fuel-qa/run_system_test.py
Dmitry Tyzhnenko 89e2c36d46 Use set in groups iteration
Change-Id: Ia87fc007a89cc8f5255a5c9b185b5ee3e6a72c8a
Closes-bug: #1563260
2016-03-29 13:44:48 +03:00

204 lines
6.8 KiB
Python
Executable File

#!/usr/bin/env python
import sys
import argparse
from proboscis import TestProgram
from proboscis import register
from fuelweb_test.helpers.utils import pretty_log
from gates_tests.helpers.utils import map_test_review_in_fuel_library
from system_test import register_system_test_cases
from system_test import get_groups
from system_test import define_custom_groups
from system_test import discover_import_tests
from system_test import tests_directory
from system_test import collect_yamls
from system_test import get_path_to_config
from system_test import get_list_confignames
from system_test import get_basepath
from system_test.core.repository import split_group_config
basedir = get_basepath()
def print_explain(names):
groups_nums = get_groups()
if not isinstance(names, list):
names = [names]
out = []
for name in [split_group_config(i)[0] if split_group_config(i) else i
for i in names]:
for i in groups_nums[name]:
if hasattr(i, 'home'):
out.append((i.home._proboscis_entry_.parent.home, i.home))
else:
out.append(i)
print(pretty_log(out))
def clean_argv():
"""Removing argv params unused by Proboscis"""
argv = sys.argv
if '--with-config' in argv:
idx = argv.index('--with-config')
argv.pop(idx)
argv.pop(idx)
if '--explain' in argv:
idx = argv.index('--explain')
argv.pop(idx)
return argv
def cli():
cli = argparse.ArgumentParser(prog="System test runner",
description="Command line tool for run Fuel "
"System Test")
commands = cli.add_subparsers(title="Operation commands",
dest="command")
cli_run = commands.add_parser('run',
help="Run test",
description="Run some test group")
cli_run.add_argument("run_groups", nargs='*', default=None, )
cli_run.add_argument("--with-config", default=False, type=str,
action="store", dest="config_name",
help="Select name of yaml config.")
cli_run.add_argument("--explain", default=False, action="store_true",
help="Show explain for running groups. "
"Will not start Proboscis.")
cli_run.add_argument("--show-plan", default=False, action="store_true",
help="Show Proboscis test plan.")
cli_run.add_argument("--with-xunit", default=False, action="store_true",
help="Use xuint report.")
cli_run.add_argument("--nologcapture", default=False, action="store_true",
help="Disable log capture for Proboscis.")
cli_run.add_argument("-q", default=False, action="store_true",
dest="quite",
help="Run Proboscis in quite mode.")
cli_run.add_argument("-a", default=False, action="store_true",
dest="nose_attr",
help="Provide Nose attr to Proboscis.")
cli_run.add_argument("-A", default=False, action="store_true",
dest="eval_nose",
help="Eval Nose attr to Proboscis.")
cli_run.add_argument("--groups", default=None, action="append", type=str,
help="Test group for testing. "
"(backward compatibility)")
cli_explain_group = commands.add_parser("explain-group",
help="Explain selected group.")
cli_explain_group.add_argument("name",
help="Group name.")
commands.add_parser("show-all-groups",
help="Show all Proboscis groups")
commands.add_parser("show-fuelweb-groups",
help="Show Proboscis groups defined in fuelweb suite")
commands.add_parser("show-systest-groups",
help="Show Proboscis groups defined in Systest suite")
commands.add_parser("show-systest-configs",
help="Show configurations for Systest suite")
if len(sys.argv) == 1:
cli.print_help()
sys.exit(1)
return cli.parse_args()
def run(**kwargs):
config_name = kwargs.get('config_name', None)
groups = kwargs.get('run_groups', [])
old_groups = kwargs.get('groups', None)
explain = kwargs.get('explain', None)
groups_to_run = []
groups.extend(old_groups or [])
for g in set(groups):
if config_name:
register_system_test_cases(
groups=[g],
configs=[config_name])
groups_to_run.append("{0}({1})".format(g, config_name))
else:
register_system_test_cases(groups=[g])
groups_to_run.append(g)
if not set([split_group_config(i)[0] if split_group_config(i) else i
for i in groups_to_run]) < set(get_groups()):
sys.exit('There are no cases mapped to current group, '
'please be sure that you put right test group name.')
if explain:
print_explain(groups)
else:
register(groups=["run_system_test"], depends_on_groups=groups_to_run)
TestProgram(groups=['run_system_test'],
argv=clean_argv()).run_and_exit()
def explain_group(**kwargs):
"""Explain selected group."""
name = kwargs.get('name', None)
print_explain(name)
def show_all_groups(**kwargs):
"""Show all Proboscis groups"""
groups_nums = get_groups()
out = {k: len(v) for k, v in groups_nums.items()}
print(pretty_log(out))
def show_fuelweb_groups(**kwargs):
"""Show Proboscis groups defined in fuelweb suite"""
groups_nums = get_groups()
out = {k: len(v) for k, v in groups_nums.items()
if not k.startswith('system_test')}
print(pretty_log(out))
def show_systest_groups(**kwargs):
"""Show Proboscis groups defined in Systest suite"""
groups_nums = get_groups()
out = {k: len(v) for k, v in groups_nums.items()
if k.startswith('system_test')}
print(pretty_log(out))
def show_systest_configs(**kwargs):
"""Show configurations for Systest suite"""
tests_configs = collect_yamls(get_path_to_config())
for c in get_list_confignames(tests_configs):
print(c)
COMMAND_MAP = {
"run": run,
"explain-group": explain_group,
"show-all-groups": show_all_groups,
"show-fuelweb-groups": show_fuelweb_groups,
"show-systest-groups": show_systest_groups,
"show-systest-configs": show_systest_configs
}
def shell():
args = cli()
discover_import_tests(basedir, tests_directory)
define_custom_groups()
map_test_review_in_fuel_library(**vars(args))
COMMAND_MAP[args.command](**vars(args))
if __name__ == '__main__':
shell()