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

Commit e2ca646b authored by Anton Potapov's avatar Anton Potapov
Browse files

Add Kosmoses and ScreenCaptureComponentInteractorTest 

Flag: com.android.systemui.new_screen_record_toolbar
Bug: 427682447
Test: atest ScreenCaptureComponentInteractorTest
Change-Id: I714c85db6426d81624ea14bbf30028051b9ea7f2
parent 39bf2614

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

        }

}