aboutsummaryrefslogtreecommitdiffstats
path: root/testcases/VIM/OpenStack/CI/libraries/run_rally.py
blob: da5e6adca9c2bff25a03f9468affcf74f836eaf8 (plain)
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
#!/usr/bin/env python
#
# Copyright (c) 2015 Orange
# guyrodrigue.koffi@orange.com
# morgan.richomme@orange.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 re, json, os, sys, urllib2
 
def get_task_id(cmd_raw):
    """
    get task id from command rally result
    :param cmd_raw:
    :return: task_id as string
    """
    taskid_re = re.compile('^Task +(.*): started$')
    for line in cmd_raw.splitlines(True):
        line = line.strip()
        match = taskid_re.match(line)
        if match:
            return match.group(1)
    return None
 
def task_succeed(json_raw):
    """
    Parse JSON from rally JSON results
    :param json_raw:
    :return: Bool
    """
    rally_report = json.loads(json_raw)
    rally_report = rally_report[0]
    if rally_report is None:
        return False
    if rally_report.get('result') is None:
        return False
 
    for result in rally_report.get('result'):
        if len(result.get('error')) > 0:
            return False
 
    return True
 
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
    """
 
    """ get the date """
    cmd = os.popen("date '+%d%m%Y_%H%M'")
    test_date = cmd.read().rstrip()
 
    """ check directory for test scenarios files or retrieve from git otherwise"""
    tests_path = "./scenarios"
    test_file_name = '{}/opnfv-{}.json'.format(tests_path, test_name)
    if not os.path.exists(test_file_name):
        retrieve_test_cases_file(test_name, tests_path)
	print "Scenario successfully downloaded"
 
    print "Start test..."
    cmd = os.popen("rally task start --abort-on-sla-failure %s" % test_file_name)
    task_id = get_task_id(cmd.read())
 
    if task_id is None:
        print "./run_rally : failed to retrieve task_id"
        exit(-1)
 
    """ check for result directory and create it otherwise """
    report_path = "./results"
    if not os.path.exists(report_path):
        os.makedirs(report_path)
 
    report_file_name = '{}/opnfv-{}-{}.html'.format(report_path, test_name, test_date)
    
    os.popen("rally task report %s --out %s" % (task_id, report_file_name))
    cmd = os.popen("rally task results %s" % task_id)
    if task_succeed(cmd.read()):
        print "OK"
    else:
        print "KO"
 
 
def retrieve_test_cases_file(test_name, tests_path):
    """
    Retrieve from github the sample test files
    :return: void
    """
 
    """ 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)
    print 'fetching {}/{} ...'.format(url_base, test_file_name)
    response = urllib2.urlopen('{}/{}'.format(url_base, test_file_name))
    file_raw = response.read()
 
    """ check if the test path existe 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 file:
        file.write(file_raw)
 
 
def main():
    """ configure script """
    tests = ['authenticate','glance','heat','keystone','neutron','nova','tempest','vm', 'all'];
 
 
    if len(sys.argv) != 2:
        options = '{d[0]} | {d[1]} | {d[2]} | {d[3]} | {d[4]} | {d[5]} | {d[6]} | {d[7]} | {d[8]}'.format(d=tests)
        print "./run_rally [", options, "]"
        exit(-1)
    test_name = sys.argv[1]
 
    if not (test_name in tests):
        print "argument not valid"
        exit(-1)
 
    if test_name == "all":
        for test_name in tests:
            if not (test_name == 'all' or test_name == 'tempest'):
                print(test_name)
                run_task(test_name)
 
    else:
        run_task(test_name)
 
 
if __name__ == '__main__':
    main()