made changes to websocket_connection.py, added fail_request()

Author: shbenzer

py/selenium/webdriver/common/bidi/network.py

@@ -280,6 +280,15 @@ def command_builder(self, method, params):
         command = {"method": method, "params": params}
         cmd = yield command
         return cmd
+    
+    def fail_request(self):
+        """Fail this request."""
+
+        if not self.request_id:
+            raise ValueError("Request not found.")
+
+        params = {"request": self.request_id}
+        self.network.conn.execute(self.command_builder("network.failRequest", params))
 
     def continue_request(self, body=None, method=None, headers=None, cookies=None, url=None):
         """Continue after intercepting this request."""

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