Modify POSCA code into PEP8 style
[bottlenecks.git] / testsuites / posca / testcase_script / posca_factor_rx_cache_size.py
index e049ad2..cfbd278 100644 (file)
@@ -15,103 +15,123 @@ import logging
 import ConfigParser
 import json
 
-#------------------------------------------------------
+# ------------------------------------------------------
 # parser for configuration files in each test case
 # ------------------------------------------------------
 parser = argparse.ArgumentParser()
 parser.add_argument("-c", "--conf",
-                    help="configuration files for the testcase, in yaml format",
-                    default="/home/opnfv/bottlenecks/testsuites/posca/testcase_cfg/posca_factor_rx_cache_size.yaml")
+                    help="configuration files for the testcase,\
+                        in yaml format",
+                    default="/home/opnfv/bottlenecks/testsuites/posca/\
+                        testcase_cfg/posca_factor_tx_pkt_size.yaml")
 args = parser.parse_args()
 
-cmd="curl -i"
-order_arg="-H \"Content-Type: application/json\" -X POST -d \'{\"cmd\": \"start\", \"opts\":{\"output-file\": \"/tem/yardstick.out\"}, \"args\": \"../samples/netperf.yaml\"}'"
+cmd = "curl -i"
+order_arg = "-H \"Content-Type: application/json\" -X POST -d \'{\"cmd\": \
+            \"start\", \"opts\":{\"output-file\": \"/tem/yardstick.out\"}, \
+            \"args\": \"../samples/netperf.yaml\"}'"
 
-#--------------------------------------------------
+# --------------------------------------------------
 # logging configuration
-#--------------------------------------------------
+# --------------------------------------------------
 logger = logging.getLogger(__name__)
 
+
 def posca_env_check():
-    print "========== posca system bandwidth env check ==========="
-    if os.path.exists(r'/home/opnfv/bottlenecks/testsuites/posca/test_result/'):
+    print("========== posca system bandwidth env check ===========")
+    filepath = r"/home/opnfv/bottlenecks/testsuites/posca/test_result/"
+    if os.path.exists(filepath):
         return True
     else:
         os.mkdirs(r'/home/opnfv/bottlenecks/testsuites/posca/test_result/')
 
-def posca_output_result(time_new,input_1,input_2,input_3,input_4,input_5):
-    save_dic={}
-    save_dic['tx_pkt_size']=input_1
-    save_dic['tx_cache_size']=input_2
-    save_dic['throughput ']=input_3
-    save_dic['latency']=input_4
-    save_dic['cpu_load']=input_5
-    with open("/home/opnfv/bottlenecks/testsuites/posca/test_result/factor_rx_cache_size_%s.json"%(time_new),"a") as f:
-        f.write(json.dumps(save_dic,f))
+
+def posca_output_result(time_new, input_1, input_2,
+                        input_3, input_4, input_5):
+    save_dic = {}
+    save_dic['tx_pkt_size'] = input_1
+    save_dic['tx_cache_size'] = input_2
+    save_dic['throughput '] = input_3
+    save_dic['latency'] = input_4
+    save_dic['cpu_load'] = input_5
+    with open("/home/opnfv/bottlenecks/testsuites/posca/test_result/\
+            factor_tx_cache_size_%s.json" % (time_new), "a") as f:
+        f.write(json.dumps(save_dic, f))
         f.write("\n")
 
+
 def posca_config_read(config_str):
-    print "========== posca system bandwidth config read ==========="
+    print("========== posca system bandwidth config read ===========")
 
     con_dic = {}
     config = ConfigParser.ConfigParser()
-    with open(config_str,"rd") as cfgfile:
+    with open(config_str, "rd") as cfgfile:
         config.readfp(cfgfile)
