review

mozilla-services · Feb 5, 2024 · cdc880b · cdc880b
1 parent 52a0325
commit cdc880b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/load/locustfiles/stored.py b/tests/load/locustfiles/stored.py
@@ -195,8 +195,12 @@ def connect_and_hello(self) -> None:
         for _ in range(len(self.notification_records)):
             self.recv_message()
 
-    def recv_message(self):
-        self.on_ws_message(self.ws, self.ws.recv())
+    def recv_message(self) -> None:
+        data = self.ws.recv()
+        if not isinstance(data, str):
+            logger.error("recv_message unexpectedly recieved bytes")
+            data = str(data)
+        self.on_ws_message(self.ws, data)
 
     def post_notification(self, endpoint_url: str) -> None:
         """Send a notification to Autopush.