testsuites/posca/testcase_script/posca_factor_rx_pkt_size.py

   1 #!/usr/bin/env python
   2 ##############################################################################
   3 # Copyright (c) 2016 Huawei Technologies Co.,Ltd and others.
   4 #
   5 # All rights reserved. This program and the accompanying materials
   6 # are made available under the terms of the Apache License, Version 2.0
   7 # which accompanies this distribution, and is available at
   8 # http://www.apache.org/licenses/LICENSE-2.0
   9 ##############################################################################
  10
  11 import os
  12 import argparse
  13 import time
  14 import logging
  15 import ConfigParser
  16 import json
  17
  18 #------------------------------------------------------
  19 # parser for configuration files in each test case
  20 # ------------------------------------------------------
  21 parser = argparse.ArgumentParser()
  22 parser.add_argument("-c", "--conf",
  23                     help="configuration files for the testcase, in yaml format",
  24                     default="/home/opnfv/bottlenecks/testsuites/posca/testcase_cfg/posca_factor_rx_pkt_size.yaml")
  25 args = parser.parse_args()
  26
  27 cmd="curl -i"
  28 order_arg="-H \"Content-Type: application/json\" -X POST -d \'{\"cmd\": \"start\", \"opts\":{\"output-file\": \"/tem/yardstick.out\"}, \"args\": \"../samples/netperf.yaml\"}'"
  29
  30 #--------------------------------------------------
  31 # logging configuration
  32 #--------------------------------------------------
  33 logger = logging.getLogger(__name__)
  34
  35 def posca_env_check():
  36     print "========== posca system bandwidth env check ==========="
  37     if os.path.exists(r'/home/opnfv/bottlenecks/testsuites/posca/test_result/'):
  38         return True
  39     else:
  40         os.mkdirs(r'/home/opnfv/bottlenecks/testsuites/posca/test_result/')
  41
  42 def posca_output_result(time_new,input_1,input_2,input_3,input_4,input_5,input_6):
  43     save_dic={}
  44     save_dic['tx_pkt_size']=input_1
  45     save_dic['rx_cache_size']=input_2
  46     save_dic['tx_cache_size']=input_3
  47     save_dic['throughput']=input_4
  48     save_dic['latency']=input_5
  49     save_dic['cpu_load']=input_6
  50     with open("/home/opnfv/bottlenecks/testsuites/posca/test_result/factor_rx_pkt_size_%s.json"%(time_new),"a") as f:
  51         f.write(json.dumps(save_dic,f))
  52         f.write("\n")
  53
  54 def posca_config_read(config_str):
  55     print "========== posca system bandwidth config read ==========="
  56
  57     con_dic = {}
  58     config = ConfigParser.ConfigParser()
  59     with open(config_str,"rd") as cfgfile:
  60         config.readfp(cfgfile)
  61         con_dic['test_ip']=config.get("config","test_ip")
  62         con_dic['test_tool']=config.get("config","tool")
  63         con_dic['test_time']=config.get("config","test_time")
  64         con_dic['test_protocol']=config.get("config","protocol")
  65         con_dic['test_tx_pkt_s']=config.get("config","tx pkt sizes")
  66         con_dic['test_rx_pkt_s']=config.get("config","rx pkt sizes")
  67         con_dic['test_tx_cache_s']=config.get("config","tx cache sizes")
  68         con_dic['test_rx_cache_s']=config.get("config","rx cache sizes")
  69         con_dic['test_cpu_load']=config.get("config","cpu load")
  70         con_dic['test_latency']=config.get("config","latency")
  71
  72     return con_dic
  73
  74 def posca_run(con_dic):
  75     print "========== run posca system bandwidth ==========="
  76
  77     test_tx_pkt_s_a = con_dic['test_tx_pkt_s'].split(',')
  78     test_tx_cache_s_a = con_dic['test_tx_cache_s'].split(',')
  79     test_rx_cache_s_a = con_dic['test_rx_cache_s'].split(',')
  80     time_new = time.strftime('%H_%M',time.localtime(time.time()))
  81     bandwidth_tmp = 1
  82
  83     for test_rx_cache_s_e in test_rx_cache_s_a:
  84         for test_tx_cache_s_e in test_tx_cache_s_a:
  85             for test_tx_pkt_s_e in test_tx_pkt_s_a:
  86                 print "%s,%s,%s"%(test_tx_pkt_s_e,test_tx_cache_s_e,test_rx_cache_s_e)
  87                 order_excute = os.popen("%s %s http://%s/api/v3/yardstick/tasks/task %s %s %s"%(cmd,order_arg,con_dic['test_ip'],test_tx_pkt_s_e,test_rx_cache_s_e,test_tx_cache_s_e))
  88                 order_result = order_excute.read()
  89                 task_id = order_result.find("task_id")
  90                 time.sleep(con_dic['test_time'])
  91                 cmd_excute = os.popen( "%s http://%s/api/v3/yardstick/testresults?task_id=%s"%(cmd,con_dic['test_ip'],task_id))
  92                 test_result = cmd_excute.read()
  93                 bandwidth = test_result.find("bandwidth")
  94                 cpu_load = test_result.find("cpu_load")
  95                 latency = test_result.find("latency")
  96                 posca_output_result(time_new,test_tx_pkt_s_e,test_rx_cache_s_e,test_tx_cache_s_e,bandwidth,latency,cpu_load)
  97                 if (abs(bandwidth-con_dic['test_throughput'])/con_dic['test_throughput'] >0.05) and (latency < con_dic['test_latency']) and (cpu_load < con_dic['test_cpu_load']):
  98                     if (abs(bandwidth_tmp-bandwidth)/bandwidth <0.05):
  99                         print "%s,%s,%s,%s,%s,%s"%(test_tx_pkt_s_e,test_rx_cache_s_e,test_tx_cache_s_e,bandwidth,latency,cpu_load)
 100                         return True
 101                     else:
 102                         bandwidth_tmp = bandwidth
 103                 else:
 104                     print "%s,%s,%s,%s,%s,%s"%(test_tx_pkt_s_e,test_rx_cache_s_e,test_tx_cache_s_e,bandwidth,latency,cpu_load)
 105                     return False
 106
 107
 108 def main():
 109     if not (args.conf):
 110        logger.error("Configuration files do not exist for the specified testcases")
 111        exit(-1)
 112     else:
 113        testcase_cfg = args.conf
 114
 115     con_dic=posca_config_read(testcase_cfg)
 116     posca_env_check()
 117     posca_run(con_dic)
 118
 119     time.sleep(5)
 120
 121 if __name__=='__main__':
 122     main()