Support RelativeBy in type annotations

Author: gryznar

py/docs/source/index.rst

@@ -16,9 +16,12 @@
 # under the License.
 
 from hashlib import md5 as md5_hash
+from typing import List, Optional, Union
 
-from ..common.by import By
+from ..common.by import By, ByType
 from .command import Command
+from .webelement import WebElement
+from ..support.relative_locator import RelativeBy
 
 
 class ShadowRoot:
@@ -39,12 +42,16 @@ def __repr__(self) -> str:
             type(self), self.session.session_id, self._id
         )
 
-    def find_element(self, by: str = By.ID, value: str = None):
+    @property
+    def id(self) -> str:
+        return self._id
+
+    def find_element(self, by: Union[ByType, RelativeBy] = By.ID, value: Optional[str] = None) -> WebElement:
         """Find an element inside a shadow root given a By strategy and
         locator.
 
         Parameters:
-        ----------
+        -----------
         by : selenium.webdriver.common.by.By
             The locating strategy to use. Default is `By.ID`. Supported values include:
             - By.ID: Locate by element ID.
@@ -78,11 +85,11 @@ def find_element(self, by: str = By.ID, value: str = None):
 
         return self._execute(Command.FIND_ELEMENT_FROM_SHADOW_ROOT, {"using": by, "value": value})["value"]
 
-    def find_elements(self, by: str = By.ID, value: str = None):
+    def find_elements(self, by: Union[ByType, RelativeBy] = By.ID, value: Optional[str] = None) -> List[WebElement]:
         """Find elements inside a shadow root given a By strategy and locator.
 
         Parameters:
-        ----------
+        -----------
         by : selenium.webdriver.common.by.By
             The locating strategy to use. Default is `By.ID`. Supported values include:
             - By.ID: Locate by element ID.
@@ -97,11 +104,11 @@ def find_elements(self, by: str = By.ID, value: str = None):
 
         Example:
         --------
-        element = driver.find_element(By.ID, 'foo')
+        element = driver.find_elements(By.ID, 'foo')
 
         Returns:
         -------
-        WebElement
+        List[WebElement]
             list of `WebElements` matching locator strategy found on the page.
         """
         if by == By.ID: