export opcode to SocketIOUser send method to support send binary and …

…other format data
SvenskaSpel · Nov 22, 2023 · c9933ff · c9933ff
1 parent 2f2749a
commit c9933ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/locust_plugins/users/socketio.py b/locust_plugins/users/socketio.py
@@ -88,7 +88,7 @@ def receive_loop(self):
             logging.debug(f"WSR: {message}")
             self.on_message(message)
 
-    def send(self, body, name=None, context={}):
+    def send(self, body, name=None, context={}, opcode=websocket.ABNF.OPCODE_TEXT):
         if not name:
             if body == "2":
                 name = "2 heartbeat"
@@ -112,7 +112,7 @@ def send(self, body, name=None, context={}):
             context={**self.context(), **context},
         )
         logging.debug(f"WSS: {body}")
-        self.ws.send(body)
+        self.ws.send(body, opcode)
 
     def sleep_with_heartbeat(self, seconds):
         while seconds >= 0: