[py] allow using enable_bidi property on options class

enable_bidi is a boolean property that defaults to False and can be set to True by user
web_socket_url is a string property that defaults to None and is set by the remote response

Author: titusfortner

py/selenium/webdriver/common/options.py

@@ -44,12 +44,23 @@ def __init__(self, name):
         self.name = name
 
     def __get__(self, obj, cls):
+        if self.name == "enableBidi":
+            # whether BiDi is or will be enabled
+            value = obj._caps.get("webSocketUrl")
+            return value is True or isinstance(value, str)
+        if self.name == "webSocketUrl":
+            # Return socket url or None if not created yet
+            value = obj._caps.get(self.name)
+            return None if not isinstance(value, str) else value
         if self.name in ("acceptInsecureCerts", "strictFileInteractability", "setWindowRect", "se:downloadsEnabled"):
             return obj._caps.get(self.name, False)
         return obj._caps.get(self.name)
 
     def __set__(self, obj, value):
-        obj.set_capability(self.name, value)
+        if self.name == "enableBidi":
+            obj.set_capability("webSocketUrl", value)
+        else:
+            obj.set_capability(self.name, value)
 
 
 class _PageLoadStrategyDescriptor:
@@ -249,6 +260,50 @@ class BaseOptions(metaclass=ABCMeta):
         - `None`
     """
 
+    enable_bidi = _BaseOptionsDescriptor("enableBidi")
+    """Gets and Set whether the session has WebDriverBiDi enabled.
+
+    Usage
+    -----
+    - Get
+        - `self.enable_bidi`
+    - Set
+        - `self.enable_bidi` = `value`
+
+    Parameters
+    ----------
+    `value`: `bool`
+
+    Returns
+    -------
+    - Get
+        - `bool`
+    - Set
+        - `None`
+    """
+
+    web_socket_url = _BaseOptionsDescriptor("webSocketUrl")
+    """Gets and Set whether the session accepts insecure certificates.
+
+    Usage
+    -----
+    - Get
+        - `self.web_socket_url`
+    - Set
+        - `self.web_socket_url` = `value`
+
+    Parameters
+    ----------
+    `value`: `str`
+
+    Returns
+    -------
+    - Get
+        - `str` or `None`
+    - Set
+        - `None`
+    """
+
     page_load_strategy = _PageLoadStrategyDescriptor("pageLoadStrategy")
     """:Gets and Sets page load strategy, the default is "normal".
 

py/test/unit/selenium/webdriver/common/common_options_tests.py

@@ -75,3 +75,23 @@ def test_add_proxy():
 
     assert options.proxy == proxy
     assert caps.get("proxy") == proxy.to_capabilities()
+
+
+def test_default_bidi():
+    options = ArgOptions()
+    assert options.enable_bidi is False
+    assert options.web_socket_url is None
+
+
+def test_enable_bidi():
+    options = ArgOptions()
+    options.enable_bidi = True
+    assert options.enable_bidi is True
+    assert options.web_socket_url is None
+
+
+def test_set_socket_url():
+    options = ArgOptions()
+    options.web_socket_url = 'socket_url'
+    assert options.enable_bidi is True
+    assert options.web_socket_url is 'socket_url'