diff options
| -rw-r--r-- | moonv4/moon_interface/tests/apitests/plot_json.py | 163 |
1 files changed, 163 insertions, 0 deletions
diff --git a/moonv4/moon_interface/tests/apitests/plot_json.py b/moonv4/moon_interface/tests/apitests/plot_json.py new file mode 100644 index 00000000..61ebb145 --- /dev/null +++ b/moonv4/moon_interface/tests/apitests/plot_json.py @@ -0,0 +1,163 @@ +import argparse +import logging +import json +import plotly +from plotly.graph_objs import Scatter, Layout +import plotly.figure_factory as ff + + +logger = None + + +def init(): + global logger, HOST, PORT + parser = argparse.ArgumentParser() + parser.add_argument("--verbose", "-v", action='store_true', help="verbose mode") + parser.add_argument("--host", + help="Set the name of the host to test (default: 172.18.0.11).", + default="172.18.0.11") + parser.add_argument("--port", "-p", + help="Set the port of the host to test (default: 38001).", + default="38001") + parser.add_argument("--test-only", "-t", action='store_true', dest='testonly', help="Do not generate graphs") + parser.add_argument("--write", "-w", help="Write test data to a JSON file", default="/tmp/data.json") + parser.add_argument("--input", "-i", help="Get data from a JSON input file") + parser.add_argument("--legend", "-l", help="Set the legend (default: 'rbac,rbac+session')", + default='rbac,rbac+session') + parser.add_argument("--distgraph", "-d", + help="Show a distribution graph instead of a linear graph", + action='store_true') + parser.add_argument("--request-per-second", help="Number of requests per seconds", + type=int, dest="request_second", default=1) + parser.add_argument("--limit", help="Limit request to LIMIT", type=int) + parser.add_argument("--write-image", help="Write the graph to file IMAGE", dest="write_image") + parser.add_argument("--write-html", help="Write the graph to HTML file HTML", dest="write_html", default="data.html") + args = parser.parse_args() + + FORMAT = '%(asctime)-15s %(levelname)s %(message)s' + logging.basicConfig( + format=FORMAT, + level=logging.INFO) + + logger = logging.getLogger(__name__) + + HOST = args.host + PORT = args.port + return args + + +def get_delta(time_data): + time_delta = list() + time_delta_sum1 = 0 + for key in time_data: + time_delta.append(time_data[key]['delta']) + time_delta_sum1 += time_data[key]['delta'] + time_delta_average1 = time_delta_sum1 / len(time_data.keys()) + return time_delta, time_delta_average1 + + +def write_graph(legend=None, input=None, image_file=None, html_file=None): + logger.info("Writing graph") + legends = legend.split(",") + result_data = [] + for _input in input.split(","): + current_legend = legends.pop(0) + time_data2 = json.load(open(_input)) + time_delta2, time_delta_average2 = get_delta(time_data2) + for key in time_data2.keys(): + if key in time_data2: + time_delta2.append(time_data2[key]['delta']) + else: + time_delta2.append(None) + data2 = Scatter( + x=list(range(len(time_data2.keys()))), + y=time_delta2, + name=current_legend, + line=dict( + color='rgb(255, 192, 118)', + shape='spline') + ) + result_data.append(data2) + data2_a = Scatter( + x=list(range(len(time_data2.keys()))), + y=[time_delta_average2 for x in range(len(time_data2.keys()))], + name=current_legend + " average", + line=dict( + color='rgb(255, 152, 33)', + shape='spline') + ) + result_data.append(data2_a) + + if image_file: + plotly.offline.plot( + { + "data": result_data, + "layout": Layout( + title="Request times delta", + xaxis=dict(title='Requests'), + yaxis=dict(title='Request duration'), + ) + }, + filename=html_file, + image="svg", + image_filename=image_file, + image_height=1000, + image_width=1200 + ) + else: + plotly.offline.plot( + { + "data": result_data, + "layout": Layout( + title="Request times delta", + xaxis=dict(title='Requests'), + yaxis=dict(title='Request duration'), + ) + }, + filename=html_file, + ) + return 0 + + +def write_distgraph(legend=None, input=None, image_file=None, html_file=None): + + logger.info("Writing graph") + legends = legend.split(",") + result_data = [] + + for _input in input.split(","): + logger.info("Analysing input {}".format(_input)) + time_data2 = json.load(open(_input)) + time_delta2, time_delta_average2 = get_delta(time_data2) + result_data.append(time_delta2) + + # Create distplot with custom bin_size + if len(legends) < len(result_data): + for _cpt in range(len(result_data)-len(legends)): + legends.append("NC") + fig = ff.create_distplot(result_data, legends, bin_size=.2) + + # Plot! + plotly.offline.plot( + fig, + image="svg", + image_filename=image_file, + image_height=1000, + image_width=1200, + filename=html_file + ) + return 0 + + +def main(): + args = init() + if args.distgraph: + write_distgraph(legend=args.legend, input=args.input, image_file=args.write_image, + html_file=args.write_html) + else: + write_graph(legend=args.legend, input=args.input, image_file=args.write_image, + html_file=args.write_html) + + +if __name__ == "__main__": + main() |