Merge pull request #48 from appium/isaac-replacekeys

Add set_text method for Android

Author: jlipps

appium/webdriver/mobilecommand.py

@@ -49,3 +49,4 @@ class MobileCommand(object):
     SHAKE = 'shake'
     RESET = 'reset'
     HIDE_KEYBOARD = 'hideKeyboard'
+    REPLACE_KEYS = 'replaceKeys'

appium/webdriver/webdriver.py

@@ -694,6 +694,8 @@ def _addCommands(self):
             ('POST', '/session/$sessionId/ime/deactivate')
         self.command_executor._commands[Command.GET_ACTIVE_IME_ENGINE] = \
             ('GET', '/session/$sessionId/ime/active_engine')
+        self.command_executor._commands[Command.REPLACE_KEYS] = \
+            ('POST', '/session/$sessionId/appium/element/$elementId/replace_value')
 
 
 # monkeypatched method for WebElement

appium/webdriver/webelement.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from .mobilecommand import MobileCommand as Command
+
 from selenium.webdriver.common.by import By
 from selenium.webdriver.remote.webelement import WebElement as SeleniumWebElement
 
@@ -84,3 +86,20 @@ def find_elements_by_accessibility_id(self, id):
             driver.find_elements_by_accessibility_id()
         """
         return self.find_elements(by=By.ACCESSIBILITY_ID, value=id)
+
+    def set_text(self, keys=''):
+        """Sends text to the element. Previous text is removed.
+        Android only.
+
+        :Args:
+         - keys - the text to be sent to the element.
+
+        :Usage:
+            element.set_text('some text')
+        """
+        data = {
+            'elementId': self._id,
+            'value': [keys]
+        }
+        self._execute(Command.REPLACE_KEYS, data)
+        return self

test/functional/android/appium_tests.py

@@ -173,6 +173,28 @@ def test_open_notifications(self):
         sleep(1)
         self.driver.find_element_by_android_uiautomator('new UiSelector().text(":-|")')
 
+    def test_set_text(self):
+        self.driver.find_element_by_android_uiautomator('new UiScrollable(new UiSelector().scrollable(true).instance(0)).scrollIntoView(new UiSelector().text("Views").instance(0));').click()
+        self.driver.find_element_by_name('Controls').click()
+        self.driver.find_element_by_name('1. Light Theme').click()
+
+        el = self.driver.find_element_by_class_name('android.widget.EditText')
+        el.send_keys('original text')
+        el.set_text('new text')
+
+        self.assertEqual('new text', el.text)
+
+    def test_send_keys(self):
+        self.driver.find_element_by_android_uiautomator('new UiScrollable(new UiSelector().scrollable(true).instance(0)).scrollIntoView(new UiSelector().text("Views").instance(0));').click()
+        self.driver.find_element_by_name('Controls').click()
+        self.driver.find_element_by_name('1. Light Theme').click()
+
+        el = self.driver.find_element_by_class_name('android.widget.EditText')
+        el.send_keys('original text')
+        el.send_keys(' and new text')
+
+        self.assertEqual('original text and new text', el.text)
+
 
 if __name__ == "__main__":
     suite = unittest.TestLoader().loadTestsFromTestCase(AppiumTests)