From 55fa21eb9562820e885143ef52c5ce76b204e0e4 Mon Sep 17 00:00:00 2001
From: vadimt <vadimt@google.com>
Date: Thu, 25 Apr 2019 17:29:05 -0700
Subject: [PATCH] Annotating DigitalWellBeingToastTest so that it actually runs
 in lab :)

And starting a test app :)))))

Change-Id: I3b326a40651b0e4293dee26e4ef5f991cd7b7d0d
---
 .../com/android/quickstep/DigitalWellBeingToastTest.java  | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/quickstep/tests/src/com/android/quickstep/DigitalWellBeingToastTest.java b/quickstep/tests/src/com/android/quickstep/DigitalWellBeingToastTest.java
index 87981571bd..70f9c90dad 100644
--- a/quickstep/tests/src/com/android/quickstep/DigitalWellBeingToastTest.java
+++ b/quickstep/tests/src/com/android/quickstep/DigitalWellBeingToastTest.java
@@ -12,21 +12,29 @@ import android.app.PendingIntent;
 import android.app.usage.UsageStatsManager;
 import android.content.Intent;
 
+import androidx.test.filters.LargeTest;
+import androidx.test.runner.AndroidJUnit4;
+
 import com.android.launcher3.Launcher;
 import com.android.quickstep.views.DigitalWellBeingToast;
 import com.android.quickstep.views.RecentsView;
 import com.android.quickstep.views.TaskView;
 
 import org.junit.Test;
+import org.junit.runner.RunWith;
 
 import java.time.Duration;
 
+@LargeTest
+@RunWith(AndroidJUnit4.class)
 public class DigitalWellBeingToastTest extends AbstractQuickStepTest {
     private static final String CALCULATOR_PACKAGE =
             resolveSystemApp(Intent.CATEGORY_APP_CALCULATOR);
 
     @Test
     public void testToast() throws Exception {
+        startAppFast(CALCULATOR_PACKAGE);
+
         final UsageStatsManager usageStatsManager =
                 mTargetContext.getSystemService(UsageStatsManager.class);
         final int observerId = 0;