-        con_dic['test_ip']=config.get("config","test_ip")
-        con_dic['test_throughput']=config.get("config","throughput")
-        con_dic['test_tool']=config.get("config","tool")
-        con_dic['test_time']=config.get("config","test_time")
-        con_dic['test_protocol']=config.get("config","protocol")
-        con_dic['test_pkt_s']=config.get("config","pkt sizes")
-        con_dic['test_tx_cache_s']=config.get("config","tx cache sizes")
-        con_dic['test_rx_cache_s']=config.get("config","rx cache sizes")
-        con_dic['test_cpu_load']=config.get("config","cpu load")
-        con_dic['test_latency']=config.get("config","latency")
+        con_dic['test_ip'] = config.get("config", "test_ip")
+        con_dic['test_throughput'] = config.get("config", "throughput")
+        con_dic['test_tool'] = config.get("config", "tool")
+        con_dic['test_time'] = config.get("config", "test_time")
+        con_dic['test_protocol'] = config.get("config", "protocol")
+        con_dic['test_pkt_s'] = config.get("config", "pkt sizes")
+        con_dic['test_tx_cache_s'] = config.get("config", "tx cache sizes")
+        con_dic['test_rx_cache_s'] = config.get("config", "rx cache sizes")
+        con_dic['test_cpu_load'] = config.get("config", "cpu load")
+        con_dic['test_latency'] = config.get("config", "latency")
 
     return con_dic
 
+
 def posca_run(con_dic):
-    print "========== run posca system bandwidth ==========="
+    print("========== run posca system bandwidth ===========")
 
     test_pkt_s_a = con_dic['test_pkt_s'].split(',')
     test_tx_cache_s_a = con_dic['test_tx_cache_s'].split(',')
-    time_new = time.strftime('%H_%M',time.localtime(time.time()))
+    test_rx_cache_s_a = con_dic['test_rx_cache_s'].split(',')
+    time_new = time.strftime('%H_%M', time.localtime(time.time()))
     bandwidth_tmp = 1
 
     for test_pkt_s_e in test_pkt_s_a:
         for test_tx_cache_s_e in test_tx_cache_s_a:
-            print "Package size %s"%(test_pkt_s_e)
-            order_excute = os.popen("%s %s http://%s/api/v3/yardstick/tasks/task %s %s %s %s"%(cmd,order_arg,con_dic['test_ip'],test_pkt_s_e,test_pkt_s_e,test_tx_cache_s_e,con_dic['test_rx_cache_s']))
+            print("Package size %s") % (test_pkt_s_e)
+            order_excute = os.popen("%s %s http://%s/api/v3/yardstick/\
+                    tasks/task %s %s %s" % (cmd, order_arg, con_dic['test_ip'],
+                                            test_pkt_s_e, test_rx_cache_s_a,
+                                            test_tx_cache_s_e))
             order_result = order_excute.read()
             task_id = order_result.find("task_id")
             time.sleep(con_dic['test_time'])
-            cmd_excute = os.popen( "%s http://%s/api/v3/yardstick/testresults?task_id=%s"%(cmd,con_dic['test_ip'],task_id))
+            cmd_excute = os.popen("%s http://%s/api/v3/yardstick/testre\
+                    sults?task_id=%s" % (cmd, con_dic['test_ip'], task_id))
             test_result = cmd_excute.read()
             bandwidth = test_result.find("bandwidth")
             cpu_load = test_result.find("cpu_load")
             latency = test_result.find("latency")
-            posca_output_result(time_new,test_pkt_s_e,test_tx_cache_s_e,bandwidth,latency,cpu_load)
-            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']):
-                if abs(bandwidth_tmp-bandwidth)/bandwidth <0.05:
+            posca_output_result(time_new, test_pkt_s_e, test_rx_cache_s_a,
+                                bandwidth, latency, cpu_load)
+            if (cpu_load < con_dic['test_cpu_load\
+            ']) and (latency < con_dic['test_latency']):
+                if (abs(bandwidth_tmp-bandwidth)/bandwidth < 0.05):
                     return True
                 else:
-                    print "%s,%s"%(bandwidth,test_tx_cache_s_e)
+                    print("%s,%s") % (bandwidth, test_tx_cache_s_e)
             else:
-                print "%s,%s"%(bandwidth,test_tx_cache_s_e)
+                print("%s,%s") % (bandwidth, test_tx_cache_s_e)
                 return False
 
+
 def main():
     if not (args.conf):
-       logger.error("Configuration files do not exist for the specified testcases")
-       exit(-1)
+        logger.error("Configuration files do not exist \
+                    for the specified testcases")
+        os.exit(-1)
     else:
-       testcase_cfg = args.conf
+        testcase_cfg = args.conf
 
-    con_dic=posca_config_read(testcase_cfg)
+    con_dic = posca_config_read(testcase_cfg)
     posca_env_check()
     posca_run(con_dic)
 
     time.sleep(5)
 
-if __name__=='__main__':
+if __name__ == '__main__':
     main()