[py] Added more internal logging for CDP (#14668)

Co-authored-by: Viet Nguyen Duc <nguyenducviet4496@gmail.com>
SeleniumHQ · Oct 29, 2024 · f391cd0 · f391cd0
1 parent 9b8cfb1
commit f391cd0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/py/selenium/webdriver/common/bidi/cdp.py b/py/selenium/webdriver/common/bidi/cdp.py
@@ -211,13 +211,19 @@ async def execute(self, cmd: typing.Generator[dict, T, typing.Any]) -> T:
         if self.session_id:
             request["sessionId"] = self.session_id
         request_str = json.dumps(request)
+        if logger.isEnabledFor(logging.DEBUG):
+            logger.debug(f"Sending CDP message: {cmd_id} {cmd_event}: {request_str}")
         try:
             await self.ws.send_message(request_str)
         except WsConnectionClosed as wcc:
             raise CdpConnectionClosed(wcc.reason) from None
         await cmd_event.wait()
         response = self.inflight_result.pop(cmd_id)
+        if logger.isEnabledFor(logging.DEBUG):
+            logger.debug(f"Received CDP message: {response}")
         if isinstance(response, Exception):
+            if logger.isEnabledFor(logging.DEBUG):
+                logger.debug(f"Exception raised by {cmd_event} message: {type(response).__name__}")
             raise response
         return response