Merge pull request #333 from hipyard/geolocation_pr

add set_location method for setting geolocation

Author: bootstraponline

android_tests/lib/android/specs/driver.rb

@@ -175,6 +175,17 @@ def validate_path(path)
       end
     end
 
+    # simple integration sanity test to check for unexpected exceptions
+    t 'set_location' do
+      begin
+        set_location latitude: 55, longitude: -72, altitude: 33
+      rescue Selenium::WebDriver::Error::UnknownError => e
+        # on android this method is expected to fail with this message when running
+        # on a regular device, or on genymotion.
+        raise unless e.message.start_with?('ERROR running Appium command: port should be > 0 and < 65536')
+      end
+    end
+
     # settings
     t 'get settings' do
       get_settings.wont_be_nil

ios_tests/lib/ios/specs/driver.rb

@@ -185,6 +185,11 @@ def sauce?
       exists(0, 0) { fail 'error' }.must_equal false
     end
 
+    # simple integration sanity test to check for unexpected exceptions
+    t 'set_location' do
+      set_location latitude: 55, longitude: -72, altitude: 33
+    end
+
     # any script
     t 'execute_script' do
       execute_script %q(au.mainApp().getFirstWithPredicate("name contains[c] 'button'");)

lib/appium_lib/driver.rb

@@ -484,6 +484,7 @@ def start_driver
         @driver = Selenium::WebDriver.for :remote, http_client: @client, desired_capabilities: @caps, url: server_url
         # Load touch methods.
         @driver.extend Selenium::WebDriver::DriverExtensions::HasTouchScreen
+        @driver.extend Selenium::WebDriver::DriverExtensions::HasLocation
 
         # export session
         if @export_session
@@ -592,6 +593,22 @@ def find_element(*args)
       @driver.find_element(*args)
     end
 
+    # Calls @driver.set_location
+    #
+    # @note This method does not work on real devices.
+    #
+    # @param [Hash] opts consisting of:
+    # @option opts [Float] :latitude the latitude in degrees (required)
+    # @option opts [Float] :longitude the longitude in degees (required)
+    # @option opts [Float] :altitude the altitude, defaulting to 75
+    # @return [Selenium::WebDriver::Location] the location constructed by the selenium webdriver
+    def set_location(opts = {})
+      latitude = opts.fetch(:latitude)
+      longitude = opts.fetch(:longitude)
+      altitude = opts.fetch(:altitude, 75)
+      @driver.set_location(latitude, longitude, altitude)
+    end
+
     # Quit the driver and Pry.
     # quit and exit are reserved by Pry.
     # @return [void]