Merge "[Dual Shade] Fix DeviceEntrySourceInteractorTest for DualShade on." into main (f510a115) · Commits · e / os / android_frameworks_base

packages/SystemUI/multivalentTests/src/com/android/systemui/deviceentry/domain/interactor/DeviceEntrySourceInteractorTest.kt

+74 −91

Original line number	Diff line number	Diff line
		@@ -27,8 +27,6 @@ import com.android.systemui.authentication.data.repository.fakeAuthenticationRep
		import com.android.systemui.authentication.shared.model.AuthenticationMethodModel
		import com.android.systemui.biometrics.authController
		import com.android.systemui.bouncer.data.repository.keyguardBouncerRepository
		import com.android.systemui.coroutines.collectLastValue
		import com.android.systemui.deviceentry.shared.model.FaceAuthenticationStatus
		import com.android.systemui.deviceentry.shared.model.SuccessFaceAuthenticationStatus
		import com.android.systemui.flags.DisableSceneContainer
		import com.android.systemui.flags.EnableSceneContainer
		@@ -42,20 +40,22 @@ import com.android.systemui.keyguard.data.repository.verifyCallback
		import com.android.systemui.keyguard.shared.model.BiometricUnlockMode
		import com.android.systemui.keyguard.shared.model.BiometricUnlockSource
		import com.android.systemui.keyguard.shared.model.SuccessFingerprintAuthenticationStatus
		import com.android.systemui.kosmos.testScope
		import com.android.systemui.kosmos.Kosmos
		import com.android.systemui.kosmos.collectLastValue
		import com.android.systemui.kosmos.runCurrent
		import com.android.systemui.kosmos.runTest
		import com.android.systemui.scene.domain.interactor.sceneInteractor
		import com.android.systemui.scene.shared.flag.SceneContainerFlag
		import com.android.systemui.scene.shared.model.Overlays
		import com.android.systemui.scene.shared.model.Scenes
		import com.android.systemui.shade.domain.interactor.enableSingleShade
		import com.android.systemui.statusbar.phone.dozeScrimController
		import com.android.systemui.statusbar.phone.screenOffAnimationController
		import com.android.systemui.statusbar.policy.DevicePostureController.DEVICE_POSTURE_OPENED
		import com.android.systemui.statusbar.policy.devicePostureController
		import com.android.systemui.testKosmos
		import com.google.common.truth.Truth.assertThat
		import kotlinx.coroutines.flow.MutableStateFlow
		import kotlinx.coroutines.test.runCurrent
		import kotlinx.coroutines.test.runTest
		import kotlinx.coroutines.flow.flowOf
		import org.junit.Before
		import org.junit.Test
		import org.junit.runner.RunWith
		@@ -65,34 +65,34 @@ import org.mockito.kotlin.whenever
		@SmallTest
		@RunWith(AndroidJUnit4::class)
		class DeviceEntrySourceInteractorTest : SysuiTestCase() {

		private val kosmos = testKosmos()
		private val testScope = kosmos.testScope
		private lateinit var underTest: DeviceEntrySourceInteractor
		private val underTest: DeviceEntrySourceInteractor by lazy {
		kosmos.deviceEntrySourceInteractor
		}

		@Before
		fun setup() {
		with(kosmos) {
		if (SceneContainerFlag.isEnabled) {
		whenever(kosmos.authController.isUdfpsFingerDown).thenReturn(false)
		whenever(kosmos.dozeScrimController.isPulsing).thenReturn(false)
		whenever(kosmos.keyguardUpdateMonitor.isUnlockingWithBiometricAllowed(anyBoolean()))
		whenever(authController.isUdfpsFingerDown).thenReturn(false)
		whenever(dozeScrimController.isPulsing).thenReturn(false)
		whenever(keyguardUpdateMonitor.isUnlockingWithBiometricAllowed(anyBoolean()))
		.thenReturn(true)
		whenever(kosmos.screenOffAnimationController.isKeyguardShowDelayed()).thenReturn(false)
		kosmos.fakeAuthenticationRepository.setAuthenticationMethod(
		AuthenticationMethodModel.Pin
		)
		} else {
		underTest = kosmos.deviceEntrySourceInteractor
		whenever(screenOffAnimationController.isKeyguardShowDelayed()).thenReturn(false)
		fakeAuthenticationRepository.setAuthenticationMethod(AuthenticationMethodModel.Pin)
		}
		}
		}

		@DisableSceneContainer
		@Test
		fun deviceEntryFromFaceUnlock() =
		testScope.runTest {
		kosmos.runTest {
		val deviceEntryFromBiometricAuthentication by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		kosmos.fakeKeyguardRepository.setBiometricUnlockState(
		fakeKeyguardRepository.setBiometricUnlockState(
		BiometricUnlockMode.WAKE_AND_UNLOCK,
		BiometricUnlockSource.FACE_SENSOR,
		)
		@@ -105,11 +105,11 @@ class DeviceEntrySourceInteractorTest : SysuiTestCase() {
		@DisableSceneContainer
		@Test
		fun deviceEntryFromFingerprintUnlock() =
		testScope.runTest {
		kosmos.runTest {
		val deviceEntryFromBiometricAuthentication by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		kosmos.fakeKeyguardRepository.setBiometricUnlockState(
		fakeKeyguardRepository.setBiometricUnlockState(
		BiometricUnlockMode.WAKE_AND_UNLOCK,
		BiometricUnlockSource.FINGERPRINT_SENSOR,
		)
		@@ -122,11 +122,11 @@ class DeviceEntrySourceInteractorTest : SysuiTestCase() {
		@DisableSceneContainer
		@Test
		fun noDeviceEntry() =
		testScope.runTest {
		kosmos.runTest {
		val deviceEntryFromBiometricAuthentication by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		kosmos.fakeKeyguardRepository.setBiometricUnlockState(
		fakeKeyguardRepository.setBiometricUnlockState(
		BiometricUnlockMode.ONLY_WAKE, // doesn't dismiss keyguard:
		BiometricUnlockSource.FINGERPRINT_SENSOR,
		)
		@@ -138,22 +138,19 @@ class DeviceEntrySourceInteractorTest : SysuiTestCase() {
		@EnableSceneContainer
		@Test
		fun deviceEntryFromFingerprintUnlockOnLockScreen_sceneContainerEnabled() =
		testScope.runTest {
		underTest = kosmos.deviceEntrySourceInteractor
		kosmos.runTest {
		val deviceEntryFromBiometricSource by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		whenever(kosmos.keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		whenever(keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		configureDeviceEntryBiometricAuthSuccessState(isFingerprintAuth = true)
		configureBiometricUnlockState(
		primaryBouncerVisible = false,
		alternateBouncerVisible = false,
		sceneKey = Scenes.Lockscreen,
		)
		runCurrent()

		kosmos.configureKeyguardBypass(isBypassAvailable = true)
		underTest = kosmos.deviceEntrySourceInteractor
		configureKeyguardBypass(isBypassAvailable = true)

		assertThat(deviceEntryFromBiometricSource)
		.isEqualTo(BiometricUnlockSource.FINGERPRINT_SENSOR)
		@@ -162,19 +159,17 @@ class DeviceEntrySourceInteractorTest : SysuiTestCase() {
		@EnableSceneContainer
		@Test
		fun deviceEntryFromFingerprintUnlockOnAod_sceneContainerEnabled() =
		testScope.runTest {
		underTest = kosmos.deviceEntrySourceInteractor
		kosmos.runTest {
		val deviceEntryFromBiometricSource by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		whenever(kosmos.keyguardUpdateMonitor.isDeviceInteractive).thenReturn(false)
		whenever(keyguardUpdateMonitor.isDeviceInteractive).thenReturn(false)
		configureDeviceEntryBiometricAuthSuccessState(isFingerprintAuth = true)
		configureBiometricUnlockState(
		primaryBouncerVisible = false,
		alternateBouncerVisible = false,
		sceneKey = Scenes.Dream,
		)
		runCurrent()

		assertThat(deviceEntryFromBiometricSource)
		.isEqualTo(BiometricUnlockSource.FINGERPRINT_SENSOR)
		@@ -183,19 +178,17 @@ class DeviceEntrySourceInteractorTest : SysuiTestCase() {
		@EnableSceneContainer
		@Test
		fun deviceEntryFromFingerprintUnlockOnBouncer_sceneContainerEnabled() =
		testScope.runTest {
		underTest = kosmos.deviceEntrySourceInteractor
		kosmos.runTest {
		val deviceEntryFromBiometricSource by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		whenever(kosmos.keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		whenever(keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		configureDeviceEntryBiometricAuthSuccessState(isFingerprintAuth = true)
		configureBiometricUnlockState(
		primaryBouncerVisible = true,
		alternateBouncerVisible = false,
		sceneKey = Scenes.Lockscreen,
		)
		runCurrent()

		assertThat(deviceEntryFromBiometricSource)
		.isEqualTo(BiometricUnlockSource.FINGERPRINT_SENSOR)
		@@ -204,19 +197,17 @@ class DeviceEntrySourceInteractorTest : SysuiTestCase() {
		@EnableSceneContainer
		@Test
		fun deviceEntryFromFingerprintUnlockOnShade_sceneContainerEnabled() =
		testScope.runTest {
		underTest = kosmos.deviceEntrySourceInteractor
		kosmos.runTest {
		val deviceEntryFromBiometricSource by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		whenever(kosmos.keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		whenever(keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		configureDeviceEntryBiometricAuthSuccessState(isFingerprintAuth = true)
		configureBiometricUnlockState(
		primaryBouncerVisible = false,
		alternateBouncerVisible = false,
		sceneKey = Scenes.Lockscreen,
		)
		runCurrent()

		assertThat(deviceEntryFromBiometricSource)
		.isEqualTo(BiometricUnlockSource.FINGERPRINT_SENSOR)
		@@ -225,19 +216,17 @@ class DeviceEntrySourceInteractorTest : SysuiTestCase() {
		@EnableSceneContainer
		@Test
		fun deviceEntryFromFingerprintUnlockOnAlternateBouncer_sceneContainerEnabled() =
		testScope.runTest {
		underTest = kosmos.deviceEntrySourceInteractor
		kosmos.runTest {
		val deviceEntryFromBiometricSource by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		whenever(kosmos.keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		whenever(keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		configureDeviceEntryBiometricAuthSuccessState(isFingerprintAuth = true)
		configureBiometricUnlockState(
		primaryBouncerVisible = false,
		alternateBouncerVisible = true,
		sceneKey = Scenes.Lockscreen,
		)
		runCurrent()

		assertThat(deviceEntryFromBiometricSource)
		.isEqualTo(BiometricUnlockSource.FINGERPRINT_SENSOR)
		@@ -246,21 +235,19 @@ class DeviceEntrySourceInteractorTest : SysuiTestCase() {
		@EnableSceneContainer
		@Test
		fun deviceEntryFromFaceUnlockOnLockScreen_bypassAvailable_sceneContainerEnabled() =
		testScope.runTest {
		kosmos.configureKeyguardBypass(isBypassAvailable = true)
		underTest = kosmos.deviceEntrySourceInteractor
		kosmos.runTest {
		configureKeyguardBypass(isBypassAvailable = true)

		val deviceEntryFromBiometricSource by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		whenever(kosmos.keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		whenever(keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		configureDeviceEntryBiometricAuthSuccessState(isFaceAuth = true)
		configureBiometricUnlockState(
		primaryBouncerVisible = false,
		alternateBouncerVisible = false,
		sceneKey = Scenes.Lockscreen,
		)
		runCurrent()

		assertThat(deviceEntryFromBiometricSource).isEqualTo(BiometricUnlockSource.FACE_SENSOR)
		}
		@@ -268,27 +255,25 @@ class DeviceEntrySourceInteractorTest : SysuiTestCase() {
		@EnableSceneContainer
		@Test
		fun deviceEntryFromFaceUnlockOnLockScreen_bypassDisabled_sceneContainerEnabled() =
		testScope.runTest {
		kosmos.configureKeyguardBypass(isBypassAvailable = false)
		underTest = kosmos.deviceEntrySourceInteractor
		kosmos.runTest {
		configureKeyguardBypass(isBypassAvailable = false)

		collectLastValue(kosmos.keyguardBypassRepository.isBypassAvailable)
		collectLastValue(keyguardBypassRepository.isBypassAvailable)
		runCurrent()

		val postureControllerCallback = kosmos.devicePostureController.verifyCallback()
		val postureControllerCallback = devicePostureController.verifyCallback()
		postureControllerCallback.onPostureChanged(DEVICE_POSTURE_OPENED)

		val deviceEntryFromBiometricSource by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		whenever(kosmos.keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		whenever(keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		configureDeviceEntryBiometricAuthSuccessState(isFaceAuth = true)
		configureBiometricUnlockState(
		primaryBouncerVisible = false,
		alternateBouncerVisible = false,
		sceneKey = Scenes.Lockscreen,
		)
		runCurrent()

		// MODE_NONE does not dismiss keyguard
		assertThat(deviceEntryFromBiometricSource).isNull()
		@@ -297,20 +282,18 @@ class DeviceEntrySourceInteractorTest : SysuiTestCase() {
		@EnableSceneContainer
		@Test
		fun deviceEntryFromFaceUnlockOnBouncer_sceneContainerEnabled() =
		testScope.runTest {
		kosmos.configureKeyguardBypass(isBypassAvailable = true)
		underTest = kosmos.deviceEntrySourceInteractor
		kosmos.runTest {
		configureKeyguardBypass(isBypassAvailable = true)
		val deviceEntryFromBiometricSource by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		whenever(kosmos.keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		whenever(keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		configureDeviceEntryBiometricAuthSuccessState(isFaceAuth = true)
		configureBiometricUnlockState(
		primaryBouncerVisible = true,
		alternateBouncerVisible = false,
		sceneKey = Scenes.Lockscreen,
		)
		runCurrent()

		assertThat(deviceEntryFromBiometricSource).isEqualTo(BiometricUnlockSource.FACE_SENSOR)
		}
		@@ -318,20 +301,20 @@ class DeviceEntrySourceInteractorTest : SysuiTestCase() {
		@EnableSceneContainer
		@Test
		fun deviceEntryFromFaceUnlockOnShade_bypassAvailable_sceneContainerEnabled() =
		testScope.runTest {
		kosmos.configureKeyguardBypass(isBypassAvailable = true)
		underTest = kosmos.deviceEntrySourceInteractor
		kosmos.runTest {
		enableSingleShade()
		runCurrent()
		configureKeyguardBypass(isBypassAvailable = true)
		val deviceEntryFromBiometricSource by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		whenever(kosmos.keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		whenever(keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		configureDeviceEntryBiometricAuthSuccessState(isFaceAuth = true)
		configureBiometricUnlockState(
		primaryBouncerVisible = false,
		alternateBouncerVisible = false,
		sceneKey = Scenes.Shade,
		)
		runCurrent()

		// MODE_NONE does not dismiss keyguard
		assertThat(deviceEntryFromBiometricSource).isNull()
		@@ -340,20 +323,18 @@ class DeviceEntrySourceInteractorTest : SysuiTestCase() {
		@EnableSceneContainer
		@Test
		fun deviceEntryFromFaceUnlockOnShade_bypassDisabled_sceneContainerEnabled() =
		testScope.runTest {
		kosmos.configureKeyguardBypass(isBypassAvailable = false)
		underTest = kosmos.deviceEntrySourceInteractor
		kosmos.runTest {
		configureKeyguardBypass(isBypassAvailable = false)
		val deviceEntryFromBiometricSource by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		whenever(kosmos.keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		whenever(keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		configureDeviceEntryBiometricAuthSuccessState(isFaceAuth = true)
		configureBiometricUnlockState(
		primaryBouncerVisible = false,
		alternateBouncerVisible = false,
		sceneKey = Scenes.Lockscreen,
		)
		runCurrent()

		assertThat(deviceEntryFromBiometricSource).isNull()
		}
		@@ -361,58 +342,60 @@ class DeviceEntrySourceInteractorTest : SysuiTestCase() {
		@EnableSceneContainer
		@Test
		fun deviceEntryFromFaceUnlockOnAlternateBouncer_sceneContainerEnabled() =
		testScope.runTest {
		kosmos.configureKeyguardBypass(isBypassAvailable = true)
		underTest = kosmos.deviceEntrySourceInteractor
		kosmos.runTest {
		configureKeyguardBypass(isBypassAvailable = true)
		val deviceEntryFromBiometricSource by
		collectLastValue(underTest.deviceEntryFromBiometricSource)

		whenever(kosmos.keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		kosmos.keyguardOcclusionRepository.setShowWhenLockedActivityInfo(onTop = true)
		whenever(keyguardUpdateMonitor.isDeviceInteractive).thenReturn(true)
		keyguardOcclusionRepository.setShowWhenLockedActivityInfo(onTop = true)
		configureDeviceEntryBiometricAuthSuccessState(isFaceAuth = true)
		configureBiometricUnlockState(
		primaryBouncerVisible = false,
		alternateBouncerVisible = true,
		sceneKey = Scenes.Lockscreen,
		)
		runCurrent()

		assertThat(deviceEntryFromBiometricSource).isEqualTo(BiometricUnlockSource.FACE_SENSOR)
		}

		private fun configureDeviceEntryBiometricAuthSuccessState(
		private fun Kosmos.configureDeviceEntryBiometricAuthSuccessState(
		isFingerprintAuth: Boolean = false,
		isFaceAuth: Boolean = false,
		) {
		if (isFingerprintAuth) {
		val successStatus = SuccessFingerprintAuthenticationStatus(0, true)
		kosmos.fakeDeviceEntryFingerprintAuthRepository.setAuthenticationStatus(successStatus)
		fakeDeviceEntryFingerprintAuthRepository.setAuthenticationStatus(
		SuccessFingerprintAuthenticationStatus(0, true)
		)
		}

		if (isFaceAuth) {
		val successStatus: FaceAuthenticationStatus =
		fakeDeviceEntryFaceAuthRepository.setAuthenticationStatus(
		SuccessFaceAuthenticationStatus(
		FaceManager.AuthenticationResult(null, null, 0, true)
		)
		kosmos.fakeDeviceEntryFaceAuthRepository.setAuthenticationStatus(successStatus)
		)
		}
		}

		private fun configureBiometricUnlockState(
		private fun Kosmos.configureBiometricUnlockState(
		primaryBouncerVisible: Boolean,
		alternateBouncerVisible: Boolean,
		sceneKey: SceneKey,
		) {
		kosmos.keyguardBouncerRepository.setAlternateVisible(alternateBouncerVisible)
		kosmos.sceneInteractor.changeScene(sceneKey, "reason")
		if (primaryBouncerVisible) kosmos.sceneInteractor.showOverlay(Overlays.Bouncer, "reason")
		kosmos.sceneInteractor.setTransitionState(
		MutableStateFlow<ObservableTransitionState>(
		keyguardBouncerRepository.setAlternateVisible(alternateBouncerVisible)
		sceneInteractor.changeScene(sceneKey, "reason")
		if (primaryBouncerVisible) {
		sceneInteractor.showOverlay(Overlays.Bouncer, "reason")
		}
		sceneInteractor.setTransitionState(
		flowOf(
		ObservableTransitionState.Idle(
		sceneKey,
		if (primaryBouncerVisible) setOf(Overlays.Bouncer) else emptySet(),
		)
		)
		)
		runCurrent()
		}
		}