commit 1a5df3f1ef7970291e4900ff54b2e5b1ad17d986
author vadimt <vadimt@google.com> Mon Sep 16 17:37:03 2019 -0700
committer android-build-merger <android-build-merger@google.com> Mon Sep 16 17:37:03 2019 -0700
tree bfa0a417c3abc35688988f4af4b50a47b48f46ee
parent fe2e430a711b7ed7f252e188d73040f43f927371
parent 87b871eaee2dd449c3b36f66cc6a074ec3409573

Adding testing for opening a widget am: 5e742772d0
am: 87b871eaee

Change-Id: I8554d43fc23b81bf06df09bedc7a0a26d757d7b9

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 1c99f31..cb586ac 100644
--- a/tests/src/com/android/launcher3/ui/widget/AddWidgetTest.java
+++ b/tests/src/com/android/launcher3/ui/widget/AddWidgetTest.java
@@ -15,6 +15,9 @@
  */
 package com.android.launcher3.ui.widget;
 
+import static com.android.launcher3.ui.TaplTestsLauncher3.getAppPackageName;
+
+import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertTrue;
 
 import androidx.test.filters.LargeTest;
@@ -22,6 +25,7 @@
 
 import com.android.launcher3.LauncherAppWidgetInfo;
 import com.android.launcher3.LauncherAppWidgetProviderInfo;
+import com.android.launcher3.tapl.Widget;
 import com.android.launcher3.ui.AbstractLauncherUiTest;
 import com.android.launcher3.ui.TestViewHelpers;
 import com.android.launcher3.util.rule.ShellCommandRule;
@@ -59,5 +63,10 @@
                 (info, view) -> info instanceof LauncherAppWidgetInfo &&
                         ((LauncherAppWidgetInfo) info).providerName.getClassName().equals(
                                 widgetInfo.provider.getClassName())).call());
+
+        final Widget widget = mLauncher.getWorkspace().tryGetWidget(widgetInfo.label,
+                DEFAULT_UI_TIMEOUT);
+        assertNotNull("Widget not found on the workspace", widget);
+        widget.launch(getAppPackageName());
     }
 }