Skip to content

Commit 431ba5b

Browse files
committed
[py] Add input validation for timeout and interval
1 parent 0f251c5 commit 431ba5b

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

py/selenium/webdriver/remote/websocket_connection.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,12 +32,17 @@ class WebSocketConnection:
3232
_max_log_message_size = 9999
3333

3434
def __init__(self, url, timeout, interval):
35-
self.callbacks = {}
36-
self.session_id = None
35+
if not isinstance(timeout, (int, float)) or timeout < 0:
36+
raise WebDriverException("timeout must be a positive number")
37+
if not isinstance(interval, (int, float)) or timeout < 0:
38+
raise WebDriverException("interval must be a positive number")
39+
3740
self.url = url
3841
self.response_wait_timeout = timeout
3942
self.response_wait_interval = interval
4043

44+
self.callbacks = {}
45+
self.session_id = None
4146
self._id = 0
4247
self._messages = {}
4348
self._started = False

0 commit comments

Comments
 (0)