Merge "CTS Keyguard*Tests don't assume screenlock" into oc-dev
diff --git a/hostsidetests/services/activityandwindowmanager/activitymanager/src/android/server/cts/KeyguardTests.java b/hostsidetests/services/activityandwindowmanager/activitymanager/src/android/server/cts/KeyguardTests.java
index a6a7cd8..89bbb07 100644
--- a/hostsidetests/services/activityandwindowmanager/activitymanager/src/android/server/cts/KeyguardTests.java
+++ b/hostsidetests/services/activityandwindowmanager/activitymanager/src/android/server/cts/KeyguardTests.java
@@ -25,6 +25,22 @@
  */
 public class KeyguardTests extends KeyguardTestBase {
 
+    @Override
+    public void setUp() throws Exception {
+        super.setUp();
+
+        // Set screen lock (swipe)
+        mDevice.executeShellCommand("locksettings set-disabled false");
+    }
+
+    @Override
+    public void tearDown() throws Exception {
+        super.tearDown();
+
+        // Remove screen lock
+        mDevice.executeShellCommand("locksettings set-disabled true");
+    }
+
     public void testKeyguardHidesActivity() throws Exception {
         if (!isHandheld()) {
             return;
diff --git a/hostsidetests/services/activityandwindowmanager/activitymanager/src/android/server/cts/KeyguardTransitionTests.java b/hostsidetests/services/activityandwindowmanager/activitymanager/src/android/server/cts/KeyguardTransitionTests.java
index e67cdd9..05065aa 100644
--- a/hostsidetests/services/activityandwindowmanager/activitymanager/src/android/server/cts/KeyguardTransitionTests.java
+++ b/hostsidetests/services/activityandwindowmanager/activitymanager/src/android/server/cts/KeyguardTransitionTests.java
@@ -28,6 +28,22 @@
  */
 public class KeyguardTransitionTests extends ActivityManagerTestBase {
 
+    @Override
+    public void setUp() throws Exception {
+        super.setUp();
+
+        // Set screen lock (swipe)
+        mDevice.executeShellCommand("locksettings set-disabled false");
+    }
+
+    @Override
+    public void tearDown() throws Exception {
+        super.tearDown();
+
+        // Remove screen lock
+        mDevice.executeShellCommand("locksettings set-disabled true");
+    }
+
     public void testUnlock() throws Exception {
         if (!isHandheld()) {
             return;