Add TopTaskTrackerTest

Fix: 406325205 Test: TopTaskTrackerTest Flag: EXEMPT adding test Change-Id: Ia4e9de25ff7d19f952e5229edf0b08480c109860
2026-02-20 11:18:21 +00:00 · 2025-06-05 00:38:55 +00:00
parent bd029b8887
commit 16e53eec2b
1 changed files with 212 additions and 0 deletions
--- a/quickstep/tests/multivalentTests/src/com/android/quickstep/TopTaskTrackerTest.kt
+++ b/quickstep/tests/multivalentTests/src/com/android/quickstep/TopTaskTrackerTest.kt
@@ -0,0 +1,212 @@
+/*
+ * Copyright (C) 2025 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.android.quickstep
+
+import android.app.ActivityManager
+import android.app.TaskInfo
+import android.app.WindowConfiguration.ACTIVITY_TYPE_STANDARD
+import android.app.WindowConfiguration.WINDOWING_MODE_FREEFORM
+import android.app.WindowConfiguration.WINDOWING_MODE_FULLSCREEN
+import android.content.ComponentName
+import android.content.Context
+import android.content.Intent
+import android.content.res.Resources
+import android.platform.test.annotations.DisableFlags
+import android.platform.test.annotations.EnableFlags
+import android.platform.test.flag.junit.SetFlagsRule
+import android.view.Display.DEFAULT_DISPLAY
+import androidx.test.ext.junit.runners.AndroidJUnit4
+import com.android.internal.R
+import com.android.launcher3.statehandlers.DesktopVisibilityController.Companion.INACTIVE_DESK_ID
+import com.android.window.flags.Flags.FLAG_ENABLE_MULTIPLE_DESKTOPS_BACKEND
+import com.android.window.flags.Flags.FLAG_ENABLE_MULTIPLE_DESKTOPS_FRONTEND
+import com.android.wm.shell.Flags.FLAG_ENABLE_SHELL_TOP_TASK_TRACKING
+import com.android.wm.shell.shared.GroupedTaskInfo
+import com.android.wm.shell.shared.GroupedTaskInfo.TYPE_DESK
+import com.android.wm.shell.shared.GroupedTaskInfo.TYPE_FULLSCREEN
+import com.android.wm.shell.shared.GroupedTaskInfo.TYPE_SPLIT
+import com.google.common.truth.Truth.assertThat
+import org.junit.Before
+import org.junit.Rule
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.mockito.kotlin.doReturn
+import org.mockito.kotlin.eq
+import org.mockito.kotlin.mock
+import org.mockito.kotlin.whenever
+
+/** Test for [TopTaskTracker] */
+@RunWith(AndroidJUnit4::class)
+class TopTaskTrackerTest {
+
+    @get:Rule val setFlagsRule = SetFlagsRule(SetFlagsRule.DefaultInitValueType.DEVICE_DEFAULT)
+
+    private val mockContext = mock<Context>()
+    private val mockResources = mock<Resources>()
+
+    @Before
+    fun setUp() {
+        doReturn(mockResources).whenever(mockContext).resources
+    }
+
+    @Test
+    @EnableFlags(FLAG_ENABLE_SHELL_TOP_TASK_TRACKING)
+    fun getPlaceholderGroupedTaskInfo_shellTopTaskTrackingEnabled_noVisibleTasks() {
+        val cachedTaskInfo = TopTaskTracker.CachedTaskInfo(null)
+        val result = cachedTaskInfo.getPlaceholderGroupedTaskInfo(null)
+        assertThat(result).isNull()
+    }
+
+    @Test
+    @EnableFlags(FLAG_ENABLE_SHELL_TOP_TASK_TRACKING)
+    fun getPlaceholderGroupedTaskInfo_shellTopTaskTrackingEnabled_withVisibleTasks() {
+        val taskInfo = createTaskInfo(1, DEFAULT_DISPLAY)
+        val groupedTaskInfo = GroupedTaskInfo.forFullscreenTasks(taskInfo)
+        val cachedTaskInfo = TopTaskTracker.CachedTaskInfo(groupedTaskInfo)
+
+        val result = cachedTaskInfo.getPlaceholderGroupedTaskInfo(null)
+        assertThat(result).isEqualTo(groupedTaskInfo)
+    }
+
+    @Test
+    @DisableFlags(FLAG_ENABLE_SHELL_TOP_TASK_TRACKING)
+    fun getPlaceholderGroupedTaskInfo_shellTopTaskTrackingDisabled_noTasks() {
+        val cachedTaskInfo =
+            TopTaskTracker.CachedTaskInfo(
+                emptyList(),
+                mockContext,
+                DEFAULT_DISPLAY,
+                INACTIVE_DESK_ID,
+            )
+
+        val result = cachedTaskInfo.getPlaceholderGroupedTaskInfo(null)
+
+        assertThat(result).isNull()
+    }
+
+    @Test
+    @DisableFlags(FLAG_ENABLE_SHELL_TOP_TASK_TRACKING)
+    fun getPlaceholderGroupedTaskInfo_shellTopTaskTrackingDisabled_withFullscreenTask() {
+        val taskInfo = createTaskInfo(1, DEFAULT_DISPLAY)
+        val tasks = listOf(taskInfo)
+        val cachedTaskInfo =
+            TopTaskTracker.CachedTaskInfo(tasks, mockContext, DEFAULT_DISPLAY, INACTIVE_DESK_ID)
+
+        val result = cachedTaskInfo.getPlaceholderGroupedTaskInfo(null)
+
+        assertThat(result).isNotNull()
+        assertThat(result!!.isBaseType(TYPE_FULLSCREEN)).isTrue()
+        assertThat(result.taskInfo1).isEqualTo(taskInfo)
+    }
+
+    @Test
+    @DisableFlags(FLAG_ENABLE_SHELL_TOP_TASK_TRACKING)
+    fun getPlaceholderGroupedTaskInfo_shellTopTaskTrackingDisabled_withSplitTasks() {
+        val taskInfo1 = createTaskInfo(1, DEFAULT_DISPLAY)
+        val taskInfo2 = createTaskInfo(2, DEFAULT_DISPLAY)
+        val tasks = listOf(taskInfo1, taskInfo2)
+        val splitTaskIds = intArrayOf(1, 2)
+        val cachedTaskInfo =
+            TopTaskTracker.CachedTaskInfo(tasks, mockContext, DEFAULT_DISPLAY, INACTIVE_DESK_ID)
+
+        val result = cachedTaskInfo.getPlaceholderGroupedTaskInfo(splitTaskIds)
+
+        assertThat(result).isNotNull()
+        assertThat(result!!.isBaseType(TYPE_SPLIT)).isTrue()
+        assertThat(result.taskInfo1).isEqualTo(taskInfo1)
+        assertThat(result.taskInfo2).isEqualTo(taskInfo2)
+    }
+
+    @Test
+    @DisableFlags(FLAG_ENABLE_SHELL_TOP_TASK_TRACKING)
+    @EnableFlags(FLAG_ENABLE_MULTIPLE_DESKTOPS_FRONTEND, FLAG_ENABLE_MULTIPLE_DESKTOPS_BACKEND)
+    fun getPlaceholderGroupedTaskInfo_shellTopTaskTrackingDisabled_withDesktopEnabled_noActiveDesk() {
+        doReturn(true).whenever(mockResources).getBoolean(eq(R.bool.config_isDesktopModeSupported))
+        doReturn(true)
+            .whenever(mockResources)
+            .getBoolean(eq(R.bool.config_canInternalDisplayHostDesktops))
+        val taskInfo = createDesktopTaskInfo(1, DEFAULT_DISPLAY)
+        val tasks = listOf(taskInfo)
+        val cachedTaskInfo =
+            TopTaskTracker.CachedTaskInfo(tasks, mockContext, DEFAULT_DISPLAY, INACTIVE_DESK_ID)
+
+        val result = cachedTaskInfo.getPlaceholderGroupedTaskInfo(null)
+
+        assertThat(result).isNotNull()
+        assertThat(result!!.isBaseType(TYPE_FULLSCREEN)).isTrue()
+        assertThat(result.deskId).isEqualTo(INACTIVE_DESK_ID)
+        assertThat(result.taskInfo1).isEqualTo(taskInfo)
+    }
+
+    @Test
+    @DisableFlags(FLAG_ENABLE_SHELL_TOP_TASK_TRACKING)
+    @EnableFlags(FLAG_ENABLE_MULTIPLE_DESKTOPS_FRONTEND, FLAG_ENABLE_MULTIPLE_DESKTOPS_BACKEND)
+    fun getPlaceholderGroupedTaskInfo_shellTopTaskTrackingDisabled_withDesktopEnabled_withActiveDesk() {
+        doReturn(true).whenever(mockResources).getBoolean(eq(R.bool.config_isDesktopModeSupported))
+        doReturn(true)
+            .whenever(mockResources)
+            .getBoolean(eq(R.bool.config_canInternalDisplayHostDesktops))
+        val activeDeskId = 10
+        val taskInfo = createDesktopTaskInfo(1, DEFAULT_DISPLAY)
+        val tasks = listOf(taskInfo)
+        val cachedTaskInfo =
+            TopTaskTracker.CachedTaskInfo(tasks, mockContext, DEFAULT_DISPLAY, activeDeskId)
+
+        val result = cachedTaskInfo.getPlaceholderGroupedTaskInfo(null)
+
+        assertThat(result).isNotNull()
+        assertThat(result!!.isBaseType(TYPE_DESK)).isTrue()
+        assertThat(result.deskId).isEqualTo(activeDeskId)
+        assertThat(result.getTaskInfoList()).isEqualTo(tasks)
+    }
+
+    @Test
+    @DisableFlags(FLAG_ENABLE_SHELL_TOP_TASK_TRACKING, FLAG_ENABLE_MULTIPLE_DESKTOPS_FRONTEND)
+    fun getPlaceholderGroupedTaskInfo_shellTopTaskTrackingDisabled_withDesktopDisabled_withActiveDesk() {
+        doReturn(true).whenever(mockResources).getBoolean(eq(R.bool.config_isDesktopModeSupported))
+        doReturn(true)
+            .whenever(mockResources)
+            .getBoolean(eq(R.bool.config_canInternalDisplayHostDesktops))
+        val taskInfo = createDesktopTaskInfo(1, DEFAULT_DISPLAY)
+        val tasks = listOf(taskInfo)
+        val cachedTaskInfo =
+            TopTaskTracker.CachedTaskInfo(tasks, mockContext, DEFAULT_DISPLAY, INACTIVE_DESK_ID)
+        val result = cachedTaskInfo.getPlaceholderGroupedTaskInfo(null)
+
+        assertThat(result).isNotNull()
+        assertThat(result!!.isBaseType(TYPE_DESK)).isTrue()
+        assertThat(result.deskId).isEqualTo(INACTIVE_DESK_ID)
+        assertThat(result.taskInfo1).isEqualTo(taskInfo)
+    }
+
+    private fun createTaskInfo(taskId: Int, displayId: Int): TaskInfo {
+        val taskInfo = ActivityManager.RunningTaskInfo()
+        taskInfo.taskId = taskId
+        taskInfo.displayId = displayId
+        taskInfo.baseIntent = Intent()
+        taskInfo.baseActivity = ComponentName("test", "test")
+        taskInfo.configuration.windowConfiguration.setActivityType(ACTIVITY_TYPE_STANDARD)
+        taskInfo.configuration.windowConfiguration.setWindowingMode(WINDOWING_MODE_FULLSCREEN)
+        return taskInfo
+    }
+
+    private fun createDesktopTaskInfo(taskId: Int, displayId: Int): TaskInfo {
+        val taskInfo = createTaskInfo(taskId, displayId)
+        taskInfo.configuration.windowConfiguration.setWindowingMode(WINDOWING_MODE_FREEFORM)
+        return taskInfo
+    }
+}