Add new locator strategy find_element_by_windows_uiautomation

manoj9788 · manoj9788 · commit b6c35035652d · 2019-11-02T11:40:05.000+05:30
diff --git a/appium/webdriver/common/mobileby.py b/appium/webdriver/common/mobileby.py
@@ -22,6 +22,7 @@ class MobileBy(By):
     ANDROID_UIAUTOMATOR = '-android uiautomator'
     ANDROID_VIEWTAG = '-android viewtag'
     ANDROID_DATA_MATCHER = '-android datamatcher'
+    WINDOWS_UI_AUTOMATION = '-windows uiautomation'
     ACCESSIBILITY_ID = 'accessibility id'
     IMAGE = '-image'
     CUSTOM = '-custom'
diff --git a/appium/webdriver/webdriver.py b/appium/webdriver/webdriver.py
@@ -159,6 +159,7 @@ def __init__(self, command_executor='http://127.0.0.1:4444/wd/hub',
         By.IOS_CLASS_CHAIN = MobileBy.IOS_CLASS_CHAIN
         By.ANDROID_UIAUTOMATOR = MobileBy.ANDROID_UIAUTOMATOR
         By.ANDROID_VIEWTAG = MobileBy.ANDROID_VIEWTAG
+        By.WINDOWS_UI_AUTOMATION = MobileBy.WINDOWS_UI_AUTOMATION
         By.ACCESSIBILITY_ID = MobileBy.ACCESSIBILITY_ID
         By.IMAGE = MobileBy.IMAGE
         By.CUSTOM = MobileBy.CUSTOM
diff --git a/appium/webdriver/webelement.py b/appium/webdriver/webelement.py
@@ -216,6 +216,20 @@ def find_elements_by_accessibility_id(self, accessibility_id):
         """
         return self.find_elements(by=MobileBy.ACCESSIBILITY_ID, value=accessibility_id)
 
+    def find_element_by_windows_uiautomation(self, win_uiautomation):
+        """Finds an element by windows uiautomation
+
+        Args:
+            win_uiautomation (str): The element name in the windows UIAutomation selector
+
+        Usage:
+            driver.find_element_by_windows_uiautomation()
+
+        Returns:
+            `appium.webdriver.webelement.WebElement`
+        """
+        return self.find_element(by=MobileBy.WINDOWS_UI_AUTOMATION, value=win_uiautomation)
+
     def find_element(self, by=By.ID, value=None):
         """Find an element given a By strategy and locator
 
