Added locking tests for iOS

Author: Mykola Mokhnach

src/test/java/io/appium/java_client/android/AndroidDriverTest.java

@@ -102,10 +102,13 @@ public class AndroidDriverTest extends BaseAndroidTest {
     }
 
     @Test public void lockTest() {
-        driver.lockDevice();
-        assertEquals(true, driver.isLocked());
-        driver.unlockDevice();
-        assertEquals(false, driver.isLocked());
+        try {
+            driver.lockDevice();
+            assertTrue(driver.isDeviceLocked());
+        } finally {
+            driver.unlockDevice();
+            assertFalse(driver.isDeviceLocked());
+        }
     }
 
     @Test public void runAppInBackgroundTest() {

src/test/java/io/appium/java_client/ios/IOSDriverTest.java

@@ -17,6 +17,7 @@
 package io.appium.java_client.ios;
 
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertTrue;
 
 import io.appium.java_client.MobileElement;
@@ -27,9 +28,6 @@
 import org.openqa.selenium.ScreenOrientation;
 import org.openqa.selenium.html5.Location;
 
-import java.time.Duration;
-import java.util.function.Supplier;
-
 public class IOSDriverTest extends AppIOSTest {
 
     //TODO There is no ability to check this function usibg simulators.
@@ -62,11 +60,13 @@ public void getDeviceTimeTest() {
     }
 
     @Test public void lockTest() {
-        Supplier<Boolean> lock = () -> {
-            driver.lockDevice(Duration.ofSeconds(20));
-            return true;
-        };
-        assertTrue(lock.get());
+        try {
+            driver.lockDevice();
+            assertTrue(driver.isDeviceLocked());
+        } finally {
+            driver.unlockDevice();
+            assertFalse(driver.isDeviceLocked());
+        }
     }
 
     @Test public void pullFileTest() {