fix get_response for bytes response

Geode-solutions · Oct 29, 2024 · 0586102 · 0586102
1 parent 0371651
commit 0586102
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/tests/conftest.py b/src/tests/conftest.py
@@ -57,9 +57,11 @@ def print_log(self):
         print(output)
 
     def get_response(self):
-        return self.ws.recv()
-        response = json.loads(self.ws.recv())
-        return response
+        response = self.ws.recv()
+        if isinstance(response, bytes):
+            return response
+        else:
+            return eval(response)
 
     def compare_image(self, nb_messages, filename):
         for message in range(nb_messages):