AppiumDriver now implements LocationContext

Author: Jonahss

src/main/java/io/appium/java_client/AppiumDriver.java

@@ -21,7 +21,10 @@
 import com.google.common.collect.ImmutableMap;
 import io.appium.java_client.internal.JsonToMobileElementConverter;
 import org.openqa.selenium.*;
+import org.openqa.selenium.html5.Location;
+import org.openqa.selenium.html5.LocationContext;
 import org.openqa.selenium.remote.*;
+import org.openqa.selenium.remote.html5.RemoteLocationContext;
 
 import javax.xml.bind.DatatypeConverter;
 import java.net.URL;
@@ -33,19 +36,23 @@
 import static io.appium.java_client.MobileCommand.*;
 
 public class AppiumDriver extends RemoteWebDriver implements MobileDriver, ContextAware, Rotatable, FindsByIosUIAutomation,
-        FindsByAndroidUIAutomator, FindsByAccessibilityId {
+        FindsByAndroidUIAutomator, FindsByAccessibilityId, LocationContext {
 
   private final static ErrorHandler errorHandler = new ErrorHandler(new ErrorCodesMobile(), true);
   private URL remoteAddress;
   private ComplexFind complexFind;
+  private RemoteLocationContext locationContext;
+  private ExecuteMethod executeMethod;
 
   public AppiumDriver(URL remoteAddress, Capabilities desiredCapabilities){
 
     super(remoteAddress, desiredCapabilities);
     this.setElementConverter(new JsonToMobileElementConverter(this));
 
+    this.executeMethod = new AppiumExecutionMethod(this);
     this.remoteAddress = remoteAddress;
     complexFind = new ComplexFind(this);
+    locationContext = new RemoteLocationContext(executeMethod);
 
     ImmutableMap.Builder<String, CommandInfo> builder = ImmutableMap.builder();
     builder
@@ -93,6 +100,10 @@ protected Response execute(String command) {
     return execute(command, ImmutableMap.<String, Object>of());
   }
 
+  @Override
+  public ExecuteMethod getExecuteMethod() {
+    return executeMethod;
+  }
 
   /**
    * Reset the currently running app for this session
@@ -626,6 +637,16 @@ public List<WebElement> findElementsByAccessibilityId(String using) {
     return findElements("accessibility id", using);
   }
 
+  @Override
+  public Location location() {
+    return locationContext.location();
+  }
+
+  @Override
+  public void setLocation(Location location) {
+    locationContext.setLocation(location);
+  }
+
   private TouchAction createTap(WebElement element, int duration) {
     TouchAction tap = new TouchAction(this);
     return tap.press(element).waitAction(duration).release();

src/main/java/io/appium/java_client/AppiumExecutionMethod.java

@@ -0,0 +1,27 @@
+package io.appium.java_client;
+
+import com.google.common.collect.ImmutableMap;
+import org.openqa.selenium.remote.ExecuteMethod;
+import org.openqa.selenium.remote.Response;
+
+import java.util.Map;
+
+public class AppiumExecutionMethod implements ExecuteMethod {
+  private final AppiumDriver driver;
+
+  public AppiumExecutionMethod(AppiumDriver driver) {
+    this.driver = driver;
+  }
+
+  public Object execute(String commandName, Map<String, ?> parameters) {
+    Response response;
+
+    if (parameters == null || parameters.isEmpty()) {
+      response = driver.execute(commandName, ImmutableMap.<String, Object>of());
+    } else {
+      response = driver.execute(commandName, parameters);
+    }
+
+    return response.getValue();
+  }
+}

src/test/java/io/appium/java_client/MobileDriverIOSTest.java

@@ -23,6 +23,7 @@
 import org.junit.Before;
 import org.junit.Test;
 import org.openqa.selenium.ScreenOrientation;
+import org.openqa.selenium.html5.Location;
 import org.openqa.selenium.remote.DesiredCapabilities;
 
 import java.io.File;
@@ -130,4 +131,10 @@ public void orientationTest() {
     assertEquals(ScreenOrientation.LANDSCAPE, driver.getOrientation());
   }
 
+  @Test
+  public void geolocationTest() {
+    Location location = new Location(45, 45, 100);
+    driver.setLocation(location);
+  }
+
 }