#!/usr/bin/python # ############################################################################## # Copyright (c) 2015 Huawei Technologies Co.,Ltd and other. # # 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 ############################################################################## # # This script is used to build dashboard ready json results # It may be used for all the test case of the Bottlenecks project # a new method format__for_dashboard(results) # v0.1: basic example with methods for Rubbos. # import os import requests import json def get_bottlenecks_cases(): """ get the list of the supported test cases TODO: update the list when adding a new test case for the dashboard """ return ["rubbos", "tu1", "tu3"] def check_bottlenecks_case_exist(case): """ check if the testcase exists if the test case is not defined or not declared in the list return False """ bottlenecks_cases = get_bottlenecks_cases() if case is None or case not in bottlenecks_cases: return False else: return True def format_bottlenecks_for_dashboard(case, results): """ generic method calling the method corresponding to the test case check that the testcase is properly declared first then build the call to the specific method """ if check_bottlenecks_case_exist(case): cmd = "format_" + case + "_for_dashboard(results)" res = eval(cmd) else: res = [] print "Test cases not declared" return res def format_rubbos_for_dashboard(results): """ Post processing for the Rubbos test case """ test_data = [{'description': 'Rubbos results'}] # Graph 1: # ******************************** new_element = [] for each_result in results: throughput_data = [record['throughput'] for record in each_result['details']] new_element.append({'x': each_result['creation_date'], 'y': max(throughput_data)}) test_data.append({'name': "Rubbos max throughput", 'info': {'type': "graph", 'xlabel': 'time', 'ylabel': 'maximal throughput'}, 'data_set': new_element}) return test_data def format_rubbos_probe_for_dashboard(results): """ Post processing for the Rubbos test case of one time """ test_data = [{'description': 'Rubbos results'}] element = [] latest_result = results[-1]["details"] for key in sorted(lastest_result): throughput = latest_result[key]["throughput"] client_num = int(key) element.append({'x': client_num, 'y': throughput}) #graph test_data.append({'name': "Rubbos throughput vs client number", 'info': {'type': "graph", 'xlabel': 'client number', 'ylabel': 'throughput'}, 'data_set': element}) return test_data def format_tu1_for_dashboard(results): test_data = [{'description': 'Tu-1 performance result'}] line_element = [] bar_element = {} last_result = results[-1]["details"] for key in sorted(last_result): bandwith = last_result[key]["Bandwidth"] pktsize = int(key) line_element.append({'x': pktsize, 'y': bandwith * 1000}) bar_element[key] = bandwith * 1000 # graph1, line test_data.append({'name': "VM2VM max single directional throughput", 'info': {'type': "graph", 'xlabel': 'pktsize', 'ylabel': 'bandwith(kpps)'}, 'data_set': line_element}) # graph2, bar test_data.append({'name': "VM2VM max single directional throughput", 'info': {"type": "bar"}, 'data_set': bar_element}) return test_data def format_tu3_for_dashboard(results): test_data = [{'description': 'Tu-3 performance result'}] new_element = [] bar_element = {} last_result = results[-1]["details"] for key in sorted(last_result): bandwith = last_result[key]["Bandwidth"] pktsize = int(key) new_element.append({'x': pktsize, 'y': bandwith * 1000}) bar_element[key] = bandwith * 1000 # graph1, line test_data.append({'name': "VM2VM max bidirectional throughput", 'info': {'type': "graph", 'xlabel': 'pktsize', 'ylabel': 'bandwith(kpps)'}, 'data_set': new_element}) # graph2, bar test_data.append({'name': "VM2VM max single directional throughput", 'info': {"type": "bar"}, 'data_set': bar_element}) return test_data ############################ For local test ################################ def _read_sample_output(filename): curr_path = os.path.dirname(os.path.abspath(__file__)) output = os.path.join(curr_path, filename) with open(output) as f: sample_output = f.read() result = json.loads(sample_output) return result # Copy form functest/testcases/Dashboard/dashboard_utils.py # and did some minor modification for local test. def _get_results(db_url, test_criteria): test_project = test_criteria["project"] testcase = test_criteria["testcase"] # Build headers headers = {'Content-Type': 'application/json'} # build the request # if criteria is all => remove criteria url = db_url + "/results?project=" + test_project + "&case=" + testcase # Send Request to Test DB myData = requests.get(url, headers=headers) # Get result as a json object myNewData = json.loads(myData.text) # Get results myDataResults = myNewData['test_results'] return myDataResults def _test(): db_url = "http://213.77.62.197" results = _get_results(db_url, {"project": "bottlenecks", "testcase": "rubbos"}) test_result = format_rubbos_probe_for_dashboard(results) print json.dumps(test_result, indent=4) results = _get_results(db_url, {"project": "bottlenecks", "testcase": "tu1"}) #results = _read_sample_output("sample") #print json.dumps(results, indent=4) test_result = format_tu1_for_dashboard(results) print json.dumps(test_result, indent=4) results = _get_results(db_url, {"project": "bottlenecks", "testcase": "tu3"}) test_result = format_tu3_for_dashboard(results) print json.dumps(test_result, indent=4) if __name__ == '__main__': _test()