commit 4b87b9e5217fcc2c02ef163354880523d3681ec1
author Hui Kang <huikang@google.com> Fri Oct 21 16:52:47 2022 -0700
committer Hui Kang <huikang@google.com> Mon Oct 24 18:09:15 2022 +0000
tree 2d0ea22fb8e81c7f73418d50cd702d2d89435ea4
parent f34811d3a3c1969a45da452fe8ba6ff336c6e9a4

Fix AddWidgetTest flake
Test: ABTD

200 all PASS: https://android-build.googleplex.com/builds/tests/view?invocationId=I20300010100881398

Without this CL, 1/100 FAIL: https://android-build.googleplex.com/builds/abtd/run/L73700000956982533

Change-Id: Ia9d7746144e9838798332f4eea223a9529551713

tests/src/com/android/launcher3/ui/widget/AddWidgetTest.java

diff --git a/tests/src/com/android/launcher3/ui/widget/AddWidgetTest.java b/tests/src/com/android/launcher3/ui/widget/AddWidgetTest.java
index 561f3cc..1f5590e 100644
--- a/tests/src/com/android/launcher3/ui/widget/AddWidgetTest.java
+++ b/tests/src/com/android/launcher3/ui/widget/AddWidgetTest.java
@@ -54,6 +54,8 @@
         clearHomescreen();
         mDevice.pressHome();
 
+        waitForLauncherCondition("Workspace didn't finish loading", l -> !l.isWorkspaceLoading());
+
         final LauncherAppWidgetProviderInfo widgetInfo =
                 TestViewHelpers.findWidgetProvider(this, false /* hasConfigureScreen */);