Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 77ce36bb authored by burakov's avatar burakov Committed by Danny Burakov
Browse files

[Dual Shade] Fix DeviceEntrySourceInteractorTest for DualShade on. 

This ensures these unit tests do not fail when Dual Shade is the default
shade experience in scene container.

Bug: 376411622
Test: Updated unit tests.
Flag: com.android.systemui.scene_container
Change-Id: I35810ff818b0a22a6de0c2f5509d40d605bc023e
parent d4027635

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()

    }

}