Wait IME visible after launch

The ImeAppAutoFocus shows the IME when the app is launched. at the moment it doesn't wait for the IME to be visible before continuing and this causes test flakiness

Test: atest FlickerTests
Change-Id: If3ab191fcc9d026553a1859e594e4d38da37f41d

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