Add toggle_location_services

Author: imurchie

appium/webdriver/mobilecommand.py

@@ -27,6 +27,7 @@ class MobileCommand(object):
     ACTIVATE_IME_ENGINE = 'activateIMEEngine'
     DEACTIVATE_IME_ENGINE = 'deactivateIMEEngine'
     GET_ACTIVE_IME_ENGINE = 'getActiveEngine'
+    TOGGLE_LOCATION_SERVICES = 'toggleLocationServices'
 
     # Appium Commands
     GET_APP_STRINGS = 'getAppStrings'

appium/webdriver/webdriver.py

@@ -672,6 +672,13 @@ def update_settings(self, settings):
         self.execute(Command.UPDATE_SETTINGS, data)
         return self
 
+    def toggle_location_services(self):
+        """Toggle the location services on the device. Android only.
+        """
+        self.execute(Command.TOGGLE_LOCATION_SERVICES, {})
+        return self
+
+
     def _addCommands(self):
         self.command_executor._commands[Command.CONTEXTS] = \
             ('GET', '/session/$sessionId/contexts')
@@ -748,6 +755,8 @@ def _addCommands(self):
             ('GET', '/session/$sessionId/appium/settings')
         self.command_executor._commands[Command.UPDATE_SETTINGS] = \
             ('POST', '/session/$sessionId/appium/settings')
+        self.command_executor._commands[Command.TOGGLE_LOCATION_SERVICES] = \
+            ('POST', '/session/$sessionId/appium/device/toggle_location_services')
 
 
 # monkeypatched method for WebElement

test/functional/android/appium_tests.py

@@ -39,7 +39,7 @@ def tearDown(self):
         self.driver.quit()
 
         # remove zipped file from `test_pull_folder`
-        if os.path.isfile(self.zipfilename):
+        if hasattr(self, 'zipfilename') and os.path.isfile(self.zipfilename):
             os.remove(self.zipfilename)
 
     def test_app_strings(self):
@@ -217,6 +217,10 @@ def test_update_settings(self):
         settings = self.driver.get_settings()
         self.assertEqual(settings["cyberdelia"], "open")
 
+    def test_toggle_location_services(self):
+        self.driver.toggle_location_services()
+
+
 if __name__ == "__main__":
     suite = unittest.TestLoader().loadTestsFromTestCase(AppiumTests)
     unittest.TextTestRunner(verbosity=2).run(suite)

test/functional/android/desired_capabilities.py

@@ -26,6 +26,7 @@ def get_desired_capabilities(app):
         'platformVersion': '4.2',
         'deviceName': 'Android Emulator',
         'app': PATH('../../apps/' + app),
+        'newCommandTimeout': 240
     }
 
     return desired_caps