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

Commit e649ae0f authored by Anton Potapov's avatar Anton Potapov Committed by Android (Google) Code Review
Browse files

Merge "Add Kosmoses and ScreenCaptureComponentInteractorTest" into main

parents 47fea8e9 e2ca646b

packages/SystemUI/multivalentTests/src/com/android/systemui/screencapture/domain/interactor/ScreenCaptureComponentInteractorTest.kt

0 → 100644
+98 −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.screencapture.domain.interactor



import android.media.projection.StopReason

import android.view.Display

import androidx.test.ext.junit.runners.AndroidJUnit4

import androidx.test.filters.SmallTest

import com.android.systemui.SysuiTestCase

import com.android.systemui.kosmos.Kosmos

import com.android.systemui.kosmos.backgroundScope

import com.android.systemui.kosmos.collectLastValue

import com.android.systemui.kosmos.runTest

import com.android.systemui.screencapture.common.shared.model.ScreenCaptureType

import com.android.systemui.screencapture.common.shared.model.ScreenCaptureUiParameters

import com.android.systemui.screencapture.common.shared.model.recordScreenCaptureUiParameters

import com.android.systemui.screenrecord.ScreenRecordingAudioSource

import com.android.systemui.screenrecord.domain.ScreenRecordingParameters

import com.android.systemui.screenrecord.domain.interactor.screenRecordingServiceInteractor

import com.android.systemui.testKosmosNew

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

import kotlinx.coroutines.launch

import org.junit.Test

import org.junit.runner.RunWith



@SmallTest

@RunWith(AndroidJUnit4::class)

class ScreenCaptureComponentInteractorTest : SysuiTestCase() {



    private val kosmos = testKosmosNew()

    private val underTest by lazy { kosmos.screenCaptureComponentInteractor }



    @Test

    fun testRecordComponentLifecycle() =

        kosmos.runTest {

            testComponentLifecycle(

                screenCaptureType = ScreenCaptureType.RECORD,

                parameters = recordScreenCaptureUiParameters,

                startCapture = {

                    screenRecordingServiceInteractor.startRecording(

                        ScreenRecordingParameters(

                            captureTarget = null,

                            audioSource = ScreenRecordingAudioSource.NONE,

                            displayId = Display.DEFAULT_DISPLAY,

                            shouldShowTaps = false,

                        )

                    )

                },

                stopCapture = {

                    screenRecordingServiceInteractor.stopRecording(StopReason.STOP_HOST_APP)

                },

            )

        }



    private fun Kosmos.testComponentLifecycle(

        screenCaptureType: ScreenCaptureType,

        parameters: ScreenCaptureUiParameters,

        startCapture: () -> Unit,

        stopCapture: () -> Unit,

    ) {

        backgroundScope.launch { underTest.initialize() }

        val component by collectLastValue(underTest.screenCaptureComponent(screenCaptureType))

        assertThat(component).isNull()



        screenCaptureUiInteractor.show(parameters)

        assertThat(component).isNotNull()



        screenCaptureUiInteractor.hide(screenCaptureType)

        assertThat(component).isNull()



        screenCaptureUiInteractor.show(parameters)

        startCapture()

        screenCaptureUiInteractor.hide(screenCaptureType)

        val capturingComponent = component

        assertThat(capturingComponent).isNotNull()



        screenCaptureUiInteractor.show(parameters)

        assertThat(component).isSameInstanceAs(capturingComponent)



        screenCaptureUiInteractor.hide(screenCaptureType)

        stopCapture()

        assertThat(component).isNull()

    }

}

packages/SystemUI/tests/utils/src/com/android/systemui/screencapture/common/ScreenCaptureUiComponentKosmos.kt

0 → 100644
+48 −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.screencapture.common



import android.view.Display

import android.view.Window

import com.android.systemui.kosmos.Kosmos

import com.android.systemui.screencapture.common.shared.model.ScreenCaptureType

import com.android.systemui.screencapture.common.ui.compose.ScreenCaptureContent

