Upload the contribution of vstf as bottleneck network framework.

[bottlenecks.git] / vstf / vstf / controller / spirent / appliance.py

diff --git a/vstf/vstf/controller/spirent/appliance.py b/vstf/vstf/controller/spirent/appliance.py
new file mode 100755 (executable)
index 0000000..a06bf45
--- /dev/null
+++ b/vstf/vstf/controller/spirent/appliance.py
@@ -0,0 +1,92 @@
+#!/usr/bin/python
+"""
+    @author: l00190809
+    @group: Huawei Ltd 
+"""
+import os
+import logging
+
+import stevedore
+from vstf.controller.spirent.common.result_analysis import analysis_instance as analysis_instance
+LOG = logging.getLogger(__name__)
+
+
+class spirentSTC(object):
+    def __init__(self):
+        super(spirentSTC, self).__init__()
+        self.runmodel = None
+
+    def init(self, conner="", measurand="", model="", **kwargs):
+        """
+        :param str    conner: the spirent tester, the agent id of spirent vm
+        :param list   measurand: the tested host's agent id
+        :param str    model: the model used of the tested host
+        
+        """
+        mgr = stevedore.driver.DriverManager(namespace="spirent.model.plugins",
+                                             name=model,
+                                             invoke_on_load=False)
+        self.TempMod = mgr.driver(kwargs)
+        self.conner = conner
+        self.measurand = measurand
+
+    @property
+    def run(self):
+        LOG.info(vars(self.runmodel))
+        return True
+
+
+def run(config):
+    # test option parser 
+    if not os.path.exists(config['configfile']):
+        LOG.error('The config file %s does exist.', config.get("configfile"))
+        return False
+
+    runmodel = None  # Tnv_Model(config = config)
+
+    # check parameter valid
+    flag = runmodel.check_parameter_invalid()
+    if not flag:
+        LOG.error("[ERROR]Check parameter invalid.")
+        return False
+
+    # check logical parameter in the 
+    flag = runmodel.check_logic_invalid
+    if not flag:
+        LOG.error("[ERROR]Check logic parameter with host invalid.")
+        return False
+
+    init_flows_tables = runmodel.read_flow_init
+    LOG.info(init_flows_tables)
+
+    # print init_flows_tables
+    update_flows = runmodel.flow_build
+    # print update_flows
+    LOG.info(update_flows)
+
+    flag = runmodel.affinity_bind(aff_strategy=1)
+    if not flag:
+        LOG.error("runmodel affinity bind failed.")
+        return False
+
+    # Get the result
+    result = {}
+    for suite in ["frameloss", "throughput"]:
+        ret, test_result = runmodel.Test_Run(suite)
+        if not ret:
+            LOG.error("[ERROR]Run rfc2544 %s test failed.", suite)
+            return False
+        try:
+            ret, result_dict = restrucData(test_result)
+        except:
+            LOG.error("[ERROR]Restructure the test data failed.")
+        perfdata = getResult(result_dict)
+        columndata = getResultColumn(result_dict)
+        column_array, data_array = analysis_instance.analyseResult(suite, columndata, perfdata)
+        temp = {'columns': column_array, 'data': data_array}
+        result[suite] = temp
+    return result
+
+
+if __name__ == "__main__":
+    run(None)