From 01e8871b54ec71bbbdf97e8189b6264d5a9ff911 Mon Sep 17 00:00:00 2001
From: vadimt <vadimt@google.com>
Date: Tue, 29 Jan 2019 20:00:27 -0800
Subject: [PATCH] Support for a test for pausing an app via Overview.

Bug: 117888000
Change-Id: If9fc76ddfbc756acf39ad5209904d3099c2e4e5f
Tests: on the Nexus side
---
 .../src/com/android/quickstep/views/TaskMenuView.java | 11 +++++++++++
 .../src/com/android/quickstep/views/TaskView.java     |  4 ++++
 tests/src/com/android/launcher3/util/TestUtil.java    |  7 +++++++
 3 files changed, 22 insertions(+)

diff --git a/quickstep/src/com/android/quickstep/views/TaskMenuView.java b/quickstep/src/com/android/quickstep/views/TaskMenuView.java
index bea646a556..682152e9a2 100644
--- a/quickstep/src/com/android/quickstep/views/TaskMenuView.java
+++ b/quickstep/src/com/android/quickstep/views/TaskMenuView.java
@@ -28,6 +28,7 @@ import android.graphics.drawable.Drawable;
 import android.util.AttributeSet;
 import android.view.Gravity;
 import android.view.MotionEvent;
+import android.view.View;
 import android.view.ViewGroup;
 import android.widget.LinearLayout;
 import android.widget.TextView;
@@ -280,4 +281,14 @@ public class TaskMenuView extends AbstractFloatingView {
         Rect toRect = new Rect(0, 0, getWidth(), getHeight());
         return new RoundedRectRevealOutlineProvider(radius, radius, fromRect, toRect);
     }
+
+    public View findMenuItemByText(String text) {
+        for (int i = mOptionLayout.getChildCount() - 1; i >= 0; --i) {
+            final ViewGroup menuOptionView = (ViewGroup) mOptionLayout.getChildAt(i);
+            if (text.equals(menuOptionView.<TextView>findViewById(R.id.text).getText())) {
+                return menuOptionView;
+            }
+        }
+        return null;
+    }
 }
diff --git a/quickstep/src/com/android/quickstep/views/TaskView.java b/quickstep/src/com/android/quickstep/views/TaskView.java
index ad63c248be..027f04b3a7 100644
--- a/quickstep/src/com/android/quickstep/views/TaskView.java
+++ b/quickstep/src/com/android/quickstep/views/TaskView.java
@@ -210,6 +210,10 @@ public class TaskView extends FrameLayout implements PageCallbacks, Reusable {
         mDigitalWellBeingToast = findViewById(R.id.digital_well_being_toast);
     }
 
+    public TaskMenuView getMenuView() {
+        return mMenuView;
+    }
+
     /**
      * Updates this task view to the given {@param task}.
      */
diff --git a/tests/src/com/android/launcher3/util/TestUtil.java b/tests/src/com/android/launcher3/util/TestUtil.java
index 1338dcb108..55e57440c8 100644
--- a/tests/src/com/android/launcher3/util/TestUtil.java
+++ b/tests/src/com/android/launcher3/util/TestUtil.java
@@ -27,6 +27,8 @@ import java.io.IOException;
 import java.io.InputStream;
 
 public class TestUtil {
+    public static final String DUMMY_PACKAGE = "com.example.android.aardwolf";
+
     public static void installDummyApp() throws IOException {
         // Copy apk from resources to a local file and install from there.
         final Resources resources = getContext().getResources();
@@ -48,4 +50,9 @@ public class TestUtil {
 
         UiDevice.getInstance(getInstrumentation()).executeShellCommand("pm install " + apkFilename);
     }
+
+    public static void uninstallDummyApp() throws IOException {
+        UiDevice.getInstance(getInstrumentation()).executeShellCommand(
+                "pm uninstall " + DUMMY_PACKAGE);
+    }
 }