From 8d6777df09c3dc441013a31f21cc50ab3b0f42a3 Mon Sep 17 00:00:00 2001 From: Billy O'Mahony Date: Fri, 29 May 2015 15:24:03 +0100 Subject: framework: Add reworked framework to repo This commit adds the vSwitch Integration Test Framework whose design, based off TOIT, is outlined in the HLD previously made availiable to the community for review. The design of this framework allows developers to add different implementations of components, specifically vSwitches, Traffic Generators, Metrics Collectors and VNFs, easily. The goal of this design is that all testcases should run regardless of what is "under the hood". This commit adds support for running the framework for a phy to phy RFC2544 testcase only. More testcases will be added by the community. vSwitches supported at this time: * Intel DPDK (r) accelerated OpenvSwitch Traffic Generators supported at this time: * IxNet - IxNetwork Implementation * Ixia - IxExplorer Implementation * Dummy - Manual Implementation Metrics Collectors supported at this time: * Linux Metrics No VNFs are supported at this time but the framework outlines how they should be integrated and provides APIs for them to adhere to. JIRA: VSPERF-27 Change-Id: I312e1a1199487ffee8f824be06cd97d4f793eee0 Signed-off-by: Stephen Finucane Signed-off-by: Meghan Halton Signed-off-by: Christopher Nolan Signed-off-by: Maryam Tahhan Signed-off-by: Ciara Loftus Signed-off-by: Mark Kavanagh Signed-off-by: Cian Ferriter Signed-off-by: Timo Puha Signed-off-by: Billy O'Mahony Signed-off-by: Michal Weglicki Signed-off-by: Rory Sexton Signed-off-by: Ian Stokes Signed-off-by: Kevin Traynor Signed-off-by: Dino Simeon Madarang Reviewed-by: Eugene Snider Reviewed-by: Aihua Li --- testcases/testcase.py | 128 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 128 insertions(+) create mode 100644 testcases/testcase.py (limited to 'testcases/testcase.py') diff --git a/testcases/testcase.py b/testcases/testcase.py new file mode 100644 index 00000000..76758c8f --- /dev/null +++ b/testcases/testcase.py @@ -0,0 +1,128 @@ +# Copyright 2015 Intel Corporation. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +"""TestCase base class +""" + +import time +import csv +import os +import logging +from collections import OrderedDict + +import core.component_factory as component_factory +from core.loader import Loader + +class TestCase(object): + """TestCase base class + + In this basic form runs RFC2544 throughput test + """ + def __init__(self, cfg, results_dir): + """Pull out fields from test config + + No external actions yet. + """ + self._logger = logging.getLogger(__name__) + self.name = cfg['Name'] + self.desc = cfg.get('Description', 'No description given.') + self._traffic_type = cfg['Traffic Type'] + self._deployment = cfg['Deployment'] + self._collector = cfg['Collector'] + self._results_dir = results_dir + + def run(self): + """Run the test + + All setup and teardown through controllers is included. + """ + self._logger.debug(self.name) + + self._logger.debug("Controllers:") + loader = Loader() + traffic_ctl = component_factory.create_traffic( + self._traffic_type, + loader.get_trafficgen_class()) + vnf_ctl = component_factory.create_vnf( + self._deployment, + loader.get_vnf_class()) + vswitch_ctl = component_factory.create_vswitch( + self._deployment, + loader.get_vswitch_class()) + collector_ctl = component_factory.create_collector( + self._collector, + loader.get_collector_class()) + + self._logger.debug("Setup:") + collector_ctl.log_cpu_stats() + with vswitch_ctl: + if vnf_ctl: + vnf_ctl.start() + #TODO 'traffic' is placeholder for traffic dict + traffic = {'test': 'rfc2544'} + with traffic_ctl: + traffic_ctl.send_traffic(traffic) + + + self._logger.debug("Traffic Results:") + traffic_ctl.print_results() + + self._logger.debug("Collector Results:") + self._logger.debug(collector_ctl.get_results()) + + + output_file = "result_" + self.name + "_" + self._deployment +".csv" + + self._write_result_to_file( + traffic_ctl.get_results(), + os.path.join(self._results_dir, output_file)) + + @staticmethod + def _write_result_to_file(results, output): + """Write list of dictionaries to a CSV file. + + Each element on list will create separate row in output file. + If output file already exists, data will be appended at the end, + otherwise it will be created. + + :param results: list of dictionaries. + :param output: path to output file. + """ + with open(output, 'a') as csvfile: + + logging.info("Write results to file: " + output) + fieldnames = TestCase._get_unique_keys(results) + + writer = csv.DictWriter(csvfile, fieldnames) + + if not csvfile.tell(): # file is now empty + writer.writeheader() + + for result in results: + writer.writerow(result) + + + @staticmethod + def _get_unique_keys(list_of_dicts): + """Gets unique key values as ordered list of strings in given dicts + + :param list_of_dicts: list of dictionaries. + + :returns: list of unique keys(strings). + """ + result = OrderedDict() + for item in list_of_dicts: + for key in item.keys(): + result[key] = '' + + return list(result.keys()) -- cgit 1.2.3-korg