POSCA test case complete

JIRA: BOTTLENECK-107

JIRA: BOTTLENECK-106

This patch has done the flowings:
1.Using the RESTful API form yardstick
2.Common function reuse.
3.If there is no test_ip it would use host ip and port as test_ip.
4.A little demo of dashboard. Now is can't work

Change-Id: I1063176c762c40238019c73f5359f23bc5aab19c
Signed-off-by: liyin <liyin11@huawei.com>

1 files changed, 122 insertions, 0 deletions

diff --git a/testsuites/posca/testcase_script/common_script.py b/testsuites/posca/testcase_script/common_script.py
new file mode 100644
index 00000000..9234d938
--- /dev/null
+++ b/testsuites/posca/testcase_script/common_script.py
@@ -0,0 +1,122 @@
+#!/usr/bin/env python
+##############################################################################
+# Copyright (c) 2016 Huawei Technologies Co.,Ltd and others.
+#
+# 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
+##############################################################################
+
+import json
+import requests
+import time
+import subprocess as sub
+from pyroute2 import IPDB
+import sys
+
+headers = {"Content-Type": "application/json"}
+
+
+def posca_tran_data(ES_ip, file_name):
+        p = sub.Popen(['curl', '-s', '-XPOST', "%s/_bulk" % ES_ip,
+                       '--data-binary', "@" + file_name], stdout=sub.PIPE)
+        for line in iter(p.stdout.readline, b''):
+            ret_dict = json.loads(line)
+            if not ret_dict['errors']:
+                print("INFO: %6s lines no errors, total cost %d ms."
+                      % (len(ret_dict['items']), ret_dict['took']))
+                return len(ret_dict['items'])
+            else:
+                print("ERROR: %6s lines have errors, total cost %d ms."
+                      % (len(ret_dict['items']), ret_dict['took']))
+
+
+def posca_config_read(config_str, con_str, config):
+    print("========== posca system bandwidth config read ===========")
+    con_dic = {}
+    print(config_str)
+    idx = 0
+    with open(config_str, "rd") as cfgfile:
+        config.readfp(cfgfile)
+        while idx < len(con_str):
+            con_dic[str(con_str[idx])] = \
+                            config.get("config", str(con_str[idx]))
+            idx += 1
+    with IPDB() as ip:
+        GATEWAY_IP = ip.routes['default'].gateway
+    if str(con_dic["test_ip"]) is "":
+        con_dic["test_ip"] = GATEWAY_IP+":3333"
+        print("test_ip is null get local ip is %s" %(con_dic["test_ip"]))
+    if con_dic["ES_ip"] is "":
+        con_dic["ES_ip"] = GATEWAY_IP+":9200"
+        print("ES_ip is null get local ip is %s" %(con_dic["ES_ip"]))
+    return con_dic
+
+
+def posca_output_result(file_config, data_reply):
+    data_head = {}
+    data_head["index"] = {}
+    data_head["index"]["_index"] = "bottlenecks"
+    data_head["index"]["_type"] = file_config["test_type"]
+    data_head["index"]["_id"] = file_config["test_id"]
+
+    data_reply["throughput"] = float(data_reply["throughput"])
+    data_reply["mean_latency"] = float(data_reply["mean_latency"])
+    data_reply["remote_cpu_util"] = float(data_reply["remote_cpu_util"])
+    data_reply["local_cpu_util"] = float(data_reply["local_cpu_util"])
+    data_reply["local_transport_retrans"] =\
+        float(data_reply["local_transport_retrans"])
+    with open(file_config["file_path"], "a") as f:
+        f.write(json.dumps(data_head, f))
+        f.write("\n")
+        f.write(json.dumps(data_reply, f))
+        f.write("\n")
+        f.close()
+
+
+def posca_get_reply(con_dic, task_id, time_test=1):
+    reply_url = "http://%s/yardstick/result/action?action=getResult&task_id=%s\
+&measurement=tc100" % (con_dic["test_ip"], task_id)
+    time.sleep(float(con_dic["test_time"]))
+    reply_response = requests.get(reply_url)
+    reply_data = json.loads(reply_response.text)
+    print(reply_data)
+    if reply_data["status"] == 1:
+        return(reply_data["result"][0])
+    if reply_data["status"] == 0:
+        if time_test == 3:
+            print("yardstick time out")
+            sys.exit()
+        posca_get_reply(con_dic, task_id, time_test=time_test+1)
+    if reply_data["status"] == 2:
+        print("yardstick error exit")
+        sys.exit()
+
+def posca_send_data(con_dic, test_config, file_config):
+    base_url = "http://%s/yardstick/test/action" % (con_dic['test_ip'])
+    print(con_dic["test_ip"])
+    test_dict = {
+            "action":"runTestCase",
+            "args":{
+                "opts": {
+                    "task-args": {
+                        'tx_msg_size': '%s' % str(test_config["tx_msg_size"]),
+                        'rx_msg_size': '%s' % str(test_config["rx_msg_size"]),
+                        'test_time': '%s' % str(int(con_dic["test_time"]) - 20),
+                        'host': 'node3.LF',
+                        'target': 'node4.LF'
+                        }
+                 },
+                 "testcase":"tc100"
+            }
+    }
+    # print(base_url)
+    reponse = requests.post(
+                        base_url, data=json.dumps(test_dict), headers=headers)
+    ask_data = json.loads(reponse.text)
+    task_id = ask_data["task_id"]
+    data_reply = posca_get_reply(con_dic, task_id)
+    data_reply.update(test_config)
+    posca_output_result(file_config, data_reply)
+    return data_reply