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

Commit d967a712 authored by Maryam Dehaini's avatar Maryam Dehaini
Browse files

Update WindowDecorCaptionRespositoryTests 

Caption state flow is not a shared flow. Updates tests to reflect this.

Bug: 412444236
Test: manual testing
Flag: EXEMPT test fix
Change-Id: I51d5ad957ae39cf9b8a6e76f57fb6eb38557711d
parent 4a1ee6d5

libs/WindowManager/Shell/tests/unittest/src/com/android/wm/shell/desktopmode/WindowDecorCaptionRepositoryTest.kt

+8 −12
Original line number Diff line number Diff line
@@ -27,6 +27,8 @@ import androidx.test.filters.SmallTest 
import com.android.wm.shell.windowdecor.viewholder.AppHandleIdentifier

import com.android.wm.shell.windowdecor.viewholder.AppHandleIdentifier.AppHandleWindowingMode

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

import kotlinx.coroutines.flow.first

import kotlinx.coroutines.test.runTest

import org.junit.Before

import org.junit.Test

import org.junit.runner.RunWith
@@ -42,13 +44,7 @@ class WindowDecorCaptionRepositoryTest { 
    }



    @Test

    fun initialState_noAction_returnsNoCaption() {

        // Check the initial value of `captionStateFlow`.

        assertThat(captionRepository.captionStateFlow.value).isEqualTo(CaptionState.NoCaption())

    }



    @Test

    fun notifyCaptionChange_toAppHandleVisible_updatesStateWithCorrectData() {

    fun notifyCaptionChange_toAppHandleVisible_updatesStateWithCorrectData() = runTest {

        val taskInfo = createTaskInfo(WINDOWING_MODE_FULLSCREEN, GMAIL_PACKAGE_NAME)

        val appHandleCaptionState =

            CaptionState.AppHandle(
@@ -63,11 +59,11 @@ class WindowDecorCaptionRepositoryTest { 


        captionRepository.notifyCaptionChanged(appHandleCaptionState)



        assertThat(captionRepository.captionStateFlow.value).isEqualTo(appHandleCaptionState)

        assertThat(captionRepository.captionStateFlow.first()).isEqualTo(appHandleCaptionState)

    }



    @Test

    fun notifyCaptionChange_toAppChipVisible_updatesStateWithCorrectData() {

    fun notifyCaptionChange_toAppChipVisible_updatesStateWithCorrectData() = runTest {

        val taskInfo = createTaskInfo(WINDOWING_MODE_FREEFORM, GMAIL_PACKAGE_NAME)

        val appHeaderCaptionState =

            CaptionState.AppHeader(
@@ -81,14 +77,14 @@ class WindowDecorCaptionRepositoryTest { 


        captionRepository.notifyCaptionChanged(appHeaderCaptionState)



        assertThat(captionRepository.captionStateFlow.value).isEqualTo(appHeaderCaptionState)

        assertThat(captionRepository.captionStateFlow.first()).isEqualTo(appHeaderCaptionState)

    }



    @Test

    fun notifyCaptionChange_toNoCaption_updatesState() {

    fun notifyCaptionChange_toNoCaption_updatesState() = runTest {

        captionRepository.notifyCaptionChanged(CaptionState.NoCaption())



        assertThat(captionRepository.captionStateFlow.value).isEqualTo(CaptionState.NoCaption())

        assertThat(captionRepository.captionStateFlow.first()).isEqualTo(CaptionState.NoCaption())

    }



    private fun createTaskInfo(