diff --git a/appium/webdriver/webdriver.py b/appium/webdriver/webdriver.py index e5459335..323a81ea 100644 --- a/appium/webdriver/webdriver.py +++ b/appium/webdriver/webdriver.py @@ -28,6 +28,7 @@ from appium.common.logger import logger from appium.options.common.base import AppiumOptions +from appium.protocols.webdriver.can_find_elements import CanFindElements from appium.webdriver.common.appiumby import AppiumBy from .appium_connection import AppiumConnection @@ -237,6 +238,7 @@ class WebDriver( Settings, Sms, SystemBars, + CanFindElements, ): def __init__( self, diff --git a/appium/webdriver/webelement.py b/appium/webdriver/webelement.py index 52ea4e94..f6b7faa3 100644 --- a/appium/webdriver/webelement.py +++ b/appium/webdriver/webelement.py @@ -19,10 +19,12 @@ from selenium.webdriver.remote.webelement import WebElement as SeleniumWebElement from typing_extensions import Self +from appium.protocols.webdriver.can_find_elements import CanFindElements + from .mobilecommand import MobileCommand as Command -class WebElement(SeleniumWebElement): +class WebElement(SeleniumWebElement, CanFindElements): _execute: Callable _id: str