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

Commit b2ab705e authored by Chandru S's avatar Chandru S
Browse files

Fix KeyguardIndicationAreaViewModelTest that fails when flexi is enabled 

KeyguardIndicationAreaViewModel uses communalSceneInteractor#isCommunalVisible which relies on SceneInteractor to determine the current scene.

Bug: 409851925
Test: unit test passes with flexi enabled
Flag: com.android.systemui.scene_container
Change-Id: I335be5f994ba93dc4516f89736f55dfd751e84b1
parent 3958663f

packages/SystemUI/tests/src/com/android/systemui/keyguard/ui/viewmodel/KeyguardIndicationAreaViewModelTest.kt

+32 −0
Original line number Diff line number Diff line
@@ -25,6 +25,8 @@ import com.android.systemui.communal.domain.interactor.communalSceneInteractor 
import com.android.systemui.communal.shared.model.CommunalScenes

import com.android.systemui.coroutines.collectLastValue

import com.android.systemui.doze.util.BurnInHelperWrapper

import com.android.systemui.flags.DisableSceneContainer

import com.android.systemui.flags.EnableSceneContainer

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

import com.android.systemui.keyguard.domain.interactor.BurnInInteractor

import com.android.systemui.keyguard.domain.interactor.keyguardInteractor
@@ -34,10 +36,14 @@ import com.android.systemui.keyguard.shared.model.StatusBarState 
import com.android.systemui.keyguard.shared.quickaffordance.KeyguardQuickAffordancePosition

import com.android.systemui.kosmos.testDispatcher

import com.android.systemui.kosmos.testScope

import com.android.systemui.scene.domain.interactor.sceneInteractor

import com.android.systemui.scene.shared.model.Scenes

import com.android.systemui.testKosmos

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

import kotlinx.coroutines.ExperimentalCoroutinesApi

import kotlinx.coroutines.flow.MutableStateFlow

import kotlinx.coroutines.flow.flowOf

import kotlinx.coroutines.test.runCurrent

import kotlinx.coroutines.test.runTest

import org.junit.Before

import org.junit.Test
@@ -46,6 +52,7 @@ import org.mockito.kotlin.any 
import org.mockito.kotlin.doReturn

import org.mockito.kotlin.mock



@OptIn(ExperimentalCoroutinesApi::class)

@SmallTest

class KeyguardIndicationAreaViewModelTest() : SysuiTestCase() {

    private val kosmos = testKosmos()
@@ -108,6 +115,7 @@ class KeyguardIndicationAreaViewModelTest() : SysuiTestCase() { 
        }



    @Test

    @DisableSceneContainer

    fun visibilityWhenCommunalShowing() =

        testScope.runTest {

            keyguardRepository.setStatusBarState(StatusBarState.KEYGUARD)
@@ -127,6 +135,30 @@ class KeyguardIndicationAreaViewModelTest() : SysuiTestCase() { 
            assertThat(visible).isFalse()

        }



    @Test

    @EnableSceneContainer

    fun withSceneContainerEnabled_visibilityWhenCommunalShowing() =

        testScope.runTest {

            keyguardRepository.setStatusBarState(StatusBarState.KEYGUARD)

            kosmos.sceneInteractor.setTransitionState(

                MutableStateFlow(ObservableTransitionState.Idle(currentScene = Scenes.Communal))

            )

            runCurrent()



            val visible by collectLastValue(underTest.visible)



            assertThat(visible).isTrue()

            keyguardRepository.setStatusBarState(StatusBarState.SHADE)

            runCurrent()

            assertThat(visible).isTrue()



            kosmos.sceneInteractor.setTransitionState(

                MutableStateFlow(ObservableTransitionState.Idle(currentScene = Scenes.Lockscreen))

            )

            runCurrent()

            assertThat(visible).isFalse()

        }



    companion object {

        private const val DEFAULT_BURN_IN_OFFSET = 5

        private const val RETURNED_BURN_IN_OFFSET = 3