Add new locator strategy find_elements_by_windows_uiautomation and test

manoj9788 · manoj9788 · commit 6911f7dbd69c · 2019-11-02T12:21:21.000+05:30
diff --git a/appium/webdriver/webelement.py b/appium/webdriver/webelement.py
@@ -230,6 +230,20 @@ def find_element_by_windows_uiautomation(self, win_uiautomation):
         """
         return self.find_element(by=MobileBy.WINDOWS_UI_AUTOMATION, value=win_uiautomation)
 
+    def find_elements_by_windows_uiautomation(self, win_uiautomation):
+        """Finds elements by windows uiautomation
+
+        Args:
+            win_uiautomation (str): The element name in the windows UIAutomation selector
+
+        Usage:
+            driver.find_elements_by_windows_uiautomation()
+
+        Returns:
+            :obj:`list` of :obj:`appium.webdriver.webelement.WebElement`
+        """
+        return self.find_elements(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
 
diff --git a/test/unit/webdriver/webelement_test.py b/test/unit/webdriver/webelement_test.py
@@ -80,3 +80,18 @@ def test_find_elements_by_android_data_matcher_no_value(self):
         assert d['using'] == '-android datamatcher'
         assert d['value'] == '{}'
         assert len(els) == 0
+
+    @httpretty.activate
+    def test_find_element_by_windows_uiautomation(self):
+        driver = android_w3c_driver()
+        element = MobileWebElement(driver, 'element_id', w3c=True)
+        httpretty.register_uri(
+            httpretty.POST,
+            appium_command('/session/1234567890/element/element_id/element'),
+            body='{"value": {"element-6066-11e4-a52e-4f735466cecf": "win-element-id"}}'
+        )
+        el = element.find_element_by_windows_uiautomation('win_element')
+
+        d = get_httpretty_request_body(httpretty.last_request())
+        assert d['using'] == '-windows uiautomation'
+        assert el.id == 'win-element-id'
