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

Commit 0032e9cf authored by Anton Potapov's avatar Anton Potapov
Browse files

Move commons from vendor System UI to make it accessible from the Android TV name space 

This change introduces utils/screenshot directory with a build target
for the utilitis specific to screenshot tests

Flag: EXEMPT TEST_ONLY
Bug: 387284759
Test: atest VolumeDialogScreenshotTest
Change-Id: Ifd979048046b0a235093446413dee71440877630
parent 78aa5160

packages/SystemUI/Android.bp

+9 −0
Original line number Diff line number Diff line
@@ -609,6 +609,15 @@ filegroup { 
    path: "tests/utils",

}



filegroup {

    name: "SystemUI-screenshot-tests-utils",

    srcs: [

        "tests/utils/screenshot/src/**/*.java",

        "tests/utils/screenshot/src/**/*.kt",

    ],

    path: "tests/utils/screenshot",

}



filegroup {

    name: "SystemUI-robotest-utils",

    srcs: [

packages/SystemUI/tests/utils/screenshot/src/com/android/systemui/volume/dialog/VolumeDialogScreenshotTestCommons.kt

0 → 100644
+157 −0
Original line number Diff line number Diff line 
/*

 * Copyright (C) 2025 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */



package com.android.systemui.volume.dialog



import android.app.ActivityManager

import android.content.applicationContext

import android.content.packageManager

import android.content.pm.PackageManager

import android.media.AudioManager

import android.media.session.PlaybackState

import android.provider.Settings

import android.view.View

import com.android.settingslib.notification.data.repository.updateNotificationPolicy

import com.android.settingslib.volume.shared.model.RingerMode

import com.android.systemui.accessibility.data.repository.accessibilityRepository

import com.android.systemui.kosmos.Kosmos

import com.android.systemui.kosmos.testScope

import com.android.systemui.plugins.VolumeDialogController

import com.android.systemui.plugins.fakeVolumeDialogController

import com.android.systemui.res.R

import com.android.systemui.statusbar.policy.data.repository.zenModeRepository

import com.android.systemui.statusbar.policy.devicePostureController

import com.android.systemui.statusbar.policy.fakeDevicePostureController

import com.android.systemui.statusbar.policy.fakeDeviceProvisionedController

import com.android.systemui.volume.data.repository.audioSystemRepository

import com.android.systemui.volume.data.repository.fakeAudioRepository

import com.android.systemui.volume.dialog.data.repository.volumeDialogVisibilityRepository

import com.android.systemui.volume.dialog.shared.model.VolumeDialogVisibilityModel

import com.android.systemui.volume.dialog.sliders.domain.model.VolumeDialogSliderType

import com.android.systemui.volume.dialog.sliders.domain.model.volumeDialogSliderType

import com.android.systemui.volume.localMediaController

import com.android.systemui.volume.localPlaybackStateBuilder

import com.android.systemui.volume.mediaControllerRepository

import com.android.systemui.volume.panel.component.mediaoutput.domain.interactor.FakeMediaControllerInteractor

import com.android.systemui.volume.panel.component.mediaoutput.domain.interactor.mediaControllerInteractor

import kotlin.time.Duration.Companion.seconds

import kotlinx.coroutines.runBlocking

import kotlinx.coroutines.test.TestResult

import kotlinx.coroutines.test.runTest

import kotlinx.coroutines.withTimeout

import org.mockito.kotlin.whenever

import platform.test.screenshot.utils.compose.ComposeScreenshotTestRule



/** Timeout for the data and domain logic to be ready for the screenshot */

private val initializationTimeout = 5.seconds



/**

 * Timeout of each test.

 *

 * This is a sum of the TestScope#runTest function default timeout and

 * ComposeScreenshotTestRule#screenshotTest function timeout.

 */

private val testTimeout = 20.seconds



object VolumeDialogScreenshotTestCommons {



    private fun Kosmos.setupTestDefaults() {

        accessibilityRepository.setRecommendedTimeout(10.seconds)

        devicePostureController = fakeDevicePostureController

        whenever(packageManager.hasSystemFeature(PackageManager.FEATURE_LEANBACK)).thenReturn(false)



        mediaControllerInteractor = FakeMediaControllerInteractor()

        localPlaybackStateBuilder.setState(

            /* state = */ PlaybackState.STATE_STOPPED,

            /* position = */ 0,

            /* playbackSpeed = */ 1f,

        )

        mediaControllerRepository.setActiveSessions(listOf(localMediaController))

        audioSystemRepository.setIsSingleVolume(false)

        setRingerMode(AudioManager.RINGER_MODE_NORMAL)

        volumeDialogSliderType = VolumeDialogSliderType.Stream(AudioManager.STREAM_MUSIC)



        with(fakeDeviceProvisionedController) { setUserSetup(currentUser, true) }



        with(fakeVolumeDialogController) {

            setHasVibrator(true)

            setActiveStream(AudioManager.STREAM_MUSIC)

            updateState {

                states[AudioManager.STREAM_MUSIC] =

                    VolumeDialogController.StreamState().also {

                        it.level = 10

                        it.levelMin = 0

                        it.levelMax = 20

                        it.name = R.string.stream_music

                    }

                states[AudioManager.STREAM_RING] =

                    VolumeDialogController.StreamState().also {

                        it.level = 4

                        it.levelMin = 0

                        it.levelMax = 7

                        it.name = R.string.stream_ring

                    }

            }

        }



        zenModeRepository.apply {

            updateNotificationPolicy()

            updateZenMode(Settings.Global.ZEN_MODE_OFF)

        }



        volumeDialogVisibilityRepository.updateVisibility {

            VolumeDialogVisibilityModel.Visible(

                reason = 0,

                keyguardLocked = false,

                lockTaskModeState = ActivityManager.LOCK_TASK_MODE_NONE,

            )

        }

    }



    fun Kosmos.screenshotState(

        screenshotRule: ComposeScreenshotTestRule,

        identifier: String,

        prepareState: suspend Kosmos.() -> Unit,

        onDialogCreated: Kosmos.() -> Unit = {},

    ): TestResult =

        testScope.runTest(testTimeout) {

            screenshotRule.dialogScreenshotTest(

                goldenIdentifier = identifier,

                frameLimit = 500,

                shouldWaitForTheDialog = { dialog ->

                    val isDialogFullyVisible =

                        dialog.requireViewById<View>(R.id.volume_dialog).translationX == 0f

                    !isDialogFullyVisible

                },

                waitForIdle = { onDialogCreated() },

            ) { activity ->

                runBlocking {

                    withTimeout(initializationTimeout) {

                        applicationContext = activity

                        setupTestDefaults()

                        prepareState()

                    }

                }



                volumeDialog

            }

        }



    fun Kosmos.setRingerMode(mode: Int) {

        fakeVolumeDialogController.updateState { this.ringerModeInternal = mode }

        fakeAudioRepository.setRingerMode(RingerMode(mode))

    }

}

packages/SystemUI/tests/utils/src/com/android/systemui/volume/dialog/ui/binder/VolumeDialogViewBinderKosmos.kt

+12 −5
Original line number Diff line number Diff line
@@ -25,6 +25,17 @@ import com.android.systemui.volume.dialog.ui.utils.jankListenerFactory 
import com.android.systemui.volume.dialog.ui.viewmodel.volumeDialogViewModel

import com.android.systemui.volume.dialog.utils.volumeTracer



val Kosmos.systemUiVolumeDialogViewBinders by

    Kosmos.Fixture {

        listOf(

            volumeDialogSlidersViewBinder,

            volumeDialogRingerViewBinder,

            volumeDialogSettingsButtonViewBinder,

        )

    }



var Kosmos.volumeDialogViewBinders: List<ViewBinder> by

    Kosmos.Fixture { systemUiVolumeDialogViewBinders }

val Kosmos.volumeDialogViewBinder by

    Kosmos.Fixture {

        VolumeDialogViewBinder(
@@ -32,10 +43,6 @@ val Kosmos.volumeDialogViewBinder by 
            volumeDialogViewModel,

            jankListenerFactory,

            volumeTracer,

            listOf(

                volumeDialogSlidersViewBinder,

                volumeDialogRingerViewBinder,

                volumeDialogSettingsButtonViewBinder,

            ),

            volumeDialogViewBinders,

        )

    }