Move monkeypatched set_value into WebElement

imurchie · imurchie · commit 4b525c33a941 · 2015-06-04T08:53:31.000-07:00
diff --git a/appium/webdriver/webdriver.py b/appium/webdriver/webdriver.py
@@ -793,15 +793,3 @@ def _addCommands(self):
             ('POST', '/session/$sessionId/location')
         self.command_executor._commands[Command.LOCATION_IN_VIEW] = \
             ('GET', '/session/$sessionId/element/$id/location_in_view')
-
-
-# monkeypatched method for WebElement
-def set_value(self, value):
-    """Set the value on this element in the application
-    """
-    data = {
-        'elementId': self.id,
-        'value': [value],
-    }
-    self._execute(Command.SET_IMMEDIATE_VALUE, data)
-    return self
diff --git a/appium/webdriver/webelement.py b/appium/webdriver/webelement.py
@@ -112,3 +112,13 @@ def location_in_view(self):
             location = element.location_in_view
         """
         return self._execute(Command.LOCATION_IN_VIEW)['value']
+
+    def set_value(self, value):
+        """Set the value on this element in the application
+        """
+        data = {
+            'elementId': self.id,
+            'value': [value],
+        }
+        self._execute(Command.SET_IMMEDIATE_VALUE, data)
+        return self
