Fix PIP keyboard test

Test was missing teardown block Bug: 218604389 Test: atest WMShellFlickerTest Change-Id: I2331a16f76b95d31b524c2b714b815a174fbcf04
2022-02-10 13:12:59 +01:00
parent 903315ba69
commit 3bbed27849
1 changed files with 4 additions and 2 deletions
--- a/libs/WindowManager/Shell/tests/flicker/src/com/android/wm/shell/flicker/bubble/BaseBubbleScreen.kt
+++ b/libs/WindowManager/Shell/tests/flicker/src/com/android/wm/shell/flicker/bubble/BaseBubbleScreen.kt
@@ -68,9 +68,11 @@ abstract class BaseBubbleScreen(protected val testSpec: FlickerTestParameter) {
            }

            teardown {
-                notifyManager.setBubblesAllowed(testApp.component.packageName,
+                test {
+                    notifyManager.setBubblesAllowed(testApp.component.packageName,
                        uid, NotificationManager.BUBBLE_PREFERENCE_NONE)
-                testApp.exit()
+                    testApp.exit()
+                }
            }

            extraSpec(this)