check 'error'

Author: navin772

py/selenium/webdriver/remote/websocket_connection.py

@@ -64,10 +64,13 @@ def execute(self, command):
         self._wait_until(lambda: current_id in self._messages)
         response = self._messages.pop(current_id)
 
-        error = response["error"]
-        if "message" in response:
-            error_msg = f"{error}: {response['message']}"
-            raise Exception(error_msg)
+        if "error" in response:
+            error = response["error"]
+            if "message" in response:
+                error_msg = f"{error}: {response['message']}"
+                raise Exception(error_msg)
+            else:
+                raise Exception(error)
         else:
             result = response["result"]
             return self._deserialize_result(result, command)