Fix task transition test

Test: atest FlickerTests Change-Id: Id1a6ddecec1d4602fd89545676f63be4bcdfa0e4
2022-07-25 15:07:35 +00:00
parent ae34fbc353
commit 0cc86f204c
2 changed files with 2 additions and 2 deletions
--- a/tests/FlickerTests/src/com/android/server/wm/flicker/helpers/NewTasksAppHelper.kt
+++ b/tests/FlickerTests/src/com/android/server/wm/flicker/helpers/NewTasksAppHelper.kt
@@ -47,7 +47,7 @@ class NewTasksAppHelper @JvmOverloads constructor(
        }
        button.click()
        wmHelper.StateSyncBuilder()
-            .withFullScreenApp(this)
+            .withAppTransitionIdle()
            .waitForAndVerify()
    }
 }
--- a/tests/FlickerTests/src/com/android/server/wm/flicker/launch/TaskTransitionTest.kt
+++ b/tests/FlickerTests/src/com/android/server/wm/flicker/launch/TaskTransitionTest.kt
@@ -78,7 +78,7 @@ class TaskTransitionTest(testSpec: FlickerTestParameter) : BaseTest(testSpec) {
            testApp.openNewTask(device, wmHelper)
            tapl.pressBack()
            wmHelper.StateSyncBuilder()
-                .withFullScreenApp(testApp)
+                .withAppTransitionIdle()
                .waitForAndVerify()
        }
    }