Add moon feature tenant testcase

JIRA: BOTTLENECKS-212

Add moon feature tenant testcase to Bottlenecks

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

1 files changed, 166 insertions, 0 deletions

diff --git a/testsuites/posca/testcase_script/posca_feature_moon_tenants.py b/testsuites/posca/testcase_script/posca_feature_moon_tenants.py
new file mode 100644
index 00000000..8f4061df
--- /dev/null
+++ b/testsuites/posca/testcase_script/posca_feature_moon_tenants.py
@@ -0,0 +1,166 @@
+#!/usr/bin/env python
+##############################################################################
+# Copyright (c) 2017 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
+##############################################################################
+'''This file realize the function of run systembandwidth script.
+for example this contain two part first run_script,
+second is algorithm, this part is about how to judge the bottlenecks.
+This test is using yardstick as a tool to begin test.'''
+
+import os
+import time
+import uuid
+import json
+import Queue
+import multiprocessing
+import utils.logger as log
+from utils.parser import Parser as conf_parser
+import utils.env_prepare.stack_prepare as stack_prepare
+import utils.infra_setup.runner.docker_env as docker_env
+import utils.infra_setup.runner.yardstick as yardstick_task
+
+# --------------------------------------------------
+# logging configuration
+# --------------------------------------------------
+LOG = log.Logger(__name__).getLogger()
+
+testfile = os.path.basename(__file__)
+testcase, file_format = os.path.splitext(testfile)
+# cidr = "/home/opnfv/repos/yardstick/samples/pvp_throughput_bottlenecks.yaml"
+runner_switch = True
+runner_DEBUG = True
+
+
+def env_pre(con_dic):
+    LOG.info("yardstick environment prepare!")
+    stack_prepare._prepare_env_daemon(True)
+
+
+def config_to_result(test_config, test_result):
+    final_data = []
+    print(test_result)
+    out_data = test_result["result"]["testcases"]
+    test_data = out_data["pvp_throughput_bottlenecks"]["tc_data"]
+    for result in test_data:
+        testdata = {}
+        testdata["vcpu"] = test_config["vcpu"]
+        testdata["memory"] = test_config["memory"]
+        testdata["nrFlows"] = result["data"]["nrFlows"]
+        testdata["packet_size"] = result["data"]["packet_size"]
+        testdata["throughput"] = result["data"]["throughput_rx_mbps"]
+        final_data.append(testdata)
+    return final_data
+
+
+def testcase_parser(runner_conf, out_file="yardstick.out", **parameter_info):
+    cidr = "/home/opnfv/repos/yardstick/" + \
+           runner_conf["yardstick_test_dir"] + \
+           runner_conf["yardstick_testcase"]
+    cmd = yardstick_task.yardstick_command_parser(debug=runner_DEBUG,
+                                                  cidr=cidr,
+                                                  outfile=out_file,
+                                                  parameter=parameter_info)
+    return cmd
+
+
+def do_test(runner_conf, test_config, Use_Dashboard, context_conf):
+    yardstick_container = docker_env.yardstick_info['container']
+    out_file = ("/tmp/yardstick_" + str(uuid.uuid4()) + ".out")
+    cmd = testcase_parser(runner_conf, out_file=out_file, **test_config)
+    print(cmd)
+    stdout = docker_env.docker_exec_cmd(yardstick_container, cmd)
+    LOG.info(stdout)
+    loop_value = 0
+    while loop_value < 60:
+        time.sleep(2)
+        loop_value = loop_value + 1
+        with open(out_file) as f:
+            data = json.load(f)
+            if data["status"] == 1:
+                LOG.info("yardstick run success")
+                break
+            elif data["status"] == 2:
+                LOG.error("yardstick error exit")
+                exit()
+    # data = json.load(output)
+
+    save_data = config_to_result(test_config, data)
+    if Use_Dashboard is True:
+        print("use dashboard")
+        # DashBoard.dashboard_send_data(context_conf, save_data)
+
+    # return save_data["data_body"]
+    return save_data
+
+
+def run(test_config):
+    load_config = test_config["load_manager"]
+    scenarios_conf = load_config["scenarios"]
+    runner_conf = test_config["runners"]
+    Use_Dashboard = False
+
+    env_pre(None)
+    if test_config["contexts"]["yardstick_ip"] is None:
+        load_config["contexts"]["yardstick_ip"] =\
+            conf_parser.ip_parser("yardstick_test_ip")
+
+    if "dashboard" in test_config["contexts"].keys():
+        if test_config["contexts"]["dashboard_ip"] is None:
+            test_config["contexts"]["dashboard_ip"] =\
+                conf_parser.ip_parser("dashboard")
+        LOG.info("Create Dashboard data")
+        Use_Dashboard = True
+        # DashBoard.dashboard_system_bandwidth(test_config["contexts"])
+
+    resources = conf_parser.str_to_list(scenarios_conf["resources"])
+    initial = conf_parser.str_to_list(scenarios_conf["initial"])
+    threshhold = conf_parser.str_to_list(scenarios_conf["threshhold"])
+    timeout = conf_parser.str_to_list(scenarios_conf["timeout"])
+    SLA = conf_parser.str_to_list(scenarios_conf["SLA"])
+    case_config = {"SLA": SLA,
+                   "resources": resources}
+
+    process_queue = Queue.Queue()
+
+    load_config["result_file"] = os.path.dirname(
+        os.path.abspath(__file__)) + "/test_case/result"
+
+    result = 0
+
+    if initial is 0:
+        tenant_number = threshhold
+    else:
+        tenant_number = initial
+
+    while runner_switch is True:
+        for tenant in range(0, tenant_number):
+            process = multiprocessing.Process(target=do_test,
+                                              args=(runner_conf,
+                                                    case_config,
+                                                    Use_Dashboard,
+                                                    test_config["contexts"],
+                                                    ))
+            process.start()
+            process_queue.put(process)
+
+        result = result + tenant_number
+        tenant_number = threshhold
+        time.sleep(timeout)
+
+    while process_queue.qsize():
+        process = process_queue.get()
+        process.terminate()
+
+    if result is initial:
+        result = 0
+    else:
+        result = result - threshhold
+
+    LOG.info("Finished bottlenecks testcase")
+    LOG.info("The result data is %s", result)
+    return result