3 # Copyright (c) 2015 Orange
4 # guyrodrigue.koffi@orange.com
5 # morgan.richomme@orange.com
6 # All rights reserved. This program and the accompanying materials
7 # are made available under the terms of the Apache License, Version 2.0
8 # which accompanies this distribution, and is available at
9 # http://www.apache.org/licenses/LICENSE-2.0
11 # 0.1 (05/2015) initial commit
12 # 0.2 (28/09/2015) extract Tempest, format json result, add ceilometer suite
13 # 0.3 (19/10/2015) remove Tempest from run_rally
14 # and push result into test DB
25 import novaclient.v2.client as novaclient
27 """ tests configuration """
28 tests = ['authenticate', 'glance', 'cinder', 'heat', 'keystone',
29 'neutron', 'nova', 'quotas', 'requests', 'vm', 'all']
30 parser = argparse.ArgumentParser()
31 parser.add_argument("repo_path", help="Path to the repository")
32 parser.add_argument("test_name",
33 help="Module name to be tested"
34 "Possible values are : "
35 "[ {d[0]} | {d[1]} | {d[2]} | {d[3]} | {d[4]} | "
36 "{d[5]} | {d[6]} | {d[7]}] "
38 "performs all the possible tests scenarios"
41 parser.add_argument("-d", "--debug", help="Debug mode", action="store_true")
42 parser.add_argument("-r", "--report",
43 help="Create json result file",
46 args = parser.parse_args()
48 sys.path.append(args.repo_path + "testcases/")
51 """ logging configuration """
52 logger = logging.getLogger("run_rally")
53 logger.setLevel(logging.DEBUG)
55 ch = logging.StreamHandler()
57 ch.setLevel(logging.DEBUG)
59 ch.setLevel(logging.INFO)
61 formatter = logging.Formatter("%(asctime)s - %(name)s - "
62 "%(levelname)s - %(message)s")
63 ch.setFormatter(formatter)
66 with open(args.repo_path+"testcases/config_functest.yaml") as f:
67 functest_yaml = yaml.safe_load(f)
70 HOME = os.environ['HOME']+"/"
71 REPO_PATH = args.repo_path
73 #SCENARIOS_DIR = REPO_PATH + functest_yaml.get("general"). \
74 # get("directories").get("dir_rally_scn")
75 SCENARIOS_DIR = REPO_PATH + "testcases/VIM/OpenStack/CI/rally_cert/"
77 TEMPLATE_DIR = SCENARIOS_DIR + "scenario/templates"
78 SUPPORT_DIR = SCENARIOS_DIR + "scenario/support"
80 FLAVOR_NAME = "m1.tiny"
83 CONTROLLERS_AMOUNT = 2
85 RESULTS_DIR = functest_yaml.get("general").get("directories"). \
87 TEST_DB = functest_yaml.get("results").get("test_db_url")
88 FLOATING_NETWORK = functest_yaml.get("general"). \
89 get("openstack").get("neutron_public_net_name")
91 GLANCE_IMAGE_NAME = functest_yaml.get("general"). \
92 get("openstack").get("image_name")
93 GLANCE_IMAGE_FILENAME = functest_yaml.get("general"). \
94 get("openstack").get("image_file_name")
95 GLANCE_IMAGE_FORMAT = functest_yaml.get("general"). \
96 get("openstack").get("image_disk_format")
97 GLANCE_IMAGE_PATH = functest_yaml.get("general"). \
98 get("directories").get("dir_functest_data") + "/" + GLANCE_IMAGE_FILENAME
99 GLANCE_IMAGE_LOCATION = "http://download.cirros-cloud.net/0.3.4/cirros-0.3.4-x86_64-disk.img"
102 def push_results_to_db(payload):
104 url = TEST_DB + "/results"
105 installer = functest_utils.get_installer_type(logger)
106 git_version = functest_utils.get_git_branch(args.repo_path)
107 pod_name = functest_utils.get_pod_name(logger)
108 # TODO pod_name hardcoded, info shall come from Jenkins
109 params = {"project_name": "functest", "case_name": "Rally",
110 "pod_name": pod_name, "installer": installer,
111 "version": git_version, "details": payload}
113 headers = {'Content-Type': 'application/json'}
114 r = requests.post(url, data=json.dumps(params), headers=headers)
118 def get_task_id(cmd_raw):
120 get task id from command rally result
122 :return: task_id as string
124 taskid_re = re.compile('^Task +(.*): started$')
125 for line in cmd_raw.splitlines(True):
127 match = taskid_re.match(line)
129 return match.group(1)
133 def create_glance_image(path, name, disk_format):
135 Create a glance image given the absolute path of the image, its name and the disk format
137 cmd = ("glance image-create --name " + name + " --visibility public "
138 "--disk-format " + disk_format + " --container-format bare --file " + path)
139 functest_utils.execute_command(cmd, logger)
143 def task_succeed(json_raw):
145 Parse JSON from rally JSON results
149 rally_report = json.loads(json_raw)
150 rally_report = rally_report[0]
151 if rally_report is None:
153 if rally_report.get('result') is None:
156 for result in rally_report.get('result'):
157 if len(result.get('error')) > 0:
163 def build_task_args(test_file_name):
164 task_args = {'service_list': [test_file_name]}
165 task_args['smoke'] = False
166 task_args['image_name'] = GLANCE_IMAGE_NAME
167 task_args['flavor_name'] = FLAVOR_NAME
168 task_args['glance_image_location'] = GLANCE_IMAGE_LOCATION
169 task_args['floating_network'] = FLOATING_NETWORK
170 task_args['tmpl_dir'] = TEMPLATE_DIR
171 task_args['sup_dir'] = SUPPORT_DIR
172 task_args['users_amount'] = USERS_AMOUNT
173 task_args['tenants_amount'] = TENANTS_AMOUNT
174 task_args['controllers_amount'] = CONTROLLERS_AMOUNT
179 def run_task(test_name):
181 # the "main" function of the script who launch rally for a task
182 # :param test_name: name for the rally test
186 logger.info('starting {} test ...'.format(test_name))
188 task_file = '{}task.yaml'.format(SCENARIOS_DIR)
189 if not os.path.exists(task_file):
190 logger.error("Task file '%s' does not exist." % task_file)
193 test_file_name = '{}opnfv-{}.yaml'.format(SCENARIOS_DIR + "scenario/", test_name)
194 if not os.path.exists(test_file_name):
195 logger.error("The scenario '%s' does not exist." % test_file_name)
198 logger.debug('Scenario fetched from : {}'.format(test_file_name))
200 cmd_line = "rally task start --abort-on-sla-failure " + \
201 "--task {} ".format(task_file) + \
202 "--task-args \"{}\" ".format(build_task_args(test_name))
203 logger.debug('running command line : {}'.format(cmd_line))
204 cmd = os.popen(cmd_line)
205 task_id = get_task_id(cmd.read())
206 logger.debug('task_id : {}'.format(task_id))
209 logger.error("failed to retrieve task_id")
212 # check for result directory and create it otherwise
213 if not os.path.exists(RESULTS_DIR):
214 logger.debug('does not exists, we create it'.format(RESULTS_DIR))
215 os.makedirs(RESULTS_DIR)
217 # write html report file
218 report_file_name = '{}opnfv-{}.html'.format(RESULTS_DIR, test_name)
219 cmd_line = "rally task report {} --out {}".format(task_id,
222 logger.debug('running command line : {}'.format(cmd_line))
225 # get and save rally operation JSON result
226 cmd_line = "rally task results %s" % task_id
227 logger.debug('running command line : {}'.format(cmd_line))
228 cmd = os.popen(cmd_line)
229 json_results = cmd.read()
230 with open('{}opnfv-{}.json'.format(RESULTS_DIR, test_name), 'w') as f:
231 logger.debug('saving json file')
232 f.write(json_results)
234 with open('{}opnfv-{}.json'
235 .format(RESULTS_DIR, test_name)) as json_file:
236 json_data = json.load(json_file)
238 # Push results in payload of testcase
240 logger.debug("Push result into DB")
241 push_results_to_db(json_data)
243 """ parse JSON operation result """
244 if task_succeed(json_results):
250 def delete_glance_image(name):
251 cmd = ("glance image-delete $(glance image-list | grep %s "
252 "| awk '{print $2}' | head -1)" % name)
253 functest_utils.execute_command(cmd, logger)
258 logger.info("Cleaning up...")
259 logger.debug("Deleting image...")
260 delete_glance_image(GLANCE_IMAGE_NAME)
266 if not (args.test_name in tests):
267 logger.error('argument not valid')
270 creds_nova = functest_utils.get_credentials("nova")
271 nova_client = novaclient.Client(**creds_nova)
273 logger.debug("Creating image '%s' from '%s'..." % (GLANCE_IMAGE_NAME, GLANCE_IMAGE_PATH))
274 create_glance_image(GLANCE_IMAGE_PATH, GLANCE_IMAGE_NAME, GLANCE_IMAGE_FORMAT)
277 # Check if the given image exists
279 nova_client.images.find(name=GLANCE_IMAGE_NAME)
280 logger.info("Glance image found '%s'" % GLANCE_IMAGE_NAME)
282 logger.error("ERROR: Glance image '%s' not found." % GLANCE_IMAGE_NAME)
283 logger.info("Available images are: ")
286 if args.test_name == "all":
287 for test_name in tests:
288 if not (test_name == 'all' or
289 test_name == 'heat' or
290 test_name == 'smoke' or
295 print(args.test_name)
296 run_task(args.test_name)
300 if __name__ == '__main__':