Merge "[AI TestGen] Test: Add tests for containsClosingTaskInDesktop" into main (935fa589) · Commits · e / os / android_frameworks_base

libs/WindowManager/Shell/tests/unittest/src/com/android/wm/shell/desktopmode/DesktopTasksTransitionObserverTest.kt

+56 −1

Original line number	Diff line number	Diff line
		@@ -18,6 +18,7 @@ package com.android.wm.shell.desktopmode

		import android.app.ActivityManager.RunningTaskInfo
		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
		@@ -263,6 +264,60 @@ class DesktopTasksTransitionObserverTest : ShellTestCase() {
		)
		}

		@Test
		@EnableFlags(Flags.FLAG_ENABLE_DESKTOP_CLOSE_TASK_ANIMATION_IN_DTC_BUGFIX)
		fun onTransitionReady_closingTaskAndExitDesktop_notAddPendingMixedTransition() {
		val mockTransition = Mockito.mock(IBinder::class.java)
		val task = createTaskInfo(1, WINDOWING_MODE_FREEFORM)
		whenever(taskRepository.isAnyDeskActive(any())).thenReturn(false)
		whenever(mixedHandler.hasTransition(mockTransition)).thenReturn(false)

		transitionObserver.onTransitionReady(
		transition = mockTransition,
		info = createCloseTransition(listOf(task)),
		startTransaction = mock(),
		finishTransaction = mock(),
		)

		verify(mixedHandler, never()).addPendingMixedTransition(any())
		}

		@Test
		@EnableFlags(Flags.FLAG_ENABLE_DESKTOP_CLOSE_TASK_ANIMATION_IN_DTC_BUGFIX)
		fun onTransitionReady_closingTaskNotFreeform_notAddPendingMixedTransition() {
		val mockTransition = Mockito.mock(IBinder::class.java)
		val task = createTaskInfo(1, WINDOWING_MODE_FULLSCREEN)
		whenever(taskRepository.isAnyDeskActive(any())).thenReturn(true)
		whenever(mixedHandler.hasTransition(mockTransition)).thenReturn(false)

		transitionObserver.onTransitionReady(
		transition = mockTransition,
		info = createCloseTransition(listOf(task)),
		startTransaction = mock(),
		finishTransaction = mock(),
		)

		verify(mixedHandler, never()).addPendingMixedTransition(any())
		}

		@Test
		@EnableFlags(Flags.FLAG_ENABLE_DESKTOP_CLOSE_TASK_ANIMATION_IN_DTC_BUGFIX)
		fun onTransitionReady_notClosingTask_notAddPendingMixedTransition() {
		val mockTransition = Mockito.mock(IBinder::class.java)
		val task = createTaskInfo(1, WINDOWING_MODE_FREEFORM)
		whenever(taskRepository.isAnyDeskActive(any())).thenReturn(true)
		whenever(mixedHandler.hasTransition(mockTransition)).thenReturn(false)

		transitionObserver.onTransitionReady(
		transition = mockTransition,
		info = createOpenChangeTransition(task),
		startTransaction = mock(),
		finishTransaction = mock(),
		)

		verify(mixedHandler, never()).addPendingMixedTransition(any())
		}

		@Test
		@EnableFlags(Flags.FLAG_ENABLE_DESKTOP_CLOSE_TASK_ANIMATION_IN_DTC_BUGFIX)
		fun closingTask_startsTransitionToRemoveFully() {