1 files changed, 14 insertions, 3 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 aeb25422..db51683a 100755
--- a/scripts/automation/trex_control_plane/client/trex_stateless_client.py
+++ b/scripts/automation/trex_control_plane/client/trex_stateless_client.py
@@ -254,7 +254,17 @@ class CTRexStatelessClient(object):
                                                         self.ack_success_test)
         pass
 
-    @force_status(owned=True, active_and_owned=True)
+
+    @force_status(owned=True)#, active_and_owned=True)
+    def get_all_streams(self, port_id, get_pkt = False):
+        if not self._is_ports_valid(port_id):
+            raise ValueError("Provided illegal port id input")
+        params = {"handler": self._conn_handler.get(port_id),
+                  "port_id": port_id,
+                  "get_pkt": get_pkt}
+        return self.transmit("get_all_streams", params)
+
+    @force_status(owned=True)#, active_and_owned=True)
     def get_stream_id_list(self, port_id=None):
         if not self._is_ports_valid(port_id):
             raise ValueError("Provided illegal port id input")
@@ -263,12 +273,13 @@ class CTRexStatelessClient(object):
         return self.transmit("get_stream_list", params)
 
     @force_status(owned=True, active_and_owned=True)
-    def get_stream(self, stream_id, port_id=None):
+    def get_stream(self, stream_id, port_id, get_pkt = False):
         if not self._is_ports_valid(port_id):
             raise ValueError("Provided illegal port id input")
         params = {"handler": self._conn_handler.get(port_id),
                   "port_id": port_id,
-                  "stream_id": stream_id}
+                  "stream_id": stream_id,
+                  "get_pkt": get_pkt}
         return self.transmit("get_stream_list", params)
 
     @force_status(owned=True)