diff --git a/tests/tapl/com/android/launcher3/tapl/Workspace.java b/tests/tapl/com/android/launcher3/tapl/Workspace.java index 4fa93ef2ca..9ac6768f42 100644 --- a/tests/tapl/com/android/launcher3/tapl/Workspace.java +++ b/tests/tapl/com/android/launcher3/tapl/Workspace.java @@ -444,6 +444,8 @@ public final class Workspace extends Home { Runnable expectLongClickEvents) { try (LauncherInstrumentation.Closable c = launcher.addContextLayer( "uninstalling app icon")) { + + final String appNameToUninstall = homeAppIcon.getAppName(); dragIconToWorkspace( launcher, homeAppIcon, @@ -468,7 +470,10 @@ public final class Workspace extends Home { try (LauncherInstrumentation.Closable c1 = launcher.addContextLayer( "uninstalled app by dragging to the drop bar")) { - return new Workspace(launcher); + final Workspace newWorkspace = new Workspace(launcher); + launcher.waitUntilLauncherObjectGone( + AppIcon.getAppIconSelector(appNameToUninstall)); + return newWorkspace; } } }