From c6755e612b322facefc4edc32e948ab2b00bb3b0 Mon Sep 17 00:00:00 2001 From: liyin Date: Tue, 24 Jan 2017 17:26:21 +0800 Subject: [PATCH] Bottlenecks POSCA testing code reconstruction JIRA:BOTTLENECK-103 This is the foundation of adding stack samples. This code change a lot code. but it's a basic. Those code will be changed in the furture. Change-Id: I8d5bbb9cc401b1aaac54ec4dffc4c005a42d17ac Signed-off-by: liyin --- setup.py | 46 ++++--- testsuites/posca/run_posca.py | 105 +++++++++------- .../posca_factor_system_bandwidth.yaml | 26 ++-- .../posca_factor_system_bandwidth.py | 135 +++++++-------------- utils/infra_setup/runner/__init__.py | 8 ++ utils/infra_setup/runner/stack.py | 61 ++++++++++ utils/infra_setup/runner/yardstick.py | 80 ++++++++++++ utils/parser.py | 33 +++-- 8 files changed, 322 insertions(+), 172 deletions(-) create mode 100644 utils/infra_setup/runner/__init__.py create mode 100644 utils/infra_setup/runner/stack.py create mode 100644 utils/infra_setup/runner/yardstick.py diff --git a/setup.py b/setup.py index 55b89cc1..d9852c95 100644 --- a/setup.py +++ b/setup.py @@ -7,26 +7,38 @@ # which accompanies this distribution, and is available at # http://www.apache.org/licenses/LICENSE-2.0 ############################################################################## +'''This file realize the function of how to setup bottlenecks to your environment +This use setuptools tool to setup''' from setuptools import setup, find_packages setup( - name="bottlenecks", - version="master", - py_modules=['bottlenecks_cli'], - packages=find_packages(), - include_package_data=True, - package_data={ - 'utils': [ - 'utils/infra_setup/heat/*.py' - ] - }, - url="https://www.opnfv.org", - install_requires=["click"], - entry_points={ - 'console_scripts': [ - 'bottlenecks=cli.bottlenecks_cli:main' - ], - }, + name="bottlenecks", + version="master", + py_modules=['bottlenecks_cli'], + packages=find_packages(), + include_package_data=True, + package_data={ + 'utils': [ + 'utils/infra_setup/heat/*.py', + 'utils/infra_setup/runner/*.py' + ], + 'config': [ + '*.yaml' + ], + 'testsuites': [ + 'posca/testcase_cfg/*', + 'posca/testcase_script/*', + 'posca/testsuite_story/*', + 'posca/testcase_dashboard/*' + ], + }, + url="https://www.opnfv.org", + install_requires=["click"], + entry_points={ + 'console_scripts': [ + 'bottlenecks=cli.bottlenecks_cli:main' + ], + }, ) diff --git a/testsuites/posca/run_posca.py b/testsuites/posca/run_posca.py index e5e11e9f..fcf35d5b 100755 --- a/testsuites/posca/run_posca.py +++ b/testsuites/posca/run_posca.py @@ -1,44 +1,61 @@ -#!/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 sys -import subprocess - -INTERPRETER = "/usr/bin/python" -# ------------------------------------------------------ -# run posca testcase -# ------------------------------------------------------ - - -def posca_run(arg): - print("========== run posca ==========") - print(arg) - if(arg == "posca_factor_system_bandwidth"): - print("========== run posca_system_bandwidth ===========") - cmd = '/home/opnfv/bottlenecks/testsuites/posca/testcase_script/\ -posca_factor_system_bandwidth.py' - pargs = [INTERPRETER, cmd] - sub_result = subprocess.Popen(pargs) - sub_result.wait() - - -def posca_env_check(): - print("========== posca env check ===========") - - -def main(): - # para_testname = sys.argv[0] - para_test_arg = sys.argv[2] - posca_env_check() - posca_run(para_test_arg) - sys.exit(0) - -if __name__ == '__main__': - main() +#!/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 +############################################################################## +'''This file realize the function of how to run posca. +In this file, The first thing is to read testcase config +for example: you could run this by use +posca_run('testcase', "Which testcase you will run") +posca_run('teststory', "Which story you will run") +and if you run "python run_posca", this will run testcase, +posca_factor_system_bandwidth by default.''' + +import importlib +import utils.parser as conf_parser +import utils.logger as log +INTERPRETER = "/usr/bin/python" + +LOG = log.Logger(__name__).getLogger() +# ------------------------------------------------------ +# run posca testcase +# ------------------------------------------------------ + + +def posca_testcase_run(testcase_script, test_config): + + module_string = "testsuites.posca.testcase_script.%s" % (testcase_script) + module = importlib.import_module(module_string) + module.run(test_config) + + +def posca_run(test_level, test_name): + if test_level == "testcase": + config = conf_parser.Parser.testcase_read("posca", test_name) + elif test_level == "teststory": + config = conf_parser.Parser.story_read("posca", test_name) + for testcase in config: + print(config[testcase]) + posca_testcase_run(testcase, config[testcase]) + if con_dic["dashboard"] == "y": + cmd = '/home/opnfv/bottlenecks/testsuites/posca/testcase_dashboard/\ +system_bandwidth.py' + pargs = [INTERPRETER, cmd] + LOG.info("\nBegin to establish dashboard.") + sub_result = subprocess.Popen(pargs) + sub_result.wait() + + +def main(): + test_level = "testcase" + test_name = "posca_factor_system_bandwidth" + posca_run(test_level, test_name) + + +if __name__ == '__main__': + main() + diff --git a/testsuites/posca/testcase_cfg/posca_factor_system_bandwidth.yaml b/testsuites/posca/testcase_cfg/posca_factor_system_bandwidth.yaml index 8235d5b4..217bff3e 100755 --- a/testsuites/posca/testcase_cfg/posca_factor_system_bandwidth.yaml +++ b/testsuites/posca/testcase_cfg/posca_factor_system_bandwidth.yaml @@ -1,11 +1,15 @@ -[config] -test_ip: -dashboard: y -ES_ip: -tool: netperf -protocol: tcp -test_time: 60 -tx_pkt_sizes: 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192, 16384, 32768, 65536, 131072 -rx_pkt_sizes: 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192, 16384, 32768, 65536, 131072 -cpu_load: 0.9 -latency: 100000 +test_config: + tool: netperf + protocol: tcp + test_time: 60 + tx_pkt_sizes: 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192, 16384, 32768, 65536, 131072 + rx_pkt_sizes: 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192, 16384, 32768, 65536, 131072 + cpu_load: 0.9 + latency: 100000 +runner_config: + dashboard: y + dashboard_ip: + stack_create: yardstick + yardstick_test_ip: + yardstick_test_dir: "samples" + yardstick_testcase: "netperf_bottlenecks" diff --git a/testsuites/posca/testcase_script/posca_factor_system_bandwidth.py b/testsuites/posca/testcase_script/posca_factor_system_bandwidth.py index 7a0fd27b..ed4de180 100644 --- a/testsuites/posca/testcase_script/posca_factor_system_bandwidth.py +++ b/testsuites/posca/testcase_script/posca_factor_system_bandwidth.py @@ -1,54 +1,59 @@ #!/usr/bin/env python ############################################################################## -# Copyright (c) 2016 Huawei Technologies Co.,Ltd and others. +# 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 argparse import time -import logging -import ConfigParser -import common_script -import datetime -import subprocess - -# ------------------------------------------------------ -# 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_system_bandwidth.yaml") -args = parser.parse_args() -headers = {"Content-Type": "application/json"} -INTERPRETER = "/usr/bin/python" - - +import utils.logger as log +import utils.infra_setup.runner.yardstick as Runner # -------------------------------------------------- # logging configuration # -------------------------------------------------- -logger = logging.getLogger(__name__) +LOG = log.Logger(__name__) + +test_dict = { + "action": "runTestCase", + "args": { + "opts": { + "task-args": {} + }, + "testcase": "netperf_bottlenecks" + } +} + + +def env_pre(): + Runner.Create_Incluxdb() -def posca_env_check(): - print("========== posca system bandwidth env check ===========") - filepath = r"/home/opnfv/bottlenecks/testsuites/posca/test_result/" - if os.path.exists(filepath): - return True - else: - os.mkdir(r'/home/opnfv/bottlenecks/testsuites/posca/test_result/') +def do_test(test_config, con_dic): + test_dict['args']['opts']['task-args'] = test_config + Task_id = Runner.Send_Data(test_dict, con_dic['runner_config']) + time.sleep(con_dic['test_config']['test_time']) + Data_Reply = Runner.Get_Reply(con_dic['runner_config'], Task_id) + test_date = Data_Reply[con_dic['runner_config']['testcase']][0] + return test_date -def system_pkt_bandwidth(test_id, data, file_config, con_dic): - date_id = test_id - print("test is is begin from %d" % test_id) +def run(con_dic): + data = {} + rx_pkt_a = con_dic['test_config']['rx_pkt_sizes'].split(',') + tx_pkt_a = con_dic['test_config']['tx_pkt_sizes'].split(',') + data["rx_pkt_sizes"] = rx_pkt_a + data["tx_pkt_sizes"] = tx_pkt_a + con_dic["result_file"] = os.path.dirname( + os.path.abspath(__file__)) + "/test_case/result" + date_id = 0 cur_role_result = 1 pre_role_result = 1 pre_reply = {} @@ -62,12 +67,11 @@ def system_pkt_bandwidth(test_id, data, file_config, con_dic): test_config = { "tx_msg_size": float(test_x), "rx_msg_size": float(test_y), + "test_time": con_dic['test_config']['test_time'] } date_id = date_id + 1 - file_config["test_id"] = date_id - data_reply = common_script.posca_send_data( - con_dic, test_config, file_config) - bandwidth = data_reply["throughput"] + data_reply = do_test(test_config, con_dic) + bandwidth = float(data_reply["throughput"]) if (data_max["throughput"] < bandwidth): data_max = data_reply if (abs(bandwidth_tmp - bandwidth) / bandwidth_tmp < 0.025): @@ -76,10 +80,9 @@ def system_pkt_bandwidth(test_id, data, file_config, con_dic): else: pre_reply = data_reply bandwidth_tmp = bandwidth - cur_role_result = pre_reply["throughput"] + cur_role_result = float(pre_reply["throughput"]) if (abs(pre_role_result - cur_role_result) / pre_role_result < 0.025): print("date_id is %d,package return at line 111\n" % date_id) - # return data_return if data_return["throughput"] < data_max["throughput"]: data_return = data_max pre_role_result = cur_role_result @@ -87,60 +90,10 @@ def system_pkt_bandwidth(test_id, data, file_config, con_dic): return data_return -def posca_run(con_dic): - print("========== run posca system bandwidth ===========") - test_con_id = 0 - file_config = {} - data = {} - rx_pkt_s_a = con_dic['rx_pkt_sizes'].split(',') - tx_pkt_s_a = con_dic['tx_pkt_sizes'].split(',') - time_new = time.strftime('%H_%M', time.localtime(time.time())) - file_config["file_path"] = "/home/opnfv/bottlenecks/testsuites/posca/\ -test_result/factor_system_system_bandwidth_%s.json" % (time_new) - file_config["test_type"] = "system_bandwidth_biggest" - data["rx_pkt_sizes"] = rx_pkt_s_a - data["tx_pkt_sizes"] = tx_pkt_s_a - print("######test package begin######") - pkt_reply = system_pkt_bandwidth( - test_con_id, data, file_config, con_dic) - - print("######find system bandwidth######") - print("rx_msg_size:%d tx_msg_size:%d\n" % - (pkt_reply["rx_msg_size"], pkt_reply["tx_msg_size"])) - common_script.posca_tran_data( - con_dic['ES_ip'], file_config["file_path"]) - return True - - def main(): - if not (args.conf): - logger.error("Configuration files do not exist for \ - the specified testcases") - os.exit(-1) - else: - testcase_cfg = args.conf + run(con_dic) - con_str = [ - 'test_ip', 'tool', 'test_time', 'protocol', - 'tx_pkt_sizes', 'rx_pkt_sizes', 'cpu_load', - 'latency', 'ES_ip', 'dashboard' - ] - posca_env_check() - starttime = datetime.datetime.now() - config = ConfigParser.ConfigParser() - con_dic = common_script.posca_config_read(testcase_cfg, con_str, config) - common_script.posca_create_incluxdb(con_dic) - posca_run(con_dic) - endtime = datetime.datetime.now() - if con_dic["dashboard"] == "y": - cmd = '/home/opnfv/bottlenecks/testsuites/posca/testcase_dashboard/\ -system_bandwidth.py' - pargs = [INTERPRETER, cmd] - print("\nBegin to establish dashboard.") - sub_result = subprocess.Popen(pargs) - sub_result.wait() - print("System Bandwidth testing time : %s" % (endtime - starttime)) - time.sleep(5) if __name__ == '__main__': main() + diff --git a/utils/infra_setup/runner/__init__.py b/utils/infra_setup/runner/__init__.py new file mode 100644 index 00000000..b124dfa9 --- /dev/null +++ b/utils/infra_setup/runner/__init__.py @@ -0,0 +1,8 @@ +############################################################################## +# 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 +############################################################################## diff --git a/utils/infra_setup/runner/stack.py b/utils/infra_setup/runner/stack.py new file mode 100644 index 00000000..fb78c360 --- /dev/null +++ b/utils/infra_setup/runner/stack.py @@ -0,0 +1,61 @@ +#!/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 use config to be born with a template +a template is use to create a stack in openstack. +This file will be amended in the furture.''' + +import utils.infra_setup.heat.template as Heat + +class stack_api(): + + def from_config_template(stack_info): + stack_name = stack_info['name'] + ext_gw_net = os.environ.get("EXTERNAL_NETWORK") + heat_parser = HeatTemplate_Parser() + heat_parser.add_security_group(stack_name) + heat_parser.add_keypair(stack_name) + for network in stack_info['networks']: + heat_parser.add_network(stack_name) + heat_parser.add_subnet(stack_name, + stack_info['networks'][network]['cidr']) + + heat_parser.add_router(stack_name, ext_gw_net) + heat_parser.add_router_interface(stack_name) + + for server in stack_info['servers']: + heat_parser.add_port(server, stack_name) + heat_parser.add_floating_ip(server, stack_name, ext_gw_net) + heat_parser.add_floating_ip_ass(server) + heat_parser.add_server(server, + stack_info['servers'][server]['image'], + stack_info['servers'][server]['flavor'], + stack_info['servers'][server]['user']) + + template = heat_parser.get_template_date() + stack = Heat.HeatTemplate(name=stack_name, heat_template=template) + try: + self.stack = stack.create() + except KeyboardInterrupt: + sys.exit("\nStack create interrupted") + except RuntimeError as err: + sys.exit("error: failed to deploy stack: '%s'" % err.args) + except Exception as err: + sys.exit("error: failed to deploy stack: '%s'" % err) + + for server in stack_info['servers']: + if len(server.ports) > 0: + # TODO(hafe) can only handle one internal network for now + port = list(server.ports.values())[0] + server.private_ip = self.stack.outputs[port["stack_name"]] + + if server.floating_ip: + server.public_ip = \ + self.stack.outputs[server.floating_ip["stack_name"]] + diff --git a/utils/infra_setup/runner/yardstick.py b/utils/infra_setup/runner/yardstick.py new file mode 100644 index 00000000..7c8cd255 --- /dev/null +++ b/utils/infra_setup/runner/yardstick.py @@ -0,0 +1,80 @@ +#!/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 contain all function about yardstick API. +At present, This file contain the following function: +1.Ask Yardstick to run testcase and get a task id. +2.use task id to ask yardstick for data. +3.Ask yardstick for InfluxDB create +4.how the process of task.''' + +import sys +import time +import requests +import json +import utils.logger as logger + +headers = {"Content-Type": "application/json"} +LOG = logger.Logger(__name__).getLogger() + + +def Get_Reply(test_config, task_id, time_test=1): + reply_url = ("http://%s/yardstick/results?task_id=%s" + % (test_config['yardstick_test_ip'], task_id)) + reply_response = requests.get(reply_url) + reply_data = json.loads(reply_response.text) + LOG.info("return data is %s" % (reply_data)) + if reply_data["status"] == 1: + return(reply_data["result"]) + if reply_data["status"] == 0: + if time_test == 10: + LOG.info("yardstick time out") + sys.exit() + time.sleep(10) + reply_result_data = Get_Reply( + test_config, task_id, time_test=time_test + 1) + return(reply_result_data) + if reply_data["status"] == 2: + LOG.error("yardstick error exit") + sys.exit() + + +def Send_Data(test_dict, test_config): + base_url = ("http://%s/yardstick/testcases/%s/action" + % (test_config['yardstick_test_ip'], + test_config['yardstick_test_dir'])) + LOG.info("test ip addr is %s" % (base_url)) + reponse = requests.post( + base_url, data=json.dumps(test_dict), headers=headers) + ask_data = json.loads(reponse.text) + task_id = ask_data["result"] + LOG.info("yardstick task id is: %s" % (task_id)) + return task_id + + +def Create_Incluxdb(con_dic): + base_url = ("http://%s/yardstick/env/action" + % (con_dic['yardstick_test_ip'])) + test_dict = { + "action": "createInfluxDBContainer", + } + requests.post( + base_url, data=json.dumps(test_dict), headers=headers) + LOG.info("waiting for creating InfluxDB") + time.sleep(30) + LOG.info("Done, creating InflxDB Container") + + +def find_condition(con_dic): + base_url = ("http://%s/yardstick/asynctask?%s" + % (con_dic['yardstick_test_ip'].id)) + requests.post( + base_url, headers=headers) + LOG.info("check for creating InfluxDB") + diff --git a/utils/parser.py b/utils/parser.py index ce5096a3..c8715859 100644 --- a/utils/parser.py +++ b/utils/parser.py @@ -7,6 +7,11 @@ # which accompanies this distribution, and is available at # http://www.apache.org/licenses/LICENSE-2.0 ############################################################################## +'''This file realize the function of how to parser a config file. +This contain Two part: +Frist is Init some variables the will be used. +Second is reading config file.''' + import os import yaml @@ -14,6 +19,7 @@ import yaml class Parser(): bottlenecks_config = {} + bottlenecks_test = {} @classmethod def config_init(cls): @@ -35,7 +41,7 @@ class Parser(): cls.config_dir_check(cls.bottlenecks_config["log_dir"]) @classmethod - def config_read(cls, testcase, story_name): + def story_read(cls, testcase, story_name): story_dir = os.path.join( cls.test_dir, testcase, @@ -44,15 +50,23 @@ class Parser(): with open(story_dir) as file: story_parser = yaml.load(file) for case_name in story_parser['testcase']: - testcase_dir = os.path.join( - cls.test_dir, - testcase, - 'testcase_cfg', - case_name) - with open(testcase_dir) as f: - cls.bottlenecks_config[case_name] = yaml.load(f) + Parser.testcase_read(cls, testcase, case_name) + + return cls.bottlenecks_test + + @classmethod + def testcase_read(cls, testcase, testcase_name): - return cls.bottlenecks_config + testcase_dir = os.path.join( + cls.test_dir, + testcase, + 'testcase_cfg', + testcase_name) + testcase_local = testcase_dir + ".yaml" + with open(testcase_local) as f: + cls.bottlenecks_test[testcase_name] = yaml.load(f) + + return cls.bottlenecks_test @classmethod def config_dir_check(cls, dirname): @@ -67,3 +81,4 @@ class Parser(): stack_cfg = testcase_cfg['stack_config'] # TO-DO add cli parameters to stack_config. return test_cfg, stack_cfg + -- 2.16.6