From b719c6bcf45130c9fb0a7fb96d2496186e5e1b94 Mon Sep 17 00:00:00 2001
From: Dave McCloskey <dlm@google.com>
Date: Wed, 2 Mar 2022 10:08:26 -0800
Subject: [PATCH] When dismissing lockscreen, wait up to 1.5 seconds for it to
 work.

Bug: 221155933
Test: atest TrustTests --iterations 5 showed no failures on acloud
Change-Id: I4c845e0f032052f6de36fd95d4aa131a03e82a8f
---
 .../src/android/trust/test/lib/ScreenLockRule.kt       | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/tests/TrustTests/src/android/trust/test/lib/ScreenLockRule.kt b/tests/TrustTests/src/android/trust/test/lib/ScreenLockRule.kt
index c682a00eb8b98..bc100ba03639e 100644
--- a/tests/TrustTests/src/android/trust/test/lib/ScreenLockRule.kt
+++ b/tests/TrustTests/src/android/trust/test/lib/ScreenLockRule.kt
@@ -59,8 +59,14 @@ class ScreenLockRule : TestRule {
     }
 
     private fun verifyKeyguardDismissed() {
-        windowManager.dismissKeyguard(null, null)
-        Thread.sleep(250)
+        val maxWaits = 30
+        var waitCount = 0
+        while (windowManager.isKeyguardLocked && waitCount < maxWaits) {
+            Log.i(TAG, "Keyguard still showing; attempting to dismiss and wait 50ms ($waitCount)")
+            windowManager.dismissKeyguard(null, null)
+            Thread.sleep(50)
+            waitCount++
+        }
         assertWithMessage("Keyguard should be unlocked")
             .that(windowManager.isKeyguardLocked)
             .isFalse()