Add handlers for lock/unlock in iOS

Author: Mykola Mokhnach

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

@@ -352,6 +352,28 @@ public static ImmutableMap<String, Object> prepareArguments(String[] params,
                 LOCK, prepareArguments("seconds", duration.getSeconds()));
     }
 
+    /**
+     * This method forms a {@link java.util.Map} of parameters for the
+     * device unlocking.
+     *
+     * @return  a key-value pair. The key is the command name. The value is a
+     * {@link java.util.Map} command arguments.
+     */
+    public static Map.Entry<String, Map<String, ?>> unlockDeviceCommand() {
+        return new AbstractMap.SimpleEntry<>(UNLOCK, ImmutableMap.<String, Object>of());
+    }
+
+    /**
+     * This method forms a {@link java.util.Map} of parameters for the
+     * device locked query.
+     *
+     * @return  a key-value pair. The key is the command name. The value is a
+     * {@link java.util.Map} command arguments.
+     */
+    public static Map.Entry<String, Map<String, ?>> getIsDeviceLockedCommand() {
+        return new AbstractMap.SimpleEntry<>(IS_LOCKED, ImmutableMap.<String, Object>of());
+    }
+
     public static Map.Entry<String, Map<String, ?>> getSettingsCommand() {
         return new AbstractMap.SimpleEntry<>(GET_SETTINGS, ImmutableMap.<String, Object>of());
     }

src/main/java/io/appium/java_client/ios/LocksIOSDevice.java

@@ -16,7 +16,9 @@
 
 package io.appium.java_client.ios;
 
+import static io.appium.java_client.MobileCommand.getIsDeviceLockedCommand;
 import static io.appium.java_client.MobileCommand.lockDeviceCommand;
+import static io.appium.java_client.MobileCommand.unlockDeviceCommand;
 
 import io.appium.java_client.CommandExecutionHelper;
 import io.appium.java_client.ExecutesMethod;
@@ -27,11 +29,30 @@ public interface LocksIOSDevice extends ExecutesMethod {
 
     /**
      * Lock the device (bring it to the lock screen) for a given number of
-     * seconds.
+     * seconds or forever (until the command for unlocking is called). The call
+     * is ignored if the device already locked.
      *
-     * @param duration for how long to lock the screen. Minimum time resolution is one second
+     * @param duration for how long to lock the screen. Minimum time resolution is one second.
+     *                 A negative/zero value will lock the device and return immediately.
      */
     default void lockDevice(Duration duration) {
         CommandExecutionHelper.execute(this, lockDeviceCommand(duration));
     }
+
+    /**
+     * Unlocks the device if it is locked or returns immediately if the device is already unlocked.
+     */
+    default void unlockDevice() {
+        CommandExecutionHelper.execute(this, unlockDeviceCommand());
+    }
+
+    /**
+     * Get the current state of the device.
+     *
+     * @return true if the device is locked or false otherwise.
+     */
+    default boolean isDeviceLocked() {
+        return CommandExecutionHelper.execute(this, getIsDeviceLockedCommand());
+    }
+
 }