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

Commit 7bc1879d authored by Treehugger Robot's avatar Treehugger Robot Committed by Android (Google) Code Review
Browse files

Merge "Migrate to Kosmos.runTest" into main

parents 3931b835 a3a5df1e

packages/SystemUI/multivalentTests/src/com/android/systemui/biometrics/ui/viewmodel/DeviceEntryUdfpsTouchOverlayViewModelTest.kt

+24 −35
Original line number Diff line number Diff line
@@ -22,20 +22,21 @@ import androidx.test.ext.junit.runners.AndroidJUnit4 
import androidx.test.filters.SmallTest

import com.android.systemui.SysuiTestCase

import com.android.systemui.bouncer.data.repository.fakeKeyguardBouncerRepository

import com.android.systemui.coroutines.collectLastValue

import com.android.systemui.flags.Flags

import com.android.systemui.flags.fakeFeatureFlagsClassic

import com.android.systemui.keyguard.data.repository.fakeKeyguardTransitionRepository

import com.android.systemui.keyguard.shared.model.KeyguardState

import com.android.systemui.keyguard.ui.viewmodel.fakeDeviceEntryIconViewModelTransition

import com.android.systemui.kosmos.Kosmos

import com.android.systemui.kosmos.collectLastValue

import com.android.systemui.kosmos.runTest

import com.android.systemui.kosmos.testScope

import com.android.systemui.kosmos.useUnconfinedTestDispatcher

import com.android.systemui.securelockdevice.data.repository.fakeSecureLockDeviceRepository

import com.android.systemui.statusbar.phone.SystemUIDialogManager

import com.android.systemui.statusbar.phone.mockSystemUIDialogManager

import com.android.systemui.testKosmos

import com.google.common.truth.Truth.assertThat

import kotlinx.coroutines.test.runCurrent

import kotlinx.coroutines.test.runTest

import org.junit.Before

import org.junit.Test

import org.junit.runner.RunWith
@@ -48,15 +49,10 @@ import org.mockito.MockitoAnnotations 
@RunWith(AndroidJUnit4::class)

class DeviceEntryUdfpsTouchOverlayViewModelTest : SysuiTestCase() {

    private val kosmos =

