Jeremy Sim
a8690f5eb6
This CL adds unit tests for the SplitSelectDataHolder class. Fixes: 288318661 Test: None Change-Id: Ia09b928be1c73dcf8a198ea5ece16f0557fb5b0d
396 lines
14 KiB
Kotlin
396 lines
14 KiB
Kotlin
/*
|
|
* Copyright (C) 2023 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.util
|
|
|
|
import android.app.ActivityManager.RunningTaskInfo
|
|
import android.app.ActivityTaskManager.INVALID_TASK_ID
|
|
import android.content.Context
|
|
import android.content.ContextWrapper
|
|
import android.content.Intent
|
|
import android.os.UserHandle
|
|
import androidx.test.platform.app.InstrumentationRegistry
|
|
import com.android.launcher3.model.data.ItemInfo
|
|
import com.android.launcher3.shortcuts.ShortcutKey
|
|
import com.android.launcher3.ui.AbstractLauncherUiTest
|
|
import com.android.launcher3.util.SplitConfigurationOptions.STAGE_POSITION_TOP_OR_LEFT
|
|
import com.android.quickstep.util.SplitSelectDataHolder.Companion.SPLIT_PENDINGINTENT_PENDINGINTENT
|
|
import com.android.quickstep.util.SplitSelectDataHolder.Companion.SPLIT_PENDINGINTENT_TASK
|
|
import com.android.quickstep.util.SplitSelectDataHolder.Companion.SPLIT_SHORTCUT_TASK
|
|
import com.android.quickstep.util.SplitSelectDataHolder.Companion.SPLIT_SINGLE_INTENT_FULLSCREEN
|
|
import com.android.quickstep.util.SplitSelectDataHolder.Companion.SPLIT_SINGLE_SHORTCUT_FULLSCREEN
|
|
import com.android.quickstep.util.SplitSelectDataHolder.Companion.SPLIT_SINGLE_TASK_FULLSCREEN
|
|
import com.android.quickstep.util.SplitSelectDataHolder.Companion.SPLIT_TASK_PENDINGINTENT
|
|
import com.android.quickstep.util.SplitSelectDataHolder.Companion.SPLIT_TASK_SHORTCUT
|
|
import com.android.quickstep.util.SplitSelectDataHolder.Companion.SPLIT_TASK_TASK
|
|
import org.junit.Assert.assertEquals
|
|
import org.junit.Assert.assertFalse
|
|
import org.junit.Assert.assertNotEquals
|
|
import org.junit.Assert.assertNotNull
|
|
import org.junit.Assert.assertNull
|
|
import org.junit.Assert.assertTrue
|
|
import org.junit.Before
|
|
import org.junit.Test
|
|
|
|
class SplitSelectDataHolderTest {
|
|
private lateinit var splitSelectDataHolder: SplitSelectDataHolder
|
|
|
|
private val context: Context =
|
|
ContextWrapper(InstrumentationRegistry.getInstrumentation().targetContext)
|
|
private val sampleTaskInfo = RunningTaskInfo()
|
|
private val sampleTaskId = 10
|
|
private val sampleTaskId2 = 11
|
|
private val sampleUser = UserHandle(0)
|
|
private val sampleIntent = Intent()
|
|
private val sampleIntent2 = Intent()
|
|
private val sampleShortcut = Intent()
|
|
private val sampleShortcut2 = Intent()
|
|
private val sampleItemInfo = ItemInfo()
|
|
private val samplePackage =
|
|
AbstractLauncherUiTest.resolveSystemApp(Intent.CATEGORY_APP_CALCULATOR)
|
|
|
|
@Before
|
|
fun setup() {
|
|
splitSelectDataHolder = SplitSelectDataHolder(context)
|
|
|
|
sampleTaskInfo.taskId = sampleTaskId
|
|
sampleItemInfo.user = sampleUser
|
|
sampleIntent.setPackage(samplePackage)
|
|
sampleIntent2.setPackage(samplePackage)
|
|
sampleShortcut.setPackage(samplePackage)
|
|
sampleShortcut2.setPackage(samplePackage)
|
|
sampleShortcut.putExtra(ShortcutKey.EXTRA_SHORTCUT_ID, "sampleShortcut")
|
|
sampleShortcut2.putExtra(ShortcutKey.EXTRA_SHORTCUT_ID, "sampleShortcut2")
|
|
}
|
|
|
|
@Test
|
|
fun setInitialAsTask() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleTaskInfo,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
null,
|
|
null
|
|
)
|
|
assertTrue(splitSelectDataHolder.isSplitSelectActive())
|
|
}
|
|
|
|
@Test
|
|
fun setInitialAsIntent() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleIntent,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null,
|
|
INVALID_TASK_ID
|
|
)
|
|
assertTrue(splitSelectDataHolder.isSplitSelectActive())
|
|
}
|
|
|
|
@Test
|
|
fun setInitialAsIntentWithAlreadyRunningTask() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleIntent,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null,
|
|
sampleTaskId
|
|
)
|
|
assertTrue(splitSelectDataHolder.isSplitSelectActive())
|
|
}
|
|
|
|
@Test
|
|
fun setInitialAsShortcut() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleShortcut,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null,
|
|
INVALID_TASK_ID
|
|
)
|
|
assertTrue(splitSelectDataHolder.isSplitSelectActive())
|
|
}
|
|
|
|
@Test
|
|
fun setSecondAsTask() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleShortcut,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null,
|
|
INVALID_TASK_ID
|
|
)
|
|
splitSelectDataHolder.setSecondTask(sampleTaskId)
|
|
assertTrue(splitSelectDataHolder.isBothSplitAppsConfirmed())
|
|
}
|
|
|
|
@Test
|
|
fun setSecondAsIntent() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleTaskInfo,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
null,
|
|
null
|
|
)
|
|
splitSelectDataHolder.setSecondTask(sampleIntent, sampleUser)
|
|
assertTrue(splitSelectDataHolder.isBothSplitAppsConfirmed())
|
|
}
|
|
|
|
@Test
|
|
fun setSecondAsShortcut() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleIntent,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null,
|
|
INVALID_TASK_ID
|
|
)
|
|
splitSelectDataHolder.setSecondTask(sampleShortcut, sampleUser)
|
|
assertTrue(splitSelectDataHolder.isBothSplitAppsConfirmed())
|
|
}
|
|
|
|
@Test
|
|
fun generateLaunchData_Task_Task() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleTaskInfo,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null
|
|
)
|
|
splitSelectDataHolder.setSecondTask(sampleTaskId2)
|
|
val launchData = splitSelectDataHolder.getSplitLaunchData()
|
|
|
|
assertEquals(launchData.splitLaunchType, SPLIT_TASK_TASK)
|
|
|
|
// should contain a valid task ID for first app, and no intent or shortcut
|
|
assertNotEquals(launchData.initialTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.initialPendingIntent)
|
|
assertNull(launchData.initialShortcut)
|
|
|
|
// should contain a valid task ID for second app, and no intent or shortcut
|
|
assertNotEquals(launchData.secondTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.secondPendingIntent)
|
|
assertNull(launchData.secondShortcut)
|
|
}
|
|
|
|
@Test
|
|
fun generateLaunchData_Task_Intent() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleTaskInfo,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null
|
|
)
|
|
splitSelectDataHolder.setSecondTask(sampleIntent, sampleUser)
|
|
val launchData = splitSelectDataHolder.getSplitLaunchData()
|
|
|
|
assertEquals(launchData.splitLaunchType, SPLIT_TASK_PENDINGINTENT)
|
|
|
|
// should contain a valid task ID for first app, and no intent or shortcut
|
|
assertNotEquals(launchData.initialTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.initialPendingIntent)
|
|
assertNull(launchData.initialShortcut)
|
|
|
|
// should contain a valid intent for second app, and no task ID or shortcut
|
|
assertNotNull(launchData.secondPendingIntent)
|
|
assertEquals(launchData.secondTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.secondShortcut)
|
|
}
|
|
|
|
@Test
|
|
fun generateLaunchData_Task_Shortcut() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleTaskInfo,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null
|
|
)
|
|
splitSelectDataHolder.setSecondTask(sampleShortcut, sampleUser)
|
|
val launchData = splitSelectDataHolder.getSplitLaunchData()
|
|
|
|
assertEquals(launchData.splitLaunchType, SPLIT_TASK_SHORTCUT)
|
|
|
|
// should contain a valid task ID for first app, and no intent or shortcut
|
|
assertNotEquals(launchData.initialTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.initialPendingIntent)
|
|
assertNull(launchData.initialShortcut)
|
|
|
|
// should contain a valid shortcut and intent for second app, and no task ID
|
|
assertNotNull(launchData.secondShortcut)
|
|
assertNotNull(launchData.secondPendingIntent)
|
|
assertEquals(launchData.secondTaskId, INVALID_TASK_ID)
|
|
}
|
|
|
|
@Test
|
|
fun generateLaunchData_Intent_Task() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleIntent,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null,
|
|
INVALID_TASK_ID
|
|
)
|
|
splitSelectDataHolder.setSecondTask(sampleTaskId)
|
|
val launchData = splitSelectDataHolder.getSplitLaunchData()
|
|
|
|
assertEquals(launchData.splitLaunchType, SPLIT_PENDINGINTENT_TASK)
|
|
|
|
// should contain a valid intent for first app, and no task ID or shortcut
|
|
assertNotNull(launchData.initialPendingIntent)
|
|
assertEquals(launchData.initialTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.initialShortcut)
|
|
|
|
// should contain a valid task ID for second app, and no intent or shortcut
|
|
assertNotEquals(launchData.secondTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.secondPendingIntent)
|
|
assertNull(launchData.secondShortcut)
|
|
}
|
|
|
|
@Test
|
|
fun generateLaunchData_Shortcut_Task() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleShortcut,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null,
|
|
INVALID_TASK_ID
|
|
)
|
|
splitSelectDataHolder.setSecondTask(sampleTaskId)
|
|
val launchData = splitSelectDataHolder.getSplitLaunchData()
|
|
|
|
assertEquals(launchData.splitLaunchType, SPLIT_SHORTCUT_TASK)
|
|
|
|
// should contain a valid shortcut and intent for first app, and no task ID
|
|
assertNotNull(launchData.initialShortcut)
|
|
assertNotNull(launchData.initialPendingIntent)
|
|
assertEquals(launchData.initialTaskId, INVALID_TASK_ID)
|
|
|
|
// should contain a valid task ID for second app, and no intent or shortcut
|
|
assertNotEquals(launchData.secondTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.secondPendingIntent)
|
|
assertNull(launchData.secondShortcut)
|
|
}
|
|
|
|
@Test
|
|
fun generateLaunchData_Intent_Intent() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleIntent,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null,
|
|
INVALID_TASK_ID
|
|
)
|
|
splitSelectDataHolder.setSecondTask(sampleIntent2, sampleUser)
|
|
val launchData = splitSelectDataHolder.getSplitLaunchData()
|
|
|
|
assertEquals(launchData.splitLaunchType, SPLIT_PENDINGINTENT_PENDINGINTENT)
|
|
|
|
// should contain a valid intent for first app, and no task ID or shortcut
|
|
assertNotNull(launchData.initialPendingIntent)
|
|
assertEquals(launchData.initialTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.initialShortcut)
|
|
|
|
// should contain a valid intent for second app, and no task ID or shortcut
|
|
assertNotNull(launchData.secondPendingIntent)
|
|
assertEquals(launchData.secondTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.secondShortcut)
|
|
}
|
|
|
|
@Test
|
|
fun generateLaunchData_Single_Task() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleTaskInfo,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null
|
|
)
|
|
val launchData = splitSelectDataHolder.getFullscreenLaunchData()
|
|
|
|
assertEquals(launchData.splitLaunchType, SPLIT_SINGLE_TASK_FULLSCREEN)
|
|
|
|
// should contain a valid task ID for first app, and no intent or shortcut
|
|
assertNotEquals(launchData.initialTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.initialPendingIntent)
|
|
assertNull(launchData.initialShortcut)
|
|
|
|
// should contain no task ID, intent, or shortcut for second app
|
|
assertEquals(launchData.secondTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.secondPendingIntent)
|
|
assertNull(launchData.secondShortcut)
|
|
}
|
|
|
|
@Test
|
|
fun generateLaunchData_Single_Intent() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleIntent,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null,
|
|
INVALID_TASK_ID
|
|
)
|
|
val launchData = splitSelectDataHolder.getFullscreenLaunchData()
|
|
|
|
assertEquals(launchData.splitLaunchType, SPLIT_SINGLE_INTENT_FULLSCREEN)
|
|
|
|
// should contain a valid intent for first app, and no task ID or shortcut
|
|
assertNotNull(launchData.initialPendingIntent)
|
|
assertEquals(launchData.initialTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.initialShortcut)
|
|
|
|
// should contain no task ID, intent, or shortcut for second app
|
|
assertEquals(launchData.secondTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.secondPendingIntent)
|
|
assertNull(launchData.secondShortcut)
|
|
}
|
|
|
|
@Test
|
|
fun generateLaunchData_Single_Shortcut() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleShortcut,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
sampleItemInfo,
|
|
null,
|
|
INVALID_TASK_ID
|
|
)
|
|
val launchData = splitSelectDataHolder.getFullscreenLaunchData()
|
|
|
|
assertEquals(launchData.splitLaunchType, SPLIT_SINGLE_SHORTCUT_FULLSCREEN)
|
|
|
|
// should contain a valid shortcut and intent for first app, and no task ID
|
|
assertNotNull(launchData.initialShortcut)
|
|
assertNotNull(launchData.initialPendingIntent)
|
|
assertEquals(launchData.initialTaskId, INVALID_TASK_ID)
|
|
|
|
// should contain no task ID, intent, or shortcut for second app
|
|
assertEquals(launchData.secondTaskId, INVALID_TASK_ID)
|
|
assertNull(launchData.secondPendingIntent)
|
|
assertNull(launchData.secondShortcut)
|
|
}
|
|
|
|
@Test
|
|
fun clearState() {
|
|
splitSelectDataHolder.setInitialTaskSelect(
|
|
sampleTaskInfo,
|
|
STAGE_POSITION_TOP_OR_LEFT,
|
|
null,
|
|
null
|
|
)
|
|
splitSelectDataHolder.setSecondTask(sampleIntent, sampleUser)
|
|
splitSelectDataHolder.resetState()
|
|
assertFalse(splitSelectDataHolder.isSplitSelectActive())
|
|
}
|
|
}
|