X-Git-Url: http://git.lttng.org./?a=blobdiff_plain;f=scripts%2Fsystem-tests%2Flava2-submit.py;h=2fa8145bf0c7cac175c099a0c61c3bb12bcc78f2;hb=056f7519e4772cdd4636d386a5252d9e23866729;hp=5fb3b48837170c8fbf6660fa905c36e52045ed74;hpb=0425e1dda320507d0bedf8ce6f9be807f334079e;p=lttng-ci.git diff --git a/scripts/system-tests/lava2-submit.py b/scripts/system-tests/lava2-submit.py index 5fb3b48..2fa8145 100644 --- a/scripts/system-tests/lava2-submit.py +++ b/scripts/system-tests/lava2-submit.py @@ -1,4 +1,4 @@ -#!/usr/bin/python +#!/usr/bin/python3 # Copyright (C) 2016 - Francis Deslauriers # # This program is free software: you can redistribute it and/or modify @@ -15,35 +15,36 @@ # along with this program. If not, see . import argparse -import base64 import json import os import random import sys import time -import yaml import xmlrpc.client -import pprint - -from jinja2 import Environment, FileSystemLoader, meta +from urllib.parse import urljoin +from urllib.request import urlretrieve +import yaml +from jinja2 import Environment, FileSystemLoader USERNAME = 'lava-jenkins' HOSTNAME = 'lava-master-02.internal.efficios.com' OBJSTORE_URL = "https://obj.internal.efficios.com/lava/results/" class TestType(): - baremetal_benchmarks=1 - baremetal_tests=2 - kvm_tests=3 - kvm_fuzzing_tests=4 + """ Enum like for test type """ + baremetal_benchmarks = 1 + baremetal_tests = 2 + kvm_tests = 3 + kvm_fuzzing_tests = 4 values = { 'baremetal-benchmarks' : baremetal_benchmarks, 'baremetal-tests' : baremetal_tests, 'kvm-tests' : kvm_tests, - 'kvm-fuzzin-tests' : kvm_fuzzing_tests, + 'kvm-fuzzing-tests' : kvm_fuzzing_tests, } class DeviceType(): + """ Enum like for device type """ x86 = 'x86' kvm = 'qemu' values = { @@ -55,21 +56,23 @@ def get_job_bundle_content(server, job): try: bundle_sha = server.scheduler.job_status(str(job))['bundle_sha1'] bundle = server.dashboard.get(bundle_sha) - except xmlrpc.client.Fault as f: - print('Error while fetching results bundle', f.faultString) - raise f + except xmlrpc.client.Fault as error: + print('Error while fetching results bundle', error.faultString) + raise error return json.loads(bundle['content']) -# Parse the results bundle to see the run-tests testcase -# of the lttng-kernel-tests passed successfully def check_job_all_test_cases_state_count(server, job): + """ + Parse the results bundle to see the run-tests testcase + of the lttng-kernel-tests passed successfully + """ print("Testcase result:") content = server.results.get_testjob_results_yaml(str(job)) testcases = yaml.load(content) - passed_tests=0 - failed_tests=0 + passed_tests = 0 + failed_tests = 0 for testcase in testcases: if testcase['result'] != 'pass': print("\tFAILED {}\n\t\t See http://{}{}".format( @@ -77,27 +80,32 @@ def check_job_all_test_cases_state_count(server, job): HOSTNAME, testcase['url'] )) - failed_tests+=1 + failed_tests += 1 else: - passed_tests+=1 + passed_tests += 1 return (passed_tests, failed_tests) -# Get the benchmark results from the objstore -# save them as CSV files localy def fetch_benchmark_results(build_id): + """ + Get the benchmark results from the objstore + save them as CSV files localy + """ testcases = ['processed_results_close.csv', - 'processed_results_ioctl.csv', - 'processed_results_open_efault.csv', - 'processed_results_open_enoent.csv', - 'processed_results_dup_close.csv', - 'processed_results_raw_syscall_getpid.csv', - 'processed_results_lttng_test_filter.csv'] + 'processed_results_ioctl.csv', + 'processed_results_open_efault.csv', + 'processed_results_open_enoent.csv', + 'processed_results_dup_close.csv', + 'processed_results_raw_syscall_getpid.csv', + 'processed_results_lttng_test_filter.csv'] for testcase in testcases: url = urljoin(OBJSTORE_URL, "{:s}/{:s}".format(build_id, testcase)) + print('Fetching {}'.format(url)) urlretrieve(url, testcase) -# Parse the attachment of the testcase to fetch the stdout of the test suite def print_test_output(server, job): + """ + Parse the attachment of the testcase to fetch the stdout of the test suite + """ job_finished, log = server.scheduler.jobs.logs(str(job)) logs = yaml.load(log.data.decode('ascii')) print_line = False @@ -114,7 +122,10 @@ def print_test_output(server, job): if print_line: print("{} {}".format(line['dt'], line['msg'])) -def get_vlttng_cmd(device, lttng_tools_commit, lttng_ust_commit=None): +def get_vlttng_cmd(lttng_tools_commit, lttng_ust_commit=None): + """ + Return vlttng cmd to be used in the job template for setup. + """ vlttng_cmd = 'vlttng --jobs=$(nproc) --profile urcu-master' \ ' --override projects.babeltrace.build-env.PYTHON=python3' \ @@ -130,17 +141,14 @@ def get_vlttng_cmd(device, lttng_tools_commit, lttng_ust_commit=None): ' --override projects.lttng-ust.checkout='+lttng_ust_commit+ \ ' --profile lttng-ust-no-man-pages' - if device is DeviceType.kvm: - vlttng_path = '/root/virtenv' - else: - vlttng_path = '/tmp/virtenv' + vlttng_path = '/tmp/virtenv' vlttng_cmd += ' ' + vlttng_path return vlttng_cmd def main(): - nfsrootfs = "https://obj.internal.efficios.com/lava/rootfs/rootfs_amd64_bionic_2018-11-29.tar.gz" + nfsrootfs = "https://obj.internal.efficios.com/lava/rootfs/rootfs_amd64_xenial_2018-12-05.tar.gz" test_type = None parser = argparse.ArgumentParser(description='Launch baremetal test using Lava') parser.add_argument('-t', '--type', required=True) @@ -164,29 +172,26 @@ def main(): if not args.debug: try: lava_api_key = os.environ['LAVA2_JENKINS_TOKEN'] - except Exception as e: - print('LAVA2_JENKINS_TOKEN not found in the environment variable. Exiting...', e ) + except Exception as error: + print('LAVA2_JENKINS_TOKEN not found in the environment variable. Exiting...', + error) return -1 jinja_loader = FileSystemLoader(os.path.dirname(os.path.realpath(__file__))) jinja_env = Environment(loader=jinja_loader, trim_blocks=True, - lstrip_blocks= True) + lstrip_blocks=True) jinja_template = jinja_env.get_template('template_lava_job.jinja2') - template_source = jinja_env.loader.get_source(jinja_env, 'template_lava_job.jinja2') - parsed_content = jinja_env.parse(template_source) - undef = meta.find_undeclared_variables(parsed_content) test_type = TestType.values[args.type] if test_type in [TestType.baremetal_benchmarks, TestType.baremetal_tests]: device_type = DeviceType.x86 - vlttng_path = '/tmp/virtenv' - else: device_type = DeviceType.kvm - vlttng_path = '/root/virtenv' - vlttng_cmd = get_vlttng_cmd(device_type, args.tools_commit, args.ust_commit) + vlttng_path = '/tmp/virtenv' + + vlttng_cmd = get_vlttng_cmd(args.tools_commit, args.ust_commit) context = dict() context['DeviceType'] = DeviceType @@ -218,7 +223,16 @@ def main(): server = xmlrpc.client.ServerProxy('http://%s:%s@%s/RPC2' % (USERNAME, lava_api_key, HOSTNAME)) - jobid = server.scheduler.submit_job(render) + for attempt in range(10): + try: + jobid = server.scheduler.submit_job(render) + except xmlrpc.client.ProtocolError as error: + print('Protocol error on submit, sleeping and retrying. Attempt #{}' + .format(attempt)) + time.sleep(5) + continue + else: + break print('Lava jobid:{}'.format(jobid)) print('Lava job URL: http://lava-master-02.internal.efficios.com/scheduler/job/{}'.format(jobid)) @@ -226,12 +240,16 @@ def main(): #Check the status of the job every 30 seconds jobstatus = server.scheduler.job_state(jobid)['job_state'] running = False - while jobstatus in ['Submitted','Scheduling','Scheduled','Running']: + while jobstatus in ['Submitted', 'Scheduling', 'Scheduled', 'Running']: if not running and jobstatus == 'Running': print('Job started running') running = True time.sleep(30) - jobstatus = server.scheduler.job_state(jobid)['job_state'] + try: + jobstatus = server.scheduler.job_state(jobid)['job_state'] + except xmlrpc.client.ProtocolError as error: + print('Protocol error, retrying') + continue print('Job ended with {} status.'.format(jobstatus)) if jobstatus != 'Finished': @@ -242,13 +260,13 @@ def main(): elif test_type is TestType.baremetal_benchmarks: fetch_benchmark_results(args.build_id) - passed, failed=check_job_all_test_cases_state_count(server, jobid) + passed, failed = check_job_all_test_cases_state_count(server, jobid) print('With {} passed and {} failed Lava test cases.'.format(passed, failed)) - if failed == 0: - return 0 - else: + if failed != 0: return -1 + return 0 + if __name__ == "__main__": sys.exit(main())