java json converter chooses Long (Number base class). Fixing RemoteNe…

…tworkConnection
Jonahss · May 6, 2014 · 30efd0c · 30efd0c
1 parent 72e85ca
commit 30efd0c
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/java/client/src/org/openqa/selenium/remote/mobile/RemoteNetworkConnection.java b/java/client/src/org/openqa/selenium/remote/mobile/RemoteNetworkConnection.java
@@ -34,15 +34,17 @@ public RemoteNetworkConnection(ExecuteMethod executeMethod) {
 
   @Override
   public ConnectionType getNetworkConnection() {
-    return new ConnectionType(
-        (Integer) executeMethod.execute(DriverCommand.GET_NETWORK_CONNECTION, null));
+    return new ConnectionType(((Number) executeMethod.execute(DriverCommand.GET_NETWORK_CONNECTION,
+                                                               null)).intValue());
   }
 
   @Override
   public ConnectionType setNetworkConnection(
       ConnectionType type) {
     Map<String, ConnectionType> mode = ImmutableMap.of("type", type);
-    return new ConnectionType((Integer) executeMethod
-        .execute(DriverCommand.SET_NETWORK_CONNECTION, ImmutableMap.of("parameters", mode)));
+    return new ConnectionType(((Number) executeMethod.execute(DriverCommand.SET_NETWORK_CONNECTION,
+                                                            ImmutableMap
+                                                                .of("parameters", mode)))
+                                  .intValue());
   }
 }