From dfa379a54408b1e2314bf8c3d12ca5403cbc587a Mon Sep 17 00:00:00 2001
From: Vincent Wang <firewall@google.com>
Date: Mon, 8 Nov 2021 15:39:45 +0800
Subject: [PATCH] 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
---
 .../wm/shell/flicker/bubble/LaunchBubbleFromLockScreen.kt       | 2 ++
 1 file changed, 2 insertions(+)

diff --git 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
index 9c172a237bfcc..bbdf997e0ee43 100644
--- 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 {