testsuites/posca/testcase_script/posca_factor_cpu_burden.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 common_script
  17 import datetime
  18 import subprocess
  19
  20 # ------------------------------------------------------
  21 # parser for configuration files in each test case
  22 # ------------------------------------------------------
  23 parser = argparse.ArgumentParser()
  24 parser.add_argument("-c", "--conf",
  25                     help="configuration files for the testcase,\
  26                         in yaml format",
  27                     default="/home/opnfv/bottlenecks/testsuites/posca\
  28 /testcase_cfg/posca_factor_system_bandwidth.yaml")
  29 args = parser.parse_args()
  30 headers = {"Content-Type": "application/json"}
  31 INTERPRETER = "/usr/bin/python"
  32
  33
  34 # --------------------------------------------------
  35 # logging configuration
  36 # --------------------------------------------------
  37 logger = logging.getLogger(__name__)
  38
  39
  40 def posca_env_check():
  41     print("========== posca system bandwidth env check ===========")
  42     filepath = r"/home/opnfv/bottlenecks/testsuites/posca/test_result/"
  43     if os.path.exists(filepath):
  44         return True
  45     else:
  46         os.mkdir(r'/home/opnfv/bottlenecks/testsuites/posca/test_result/')
  47
  48
  49 def system_cpu_burden(test_id, data, file_config, con_dic):
  50     date_id = test_id
  51     print("test is is begin from %d" % test_id)
  52     cur_role_result = 1
  53     pre_role_result = 1
  54     pre_reply = {}
  55     data_return = {}
  56     data_max = {}
  57     data_return["throughput"] = 1
  58     for test_x in data["tx_pkt_sizes"]:
  59         data_max["throughput"] = 1
  60         for test_y in data["rx_pkt_sizes"]:
  61             test_config = {
  62                         "tx_msg_size": float(test_x),
  63                         "rx_msg_size": float(test_y),
  64                     }
  65             date_id = date_id + 1
  66             file_config["test_id"] = date_id
  67             data_reply = common_script.posca_send_data(
  68                    con_dic, test_config, file_config)
  69             bandwidth = data_reply["throughput"]
  70             if (data_max["remote_cpu_util"] > con_dic["cpu_load"]):
  71                 return 1, data_reply
  72             if (data_max["local_cpu_util"] > con_dic["cpu_load"]):
  73                 return 1, data_reply
  74     print("cpu_burden don't find\n")
  75     return 0, data_return
  76
  77
  78 def posca_run(con_dic):
  79     print("========== run posca system bandwidth ===========")
  80     test_con_id = 0
  81     file_config = {}
  82     data = {}
  83     rx_pkt_s_a = con_dic['rx_pkt_sizes'].split(',')
  84     tx_pkt_s_a = con_dic['tx_pkt_sizes'].split(',')
  85     time_new = time.strftime('%H_%M', time.localtime(time.time()))
  86     file_config["file_path"] = "/home/opnfv/bottlenecks/testsuites/posca/\
  87 test_result/factor_system_system_bandwidth_%s.json" % (time_new)
  88     file_config["test_type"] = "system_bandwidth_biggest"
  89     data["rx_pkt_sizes"] = rx_pkt_s_a
  90     data["tx_pkt_sizes"] = tx_pkt_s_a
  91     print("######test package begin######")
  92     date_return, pkt_reply = system_cpu_burden(
  93             test_con_id, data, file_config, con_dic)
  94
  95     return True
  96
  97
  98 def main():
  99     if not (args.conf):
 100         logger.error("Configuration files do not exist for \
 101                     the specified testcases")
 102         os.exit(-1)
 103     else:
 104         testcase_cfg = args.conf
 105
 106     con_str = [
 107             'test_ip', 'tool', 'test_time', 'protocol',
 108             'tx_pkt_sizes', 'rx_pkt_sizes', 'cpu_load',
 109             'latency', 'ES_ip', 'dashboard'
 110     ]
 111     posca_env_check()
 112     starttime = datetime.datetime.now()
 113     config = ConfigParser.ConfigParser()
 114     con_dic = common_script.posca_config_read(testcase_cfg, con_str, config)
 115     common_script.posca_create_incluxdb(con_dic)
 116     posca_run(con_dic)
 117     endtime = datetime.datetime.now()
 118     if con_dic["dashboard"] == "y":
 119         cmd = '/home/opnfv/bottlenecks/testsuites/posca/testcase_dashboard/\
 120 system_bandwidth.py'
 121         pargs = [INTERPRETER, cmd]
 122         print("\nBegin to establish dashboard.")
 123         sub_result = subprocess.Popen(pargs)
 124         sub_result.wait()
 125     print("System Bandwidth testing time : %s" %(endtime - starttime))
 126     time.sleep(5)
 127
 128 if __name__ == '__main__':
 129     main()