From 9016e17988509705f9c3569fc668774e92640fa1 Mon Sep 17 00:00:00 2001
From: Alex Chau <alexchau@google.com>
Date: Mon, 2 Aug 2021 09:39:11 +0100
Subject: [PATCH] Add ScreenRecord for testDragCustomShortcut

- For investigation of a failing test

Bug: 195263971
Test: manual
Change-Id: If68e678a847f70c5b891c9dbcece77333fedf846
---
 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 dad4f2b045..2e8048a846 100644
--- a/tests/src/com/android/launcher3/ui/widget/AddWidgetTest.java
+++ b/tests/src/com/android/launcher3/ui/widget/AddWidgetTest.java
@@ -27,6 +27,7 @@ import com.android.launcher3.model.data.LauncherAppWidgetInfo;
 import com.android.launcher3.tapl.Widget;
 import com.android.launcher3.ui.AbstractLauncherUiTest;
 import com.android.launcher3.ui.TestViewHelpers;
+import com.android.launcher3.util.rule.ScreenRecordRule.ScreenRecord;
 import com.android.launcher3.util.rule.ShellCommandRule;
 import com.android.launcher3.widget.LauncherAppWidgetProviderInfo;
 
@@ -80,6 +81,7 @@ public class AddWidgetTest extends AbstractLauncherUiTest {
      */
     @Test
     @PortraitLandscape
+    @ScreenRecord //b/195263971
     public void testDragCustomShortcut() throws Throwable {
         clearHomescreen();
         mDevice.pressHome();