Merge branch 'rpc_intg1' into dan_stateless

1 files changed, 24 insertions, 8 deletions

diff --git a/scripts/automation/trex_control_plane/client/trex_stateless_client.py b/scripts/automation/trex_control_plane/client/trex_stateless_client.py
index 9e1c7cf3..6b1b7b94 100755
--- a/scripts/automation/trex_control_plane/client/trex_stateless_client.py
+++ b/scripts/automation/trex_control_plane/client/trex_stateless_client.py
@@ -13,6 +13,8 @@ from common.trex_stats import *
 from common.trex_streams import *
 from collections import namedtuple
 
+from trex_async_client import CTRexAsyncClient
+
 RpcCmdData = namedtuple('RpcCmdData', ['method', 'params'])
 
 class RpcResponseStatus(namedtuple('RpcResponseStatus', ['success', 'id', 'msg'])):
@@ -27,10 +29,10 @@ class RpcResponseStatus(namedtuple('RpcResponseStatus', ['success', 'id', 'msg']
 class CTRexStatelessClient(object):
     """docstring for CTRexStatelessClient"""
 
-    def __init__(self, username, server="localhost", port=5050, virtual=False):
+    def __init__(self, username, server="localhost", sync_port=5050, async_port = 4500, virtual=False):
         super(CTRexStatelessClient, self).__init__()
         self.user = username
-        self.comm_link = CTRexStatelessClient.CCommLink(server, port, virtual)
+        self.comm_link = CTRexStatelessClient.CCommLink(server, sync_port, virtual)
         self.verbose = False
         self._conn_handler = {}
         self._active_ports = set()
@@ -39,6 +41,9 @@ class CTRexStatelessClient(object):
         self._server_version = None
         self.__err_log = None
 
+        self._async_client = CTRexAsyncClient(async_port)
+
+
     # ----- decorator methods ----- #
     def force_status(owned=True, active_and_owned=False):
         def wrapper(func):
@@ -100,6 +105,12 @@ class CTRexStatelessClient(object):
             return rc, err
         return self._init_sync()
 
+    def get_stats_async (self):
+        return self._async_client.get_stats()
+
+    def get_connection_port (self):
+        return self.comm_link.port
+
     def disconnect(self):
         return self.comm_link.disconnect()
 
@@ -125,6 +136,10 @@ class CTRexStatelessClient(object):
         else:
             return port_ids
 
+    def sync_user(self):
+        return self.transmit("sync_user")
+
+
     def get_acquired_ports(self):
         return self._conn_handler.keys()
 
@@ -264,7 +279,7 @@ class CTRexStatelessClient(object):
         if isinstance(port_id, list) or isinstance(port_id, set):
             # handle as batch mode
             port_ids = set(port_id)  # convert to set to avoid duplications
-            commands = [RpcCmdData("start_traffic", {"handler": self._conn_handler.get(p_id), "port_id": p_id})
+            commands = [RpcCmdData("start_traffic", {"handler": self._conn_handler.get(p_id), "port_id": p_id, "mul": 1.0})
                         for p_id in port_ids]
             rc, resp_list = self.transmit_batch(commands)
             if rc:
@@ -272,7 +287,8 @@ class CTRexStatelessClient(object):
                                                   success_test=self.ack_success_test)
         else:
             params = {"handler": self._conn_handler.get(port_id),
-                      "port_id": port_id}
+                      "port_id": port_id,
+                      "mul": 1.0}
             command = RpcCmdData("start_traffic", params)
             return self._handle_start_traffic_response(command,
                                                        self.transmit(command.method, command.params),
@@ -299,10 +315,10 @@ class CTRexStatelessClient(object):
                                                        self.transmit(command.method, command.params),
                                                        self.ack_success_test)
 
-    def get_global_stats(self):
-        command = RpcCmdData("get_global_stats", {})
-        return self._handle_get_global_stats_response(command, self.transmit(command.method, command.params))
-        # return self.transmit("get_global_stats")
+#    def get_global_stats(self):
+#        command = RpcCmdData("get_global_stats", {})
+#        return self._handle_get_global_stats_response(command, self.transmit(command.method, command.params))
+#        # return self.transmit("get_global_stats")
 
     @force_status(owned=True, active_and_owned=True)
     def get_port_stats(self, port_id=None):