cockpit-auto/run.py at master · kasturinarra/cockpit-auto · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
# !/usr/bin/env python2.7

import sys
import logging
from utils.helpers import results_logs
from utils.helpers import generate_final_results
from utils.helpers import yaml2dict
from importlib import import_module


log = logging.getLogger("sherry")


def main():
    if sys.argv[1] == 'h':
        str = """Main function entry.
    Usage:
        python run.py [argv]

    Options:
        h,help                                          Menu
        debug_tier                                     Test debug_tier
        he_tier                                        Test he_tier
        virt_tier                                      Test virt_tier
        common_tier                                    Test common_tier
        all_tier                                       Test all_tier
    Example:
        python run.py debug_tier
    """
        print str
    else:
        tier = sys.argv[1]

        config_dict = yaml2dict('./config.yml')
        host_string = config_dict['host_string']
        host_user = config_dict['host_user']
        host_pass = config_dict['host_pass']
        browser = config_dict['browser']
        test_build = config_dict['test_build']
        results_logs.test_build = test_build

        test_scens = yaml2dict('./scen.yml')
        try:
            case_files = test_scens[tier]['cases']
            for cf in case_files:
                # Get file name from scenario file
                cf_name = cf.rstrip('.py')
                results_logs.logger_name = 'checkpoints.log'
                results_logs.get_actual_logger(cf_name)

                # Import module by case file name
                case_module = import_module('cases.checks.' + cf_name, __package__)
                testclss_name = ''
                for attr in dir(case_module):
                    if attr.startswith('Test'):
                        testclss_name = attr
                        break
                testclss = getattr(case_module, testclss_name, None)

                # Make the instance of test class
                test = testclss()
                test.host_string = host_string
                test.host_user = host_user
                test.host_pass = host_pass
                test.browser = browser
                test.build = test_build

                # Go check
                log.info(test.go_check(cf_name))
        except Exception as e:
            log.exception(e)

        generate_final_results(results_logs)


if __name__ == '__main__':
    main()