Merge "Wait IME visible after launch" into tm-dev

2022-04-20 06:28:08 +00:00
parent fc5c5e9a3f b358224e5b
commit 56d66c7dd3
1 changed files with 10 additions and 0 deletions
--- a/tests/FlickerTests/src/com/android/server/wm/flicker/helpers/ImeAppAutoFocusHelper.kt
+++ b/tests/FlickerTests/src/com/android/server/wm/flicker/helpers/ImeAppAutoFocusHelper.kt
@@ -41,6 +41,16 @@ class ImeAppAutoFocusHelper @JvmOverloads constructor(
        waitIMEShown(device, wmHelper)
    }

+    override fun launchViaIntent(
+        wmHelper: WindowManagerStateHelper,
+        expectedWindowName: String,
+        action: String?,
+        stringExtras: Map<String, String>
+    ) {
+        super.launchViaIntent(wmHelper, expectedWindowName, action, stringExtras)
+        waitIMEShown(uiDevice, wmHelper)
+    }
+
    override fun open() {
        val expectedPackage = if (rotation.isRotated()) {
            imePackageName