import com.android.systemui.screencapture.common.ui.compose.screenCaptureContents

import kotlinx.coroutines.CoroutineScope



fun Kosmos.screenCaptureUiComponentBuilder(

    type: ScreenCaptureType

): ScreenCaptureUiComponent.Builder =

    object : ScreenCaptureUiComponent.Builder {



        private lateinit var scope: CoroutineScope



        override fun setScope(scope: CoroutineScope): ScreenCaptureUiComponent.Builder {

            this.scope = scope

            return this

        }



        override fun setDisplay(display: Display): ScreenCaptureUiComponent.Builder = this



        override fun setWindow(window: Window?): ScreenCaptureUiComponent.Builder = this



        override fun build(): ScreenCaptureUiComponent =

            object : ScreenCaptureUiComponent {

                override val screenCaptureContent: ScreenCaptureContent

                    get() = screenCaptureContents.getValue(type)

            }

    }

packages/SystemUI/tests/utils/src/com/android/systemui/screencapture/common/ui/compose/ScreenCaptureContentKosmos.kt

0 → 100644
+24 −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.screencapture.common.ui.compose



import com.android.systemui.kosmos.Kosmos

import com.android.systemui.screencapture.common.shared.model.ScreenCaptureType

import com.android.systemui.screencapture.record.ui.compose.screenCaptureRecordContent



var Kosmos.screenCaptureContents: Map<ScreenCaptureType, ScreenCaptureContent> by

    Kosmos.Fixture { mapOf(ScreenCaptureType.RECORD to screenCaptureRecordContent) }

packages/SystemUI/tests/utils/src/com/android/systemui/screencapture/data/repository/ScreenCaptureComponentRepositoryKosmos.kt

0 → 100644
+21 −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.screencapture.data.repository



import com.android.systemui.kosmos.Kosmos



val Kosmos.screenCaptureComponentRepository by Kosmos.Fixture { ScreenCaptureComponentRepository() }

packages/SystemUI/tests/utils/src/com/android/systemui/screencapture/domain/interactor/ScreenCaptureComponentInteractorKosmos.kt

0 → 100644
+65 −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.screencapture.domain.interactor



import com.android.systemui.kosmos.Kosmos

import com.android.systemui.kosmos.testDispatcher

import com.android.systemui.screencapture.common.ScreenCaptureComponent

import com.android.systemui.screencapture.common.shared.model.ScreenCaptureUiParameters

import com.android.systemui.screencapture.data.repository.screenCaptureComponentRepository

import com.android.systemui.screencapture.ui.ScreenCaptureUi

import com.android.systemui.screencapture.ui.screenCaptureUiFactory

import com.android.systemui.screenrecord.domain.interactor.screenRecordingServiceInteractor

import kotlinx.coroutines.CoroutineScope



val Kosmos.screenCaptureComponentInteractor by

    Kosmos.Fixture {

        ScreenCaptureComponentInteractor(

            dispatcherContext = testDispatcher,

            repository = screenCaptureComponentRepository,

            screenCaptureUiInteractor = screenCaptureUiInteractor,

            componentBuilder = FakeScreenCaptureComponentBuilder(this),

            screenRecordingServiceInteractor = screenRecordingServiceInteractor,

        )

    }



private class FakeScreenCaptureComponentBuilder(private val kosmos: Kosmos) :

    ScreenCaptureComponent.Builder {



    private lateinit var scope: CoroutineScope

    private lateinit var parameters: ScreenCaptureUiParameters



    override fun setScope(scope: CoroutineScope): ScreenCaptureComponent.Builder {

        this.scope = scope

        return this

    }



    override fun setParameters(

        parameters: ScreenCaptureUiParameters

    ): ScreenCaptureComponent.Builder {

        this.parameters = parameters

        return this

    }



    override fun build(): ScreenCaptureComponent =

        object : ScreenCaptureComponent {

            override fun coroutineScope(): CoroutineScope = scope



            override fun screenCaptureUiFactory(): ScreenCaptureUi.Factory =

                kosmos.screenCaptureUiFactory

        }

}