Add resume check to other WorkProfileTests to fix flakiness.

- Tests were flaky with waitForStateTransitionToEnd
- Tests stopped flaking with both waitForResume and waitForStateTransitionToEnd
- waitForResume may be enough to fix the flakiness, so
  removing waitForStateTransitionToEnd for now, if flakiness
  returns we can add it back

Bug: 202735477
Test: presubmit
Change-Id: Ib9cc16ea7815387582486517a87249d0053475f6

tests/src/com/android/launcher3/ui/AbstractLauncherUiTest.java

@@ -352,14 +352,6 @@ public abstract class AbstractLauncherUiTest {
                 launcher -> launcher.getStateManager().getCurrentStableState() == state.get());
     }
 
-    // Cannot be used in TaplTests between a Tapl call injecting a gesture and a tapl call
-    // expecting the results of that gesture because the wait can hide flakeness.
-    protected void waitForStateTransitionToEnd(String message, Supplier<LauncherState> state) {
-        waitForLauncherCondition(message,
-                launcher -> launcher.getStateManager().isInStableState(state.get())
-                        && !launcher.getStateManager().isInTransition());
-    }
-
     protected void waitForResumed(String message) {
         waitForLauncherCondition(message, launcher -> launcher.hasBeenResumed());
     }