Merge "Fix flaky assertion"

2021-12-03 20:08:02 +00:00
parent b04ce6d679 cd35590771
commit 22e57705c5
1 changed files with 4 additions and 0 deletions
--- a/tests/FlickerTests/src/com/android/server/wm/flicker/ime/SwitchImeWindowsFromGestureNavTest.kt
+++ b/tests/FlickerTests/src/com/android/server/wm/flicker/ime/SwitchImeWindowsFromGestureNavTest.kt
@@ -107,9 +107,13 @@ class SwitchImeWindowsFromGestureNavTest(private val testSpec: FlickerTestParame
    fun imeAppWindowVisibility() {
        testSpec.assertWm {
            isAppWindowVisible(imeTestApp.component)
+                .then()
+                .isAppSnapshotStartingWindowVisibleFor(testApp.component, isOptional = true)
                .then()
                .isAppWindowVisible(testApp.component)
                .then()
+                .isAppSnapshotStartingWindowVisibleFor(imeTestApp.component, isOptional = true)
+                .then()
                .isAppWindowVisible(imeTestApp.component)
        }
    }