raise WebDriverException

Author: navin772

py/selenium/webdriver/remote/websocket_connection.py

@@ -22,6 +22,8 @@
 
 from websocket import WebSocketApp  # type: ignore
 
+from selenium.common import WebDriverException
+
 logger = logging.getLogger(__name__)
 
 
@@ -68,9 +70,9 @@ def execute(self, command):
             error = response["error"]
             if "message" in response:
                 error_msg = f"{error}: {response['message']}"
-                raise Exception(error_msg)
+                raise WebDriverException(error_msg)
             else:
-                raise Exception(error)
+                raise WebDriverException(error)
         else:
             result = response["result"]
             return self._deserialize_result(result, command)
@@ -102,7 +104,7 @@ def _serialize_command(self, command):
     def _deserialize_result(self, result, command):
         try:
             _ = command.send(result)
-            raise Exception("The command's generator function did not exit when expected!")
+            raise WebDriverException("The command's generator function did not exit when expected!")
         except StopIteration as exit:
             return exit.value