From 4b87b9e5217fcc2c02ef163354880523d3681ec1 Mon Sep 17 00:00:00 2001
From: Hui Kang <huikang@google.com>
Date: Fri, 21 Oct 2022 16:52:47 -0700
Subject: [PATCH] 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 | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/src/com/android/launcher3/ui/widget/AddWidgetTest.java b/tests/src/com/android/launcher3/ui/widget/AddWidgetTest.java
index 561f3ccbf2..1f5590e843 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 @@ public class AddWidgetTest extends AbstractLauncherUiTest {
         clearHomescreen();
         mDevice.pressHome();
 
+        waitForLauncherCondition("Workspace didn't finish loading", l -> !l.isWorkspaceLoading());
+
         final LauncherAppWidgetProviderInfo widgetInfo =
                 TestViewHelpers.findWidgetProvider(this, false /* hasConfigureScreen */);