Fix Bubble test error problem

Add entering lock screen logic back into LaunchBubbleFromLockScreen.kt to prvent widget not found problem. Test: atest WMShellFlickerTests:LaunchBubbleFromLockScreen Bug: 205288792 Change-Id: I7a644fd5b805f304f3755771eb36976588eb920a
2021-11-08 15:39:45 +08:00
parent 22b158d0c4
commit dfa379a544
1 changed files with 2 additions and 0 deletions
--- a/libs/WindowManager/Shell/tests/flicker/src/com/android/wm/shell/flicker/bubble/LaunchBubbleFromLockScreen.kt
+++ b/libs/WindowManager/Shell/tests/flicker/src/com/android/wm/shell/flicker/bubble/LaunchBubbleFromLockScreen.kt
@@ -48,9 +48,11 @@ class LaunchBubbleFromLockScreen(testSpec: FlickerTestParameter) : BaseBubbleScr
                eachRun {
                    val addBubbleBtn = waitAndGetAddBubbleBtn()
                    addBubbleBtn?.click() ?: error("Bubble widget not found")
+                    device.sleep()
                    wmHelper.waitFor("noAppWindowsOnTop") {
                        it.wmState.topVisibleAppWindow.isEmpty()
                    }
+                    device.wakeUp()
                }
            }
            transitions {