From 0cc86f204c1401fbd46fc3a178066741668c29b3 Mon Sep 17 00:00:00 2001
From: Nataniel Borges <natanieljr@google.com>
Date: Mon, 25 Jul 2022 15:07:35 +0000
Subject: [PATCH] Fix task transition test

Test: atest FlickerTests
Change-Id: Id1a6ddecec1d4602fd89545676f63be4bcdfa0e4
---
 .../com/android/server/wm/flicker/helpers/NewTasksAppHelper.kt  | 2 +-
 .../com/android/server/wm/flicker/launch/TaskTransitionTest.kt  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/FlickerTests/src/com/android/server/wm/flicker/helpers/NewTasksAppHelper.kt b/tests/FlickerTests/src/com/android/server/wm/flicker/helpers/NewTasksAppHelper.kt
index 79656a0bef9e2..149576e4ec1df 100644
--- 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()
     }
 }
diff --git a/tests/FlickerTests/src/com/android/server/wm/flicker/launch/TaskTransitionTest.kt b/tests/FlickerTests/src/com/android/server/wm/flicker/launch/TaskTransitionTest.kt
index 4beb5d0199576..b482e5f3c3fbf 100644
--- 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()
         }
     }