From 652037296640c3b131405b8f5672232f3b02c5a4 Mon Sep 17 00:00:00 2001
From: Ke-Yu Lu <lukeyu@google.com>
Date: Tue, 23 Aug 2022 12:54:01 +0800
Subject: [PATCH] Add getIconName to AppIcon

Bug: 237606436
Test: manual on local device
Change-Id: Ifcd18cefa35148fd744087a207e48791c098899a
---
 .../com/android/launcher3/ui/TaplTestsLauncher3.java | 12 ++++++++++++
 tests/tapl/com/android/launcher3/tapl/AppIcon.java   |  7 +++++++
 2 files changed, 19 insertions(+)

diff --git a/tests/src/com/android/launcher3/ui/TaplTestsLauncher3.java b/tests/src/com/android/launcher3/ui/TaplTestsLauncher3.java
index 03bf4af3de..ce211c1d43 100644
--- a/tests/src/com/android/launcher3/ui/TaplTestsLauncher3.java
+++ b/tests/src/com/android/launcher3/ui/TaplTestsLauncher3.java
@@ -584,4 +584,16 @@ public class TaplTestsLauncher3 extends AbstractLauncherUiTest {
     public static String getAppPackageName() {
         return getInstrumentation().getContext().getPackageName();
     }
+
+    @Test
+    public void testGetAppIconName() {
+        HomeAllApps allApps = mLauncher.getWorkspace().switchToAllApps();
+        allApps.freeze();
+        try {
+            HomeAppIcon icon = allApps.getAppIcon(APP_NAME);
+            assertEquals("Wrong app icon name.", icon.getIconName(), APP_NAME);
+        } finally {
+            allApps.unfreeze();
+        }
+    }
 }
diff --git a/tests/tapl/com/android/launcher3/tapl/AppIcon.java b/tests/tapl/com/android/launcher3/tapl/AppIcon.java
index d22125971c..2687b28bcb 100644
--- a/tests/tapl/com/android/launcher3/tapl/AppIcon.java
+++ b/tests/tapl/com/android/launcher3/tapl/AppIcon.java
@@ -18,6 +18,7 @@ package com.android.launcher3.tapl;
 
 import android.widget.TextView;
 
+import androidx.annotation.NonNull;
 import androidx.test.uiautomator.By;
 import androidx.test.uiautomator.BySelector;
 import androidx.test.uiautomator.UiObject2;
@@ -86,4 +87,10 @@ public abstract class AppIcon extends Launchable {
     protected String launchableType() {
         return "app icon";
     }
+
+    /** Return the app name of a icon */
+    @NonNull
+    public String getIconName() {
+        return getObject().getText();
+    }
 }