From b9f5602d965c018c8ac1abe2ca276c82b212a09e Mon Sep 17 00:00:00 2001
From: Jon Miranda <jonmiranda@google.com>
Date: Mon, 8 Nov 2021 11:56:55 -0800
Subject: [PATCH] Testing to see if waiting for resume will fix flaky tests.

Similar to how AbstractLauncherUITest waits for resume after
switching state to Normal.

Bug: 202735477
Test: presubmit
Change-Id: I61834d32f8005491a4e14a354dc0097b05267f7b
---
 tests/src/com/android/launcher3/ui/WorkProfileTest.java | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/src/com/android/launcher3/ui/WorkProfileTest.java b/tests/src/com/android/launcher3/ui/WorkProfileTest.java
index aca5951e64..3a338d1b6a 100644
--- a/tests/src/com/android/launcher3/ui/WorkProfileTest.java
+++ b/tests/src/com/android/launcher3/ui/WorkProfileTest.java
@@ -93,6 +93,7 @@ public class WorkProfileTest extends AbstractLauncherUiTest {
         waitForLauncherCondition("Launcher didn't start", Objects::nonNull);
         waitForStateTransitionToEnd("Launcher internal state didn't switch to Normal",
                 () -> NORMAL);
+        waitForResumed("Launcher internal state is still Background");
         executeOnLauncher(launcher -> launcher.getStateManager().goToState(ALL_APPS));
         waitForStateTransitionToEnd("Launcher internal state didn't switch to All Apps",
                 () -> ALL_APPS);