        testKosmos().apply {

        testKosmos().useUnconfinedTestDispatcher().apply {

            fakeFeatureFlagsClassic.apply { set(Flags.FULL_SCREEN_USER_SWITCHER, true) }

        }

    private val systemUIDialogManager = kosmos.mockSystemUIDialogManager

    private val bouncerRepository = kosmos.fakeKeyguardBouncerRepository

    private val keyguardTransitionRepository = kosmos.fakeKeyguardTransitionRepository

    private val testScope = kosmos.testScope

    private val deviceEntryIconViewModelTransition = kosmos.fakeDeviceEntryIconViewModelTransition

    private val underTest = kosmos.deviceEntryUdfpsTouchOverlayViewModel

    private val Kosmos.underTest by Kosmos.Fixture { deviceEntryUdfpsTouchOverlayViewModel }



    @Captor

    private lateinit var sysuiDialogListenerCaptor: ArgumentCaptor<SystemUIDialogManager.Listener>
@@ -68,13 +64,12 @@ class DeviceEntryUdfpsTouchOverlayViewModelTest : SysuiTestCase() { 


    @Test

    fun dialogShowing_shouldHandleTouchesFalse() =

        testScope.runTest {

        kosmos.runTest {

            val shouldHandleTouches by collectLastValue(underTest.shouldHandleTouches)



            deviceEntryIconViewModelTransition.setDeviceEntryParentViewAlpha(1f)

            runCurrent()

            fakeDeviceEntryIconViewModelTransition.setDeviceEntryParentViewAlpha(1f)



            verify(systemUIDialogManager).registerListener(sysuiDialogListenerCaptor.capture())

            verify(mockSystemUIDialogManager).registerListener(sysuiDialogListenerCaptor.capture())

            sysuiDialogListenerCaptor.value.shouldHideAffordances(true)



            assertThat(shouldHandleTouches).isFalse()
@@ -82,13 +77,12 @@ class DeviceEntryUdfpsTouchOverlayViewModelTest : SysuiTestCase() { 


    @Test

    fun transitionAlphaIsSmall_shouldHandleTouchesFalse() =

        testScope.runTest {

        kosmos.runTest {

            val shouldHandleTouches by collectLastValue(underTest.shouldHandleTouches)



            deviceEntryIconViewModelTransition.setDeviceEntryParentViewAlpha(.3f)

            runCurrent()

            fakeDeviceEntryIconViewModelTransition.setDeviceEntryParentViewAlpha(.3f)



            verify(systemUIDialogManager).registerListener(sysuiDialogListenerCaptor.capture())

            verify(mockSystemUIDialogManager).registerListener(sysuiDialogListenerCaptor.capture())

            sysuiDialogListenerCaptor.value.shouldHideAffordances(false)



            assertThat(shouldHandleTouches).isFalse()
@@ -96,13 +90,12 @@ class DeviceEntryUdfpsTouchOverlayViewModelTest : SysuiTestCase() { 


    @Test

    fun alphaFullyShowing_noDialog_shouldHandleTouchesTrue() =

        testScope.runTest {

        kosmos.runTest {

            val shouldHandleTouches by collectLastValue(underTest.shouldHandleTouches)



            deviceEntryIconViewModelTransition.setDeviceEntryParentViewAlpha(1f)

            runCurrent()

            fakeDeviceEntryIconViewModelTransition.setDeviceEntryParentViewAlpha(1f)



            verify(systemUIDialogManager).registerListener(sysuiDialogListenerCaptor.capture())

            verify(mockSystemUIDialogManager).registerListener(sysuiDialogListenerCaptor.capture())

            sysuiDialogListenerCaptor.value.shouldHideAffordances(false)



            assertThat(shouldHandleTouches).isTrue()
@@ -110,43 +103,39 @@ class DeviceEntryUdfpsTouchOverlayViewModelTest : SysuiTestCase() { 


    @Test

    fun deviceEntryViewAlphaZero_alternateBouncerVisible_shouldHandleTouchesTrue() =

        testScope.runTest {

        kosmos.runTest {

            val shouldHandleTouches by collectLastValue(underTest.shouldHandleTouches)



            deviceEntryIconViewModelTransition.setDeviceEntryParentViewAlpha(0f)

            runCurrent()

            fakeDeviceEntryIconViewModelTransition.setDeviceEntryParentViewAlpha(0f)



            bouncerRepository.setAlternateVisible(true)

            fakeKeyguardBouncerRepository.setAlternateVisible(true)

            assertThat(shouldHandleTouches).isTrue()

        }



    @Test

    fun transitioningToDozing_shouldHandleTouchesTrue() =

        testScope.runTest {

        kosmos.runTest {

            val shouldHandleTouches by collectLastValue(underTest.shouldHandleTouches)



            keyguardTransitionRepository.sendTransitionSteps(

            fakeKeyguardTransitionRepository.sendTransitionSteps(

                from = KeyguardState.LOCKSCREEN,

                to = KeyguardState.DOZING,

                testScope = testScope,

            )

            runCurrent()

            assertThat(shouldHandleTouches).isTrue()

        }



    @EnableFlags(FLAG_SECURE_LOCK_DEVICE)

    @Test

    fun shouldHandleTouchesTrue_duringSecureLockDeviceBiometricAuth() =

        testScope.runTest {

        kosmos.runTest {

            val shouldHandleTouches by collectLastValue(underTest.shouldHandleTouches)



            deviceEntryIconViewModelTransition.setDeviceEntryParentViewAlpha(0f)

            runCurrent()

            fakeDeviceEntryIconViewModelTransition.setDeviceEntryParentViewAlpha(0f)



            kosmos.fakeSecureLockDeviceRepository.onSecureLockDeviceEnabled()

            kosmos.fakeSecureLockDeviceRepository.onSuccessfulPrimaryAuth()

            fakeSecureLockDeviceRepository.onSecureLockDeviceEnabled()

            fakeSecureLockDeviceRepository.onSuccessfulPrimaryAuth()



            runCurrent()

            assertThat(shouldHandleTouches).isTrue()

        }

}