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

Commit b20b6356 authored by Caitlin Shkuratov's avatar Caitlin Shkuratov
Browse files

[SB][Chips] Use hydrator for HomeStatusBarViewModel.ongoingActivityChips 

Fixes: 399865510
Bug: 372657935
Flag: com.android.systemui.status_bar_chips_modernization
Test: smoke test of the status bar chips (add & remove each type of
chip)

Change-Id: I07b1931f33ecb1d91b921ebe3af273118d30870b
parent 372004b6

packages/SystemUI/multivalentTests/src/com/android/systemui/statusbar/pipeline/shared/ui/viewmodel/FakeHomeStatusBarViewModel.kt

+1 −3
Original line number Diff line number Diff line
@@ -54,9 +54,7 @@ class FakeHomeStatusBarViewModel( 
        MutableStateFlow(OngoingActivityChipModel.Inactive())



    override val ongoingActivityChips =

        MutableStateFlow(

        ChipsVisibilityModel(MultipleOngoingActivityChipsModel(), areChipsAllowed = false)

        )



    override val ongoingActivityChipsLegacy =

        MutableStateFlow(MultipleOngoingActivityChipsModelLegacy())

packages/SystemUI/multivalentTests/src/com/android/systemui/statusbar/pipeline/shared/ui/viewmodel/HomeStatusBarViewModelImplTest.kt

+14 −27
Original line number Diff line number Diff line
@@ -49,6 +49,7 @@ import com.android.systemui.kosmos.runCurrent 
import com.android.systemui.kosmos.runTest

import com.android.systemui.kosmos.testScope

import com.android.systemui.kosmos.useUnconfinedTestDispatcher

import com.android.systemui.lifecycle.activateIn

import com.android.systemui.log.assertLogsWtf

import com.android.systemui.mediaprojection.data.model.MediaProjectionState

import com.android.systemui.mediaprojection.data.repository.fakeMediaProjectionRepository
@@ -107,7 +108,8 @@ import org.junit.runner.RunWith 
@RunWith(AndroidJUnit4::class)

class HomeStatusBarViewModelImplTest : SysuiTestCase() {

    private val kosmos = testKosmos().useUnconfinedTestDispatcher()

    private val Kosmos.underTest by Kosmos.Fixture { kosmos.homeStatusBarViewModel }

    private val Kosmos.underTest by

        Kosmos.Fixture { kosmos.homeStatusBarViewModel.also { it.activateIn(kosmos.testScope) } }



    @Before

    fun setUp() {
@@ -891,32 +893,26 @@ class HomeStatusBarViewModelImplTest : SysuiTestCase() { 
    @EnableChipsModernization

    fun ongoingActivityChips_statusBarHidden_noSecureCamera_noHun_notAllowed() =

        kosmos.runTest {

            val latest by collectLastValue(underTest.ongoingActivityChips)



            // home status bar not allowed

            kosmos.sceneContainerRepository.snapToScene(Scenes.Lockscreen)

            kosmos.keyguardOcclusionRepository.setShowWhenLockedActivityInfo(false, taskInfo = null)



            assertThat(latest!!.areChipsAllowed).isFalse()

            assertThat(underTest.ongoingActivityChips.areChipsAllowed).isFalse()

        }



    @Test

    @EnableChipsModernization

    fun ongoingActivityChips_statusBarNotHidden_noSecureCamera_noHun_isAllowed() =

        kosmos.runTest {

            val latest by collectLastValue(underTest.ongoingActivityChips)



            transitionKeyguardToGone()



            assertThat(latest!!.areChipsAllowed).isTrue()

            assertThat(underTest.ongoingActivityChips.areChipsAllowed).isTrue()

        }



    @Test

    @EnableChipsModernization

    fun ongoingActivityChips_statusBarNotHidden_secureCamera_noHun_notAllowed() =

        kosmos.runTest {

            val latest by collectLastValue(underTest.ongoingActivityChips)



            fakeKeyguardTransitionRepository.sendTransitionSteps(

                from = KeyguardState.LOCKSCREEN,

                to = KeyguardState.OCCLUDED,
@@ -924,7 +920,7 @@ class HomeStatusBarViewModelImplTest : SysuiTestCase() { 
            )

            kosmos.keyguardInteractor.onCameraLaunchDetected(CAMERA_LAUNCH_SOURCE_POWER_DOUBLE_TAP)



            assertThat(latest!!.areChipsAllowed).isFalse()

            assertThat(underTest.ongoingActivityChips.areChipsAllowed).isFalse()

        }



    @Test
@@ -932,8 +928,6 @@ class HomeStatusBarViewModelImplTest : SysuiTestCase() { 
    @EnableChipsModernization

    fun ongoingActivityChips_statusBarNotHidden_noSecureCamera_hunBySystem_noHunFlagOff_notAllowed() =

        kosmos.runTest {

            val latest by collectLastValue(underTest.ongoingActivityChips)



            transitionKeyguardToGone()



            headsUpNotificationRepository.setNotifications(
@@ -943,7 +937,7 @@ class HomeStatusBarViewModelImplTest : SysuiTestCase() { 
                )

            )



            assertThat(latest!!.areChipsAllowed).isFalse()

            assertThat(underTest.ongoingActivityChips.areChipsAllowed).isFalse()

        }



    @Test
@@ -951,8 +945,6 @@ class HomeStatusBarViewModelImplTest : SysuiTestCase() { 
    @EnableChipsModernization

    fun ongoingActivityChips_statusBarNotHidden_noSecureCamera_hunByUser_noHunFlagOff_isAllowed() =

        kosmos.runTest {

            val latest by collectLastValue(underTest.ongoingActivityChips)



            transitionKeyguardToGone()



            headsUpNotificationRepository.setNotifications(
@@ -962,16 +954,14 @@ class HomeStatusBarViewModelImplTest : SysuiTestCase() { 
                )

            )



            assertThat(latest!!.areChipsAllowed).isTrue()

            assertThat(underTest.ongoingActivityChips.areChipsAllowed).isTrue()

        }



    @Test

    @EnableFlags(StatusBarNoHunBehavior.FLAG_NAME)

    @EnableChipsModernization

    fun ongoingActivityChips_tatusBarNotHidden_noSecureCamera_hunBySystem_noHunFlagOn_isAllowed() =

    fun ongoingActivityChips_statusBarNotHidden_noSecureCamera_hunBySystem_noHunFlagOn_isAllowed() =

        kosmos.runTest {

            val latest by collectLastValue(underTest.ongoingActivityChips)



            transitionKeyguardToGone()



            headsUpNotificationRepository.setNotifications(
@@ -981,7 +971,7 @@ class HomeStatusBarViewModelImplTest : SysuiTestCase() { 
                )

            )



            assertThat(latest!!.areChipsAllowed).isTrue()

            assertThat(underTest.ongoingActivityChips.areChipsAllowed).isTrue()

        }



    @Test
@@ -989,8 +979,6 @@ class HomeStatusBarViewModelImplTest : SysuiTestCase() { 
    @EnableChipsModernization

    fun ongoingActivityChips_statusBarNotHidden_noSecureCamera_hunByUser_noHunFlagOn_isAllowed() =

        kosmos.runTest {

            val latest by collectLastValue(underTest.ongoingActivityChips)



            transitionKeyguardToGone()



            headsUpNotificationRepository.setNotifications(
@@ -1000,7 +988,7 @@ class HomeStatusBarViewModelImplTest : SysuiTestCase() { 
                )

            )



            assertThat(latest!!.areChipsAllowed).isTrue()

            assertThat(underTest.ongoingActivityChips.areChipsAllowed).isTrue()

        }



    @Test
@@ -1008,17 +996,16 @@ class HomeStatusBarViewModelImplTest : SysuiTestCase() { 
    @EnableChipsModernization

    fun ongoingActivityChips_followsChipsViewModel() =

        kosmos.runTest {

            val latest by collectLastValue(underTest.ongoingActivityChips)

            transitionKeyguardToGone()



            screenRecordRepository.screenRecordState.value = ScreenRecordModel.Recording



            assertIsScreenRecordChip(latest!!.chips.active[0])

            assertIsScreenRecordChip(underTest.ongoingActivityChips.chips.active[0])



            addOngoingCallState(key = "call")



            assertIsScreenRecordChip(latest!!.chips.active[0])

            assertIsCallChip(latest!!.chips.active[1], "call", context)

            assertIsScreenRecordChip(underTest.ongoingActivityChips.chips.active[0])

            assertIsCallChip(underTest.ongoingActivityChips.chips.active[1], "call", context)

        }



    @Test

packages/SystemUI/src/com/android/systemui/statusbar/pipeline/shared/ui/composable/StatusBarRoot.kt

+1 −2
Original line number Diff line number Diff line
@@ -196,9 +196,8 @@ fun StatusBarRoot( 


                                setContent {

                                    PlatformTheme {

                                        val chipsVisibilityModel by

                                        val chipsVisibilityModel =

                                            statusBarViewModel.ongoingActivityChips

                                                .collectAsStateWithLifecycle()

                                        if (chipsVisibilityModel.areChipsAllowed) {

                                            OngoingActivityChips(

                                                chips = chipsVisibilityModel.chips,

packages/SystemUI/src/com/android/systemui/statusbar/pipeline/shared/ui/viewmodel/HomeStatusBarViewModel.kt

+17 −15
Original line number Diff line number Diff line
@@ -130,7 +130,7 @@ interface HomeStatusBarViewModel : Activatable { 
    val primaryOngoingActivityChip: StateFlow<OngoingActivityChipModel>



    /** All supported activity chips, whether they are currently active or not. */

    val ongoingActivityChips: StateFlow<ChipsVisibilityModel>

    val ongoingActivityChips: ChipsVisibilityModel



    /**

     * The multiple ongoing activity chips that should be shown on the left-hand side of the status
@@ -461,24 +461,26 @@ constructor( 
            isHomeStatusBarAllowed && !isSecureCameraActive && !hideStartSideContentForHeadsUp

        }



    override val ongoingActivityChips =

    private val chipsVisibilityModel: Flow<ChipsVisibilityModel> =

        combine(ongoingActivityChipsViewModel.chips, canShowOngoingActivityChips) { chips, canShow

            ->

            ChipsVisibilityModel(chips, areChipsAllowed = canShow)

        }

            .stateIn(

                bgScope,

                SharingStarted.WhileSubscribed(),



    override val ongoingActivityChips: ChipsVisibilityModel by

        hydrator.hydratedStateOf(

            traceName = "ongoingActivityChips",

            initialValue =

                ChipsVisibilityModel(

                    chips = MultipleOngoingActivityChipsModel(),

                    areChipsAllowed = false,

                ),

            source = chipsVisibilityModel,

        )



    private val hasOngoingActivityChips =

        if (StatusBarChipsModernization.isEnabled) {

            ongoingActivityChips.map { it.chips.active.any { chip -> !chip.isHidden } }

            chipsVisibilityModel.map { it.chips.active.any { chip -> !chip.isHidden } }

        } else if (StatusBarNotifChips.isEnabled) {

            ongoingActivityChipsLegacy.map { it.primary is OngoingActivityChipModel.Active }

        } else {