Skip to content

Commit d1f1b6f

Browse files
bastimeyerdiemol
andauthored
[py] Fix BrowserError issues (#12150)
* [py] Fix KeyError on BrowserError * [py] Fix TypeError on BrowserError --------- Co-authored-by: Diego Molina <diemol@users.noreply.github.com>
1 parent 4d70bd8 commit d1f1b6f

File tree

1 file changed

+9
-3
lines changed
  • py/selenium/webdriver/common/bidi

1 file changed

+9
-3
lines changed

py/selenium/webdriver/common/bidi/cdp.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -147,8 +147,8 @@ class BrowserError(Exception):
147147
indicates that an error occurred."""
148148

149149
def __init__(self, obj):
150-
self.code = obj["code"]
151-
self.message = obj["message"]
150+
self.code = obj.get("code")
151+
self.message = obj.get("message")
152152
self.detail = obj.get("data")
153153

154154
def __str__(self):
@@ -442,7 +442,13 @@ async def _reader_task(self):
442442
try:
443443
session = self.sessions[session_id]
444444
except KeyError:
445-
raise BrowserError(f"Browser sent a message for an invalid session: {session_id!r}")
445+
raise BrowserError(
446+
{
447+
"code": -32700,
448+
"message": "Browser sent a message for an invalid session",
449+
"data": f"{session_id!r}",
450+
}
451+
)
446452
session._handle_data(data)
447453
else:
448454
self._handle_data(data)

0 commit comments

Comments
 (0)