Run callbacks in threads

Author: jkim2492

py/selenium/webdriver/remote/websocket_connection.py

@@ -60,8 +60,9 @@ def execute(self, command):
         logger.debug(f"-> {data}"[: self._max_log_message_size])
         self._ws.send(data)
 
-        self._wait_until(lambda: self._id in self._messages)
-        response = self._messages.pop(self._id)
+        current_id = self._id
+        self._wait_until(lambda: current_id in self._messages)
+        response = self._messages.pop(current_id)
 
         if "error" in response:
             raise Exception(response["error"])
@@ -131,7 +132,7 @@ def _process_message(self, message):
         if "method" in message:
             params = message["params"]
             for callback in self.callbacks.get(message["method"], []):
-                callback(params)
+                Thread(target=callback, args=(params,)).start()
 
     def _wait_until(self, condition):
         timeout = self._response_wait_timeout