Merge "b/399823962 Add new test case for using app handle menu to go to split screen" into main

/*

 * 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.wm.shell.flicker.fundamentals

import android.platform.test.annotations.RequiresDevice

import android.tools.NavBar

import android.tools.flicker.assertions.FlickerTest

import android.tools.flicker.junit.FlickerParametersRunnerFactory

import android.tools.flicker.legacy.FlickerBuilder

import android.tools.flicker.legacy.LegacyFlickerTest

import android.tools.flicker.legacy.LegacyFlickerTestFactory

import com.android.wm.shell.flicker.DesktopModeBaseTest

import com.android.wm.shell.scenarios.EnterSplitScreenWithAppHandleMenu

import com.android.wm.shell.Utils

import com.android.wm.shell.flicker.utils.appWindowOnTopAtEnd

import com.android.wm.shell.flicker.utils.layerBecomesVisible

import com.android.wm.shell.flicker.utils.splitScreenDividerBecomesVisible

import org.junit.Rule

import org.junit.Test

import org.junit.runner.RunWith

import org.junit.runners.Parameterized

/**

 * Enter the app in split screen via app handle menu.

 */

@RequiresDevice

@RunWith(Parameterized::class)

@Parameterized.UseParametersRunnerFactory(FlickerParametersRunnerFactory::class)

class EnterSplitScreenWithAppHandleMenuFlickerTest(flicker: LegacyFlickerTest) :

    DesktopModeBaseTest(flicker) {

    inner class EnterSplitScreenWithAppHandleMenuScenario : EnterSplitScreenWithAppHandleMenu(flicker.scenario.startRotation)

    @Rule

    @JvmField

    val testSetupRule = Utils.testSetupRule(NavBar.MODE_GESTURAL, flicker.scenario.startRotation)

    val scenario = EnterSplitScreenWithAppHandleMenuScenario()

    private val firstApp = scenario.testApp

    private val secondApp = scenario.calculatorApp

    override val transition: FlickerBuilder.() -> Unit

        get() = {

            setup {

                scenario.setup()

            }

            transitions {

                scenario.enterSplitScreenFromAppHandle()

            }

            teardown {

                scenario.teardown()

            }

        }

    @Test

    fun appWindowOnTopAtEnd() = flicker.appWindowOnTopAtEnd(firstApp)

    @Test

    fun splitScreenDividerBecomesVisible() = flicker.splitScreenDividerBecomesVisible()

    @Test

    fun layerBecomesVisible() = flicker.layerBecomesVisible(secondApp)

    companion object {

        @Parameterized.Parameters(name = "{0}")

        @JvmStatic

        fun getParams(): Collection<FlickerTest> {

            return LegacyFlickerTestFactory.nonRotationTests(

                supportedNavigationModes = listOf(NavBar.MODE_GESTURAL)

            )

        }

    }

}

/*

 * 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.wm.shell.functional.fundamentals

import android.platform.test.annotations.Postsubmit

import android.platform.test.rule.ScreenRecordRule

import com.android.wm.shell.scenarios.EnterSplitScreenWithAppHandleMenu

import org.junit.runner.RunWith

import org.junit.runners.BlockJUnit4ClassRunner

/* Functional test for [EnterSplitScreenWithAppHandleMenu]. */

@RunWith(BlockJUnit4ClassRunner::class)

@Postsubmit

@ScreenRecordRule.ScreenRecord

class EnterSplitScreenWithAppHandleMenuTest : EnterSplitScreenWithAppHandleMenu()

/*

 * 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.wm.shell.scenarios

import android.app.Instrumentation

import android.tools.NavBar

import android.tools.PlatformConsts.DEFAULT_DISPLAY

import android.tools.Rotation

import android.tools.device.apphelpers.CalculatorAppHelper

import android.tools.flicker.rules.ChangeDisplayOrientationRule

import android.tools.traces.parsers.WindowManagerStateHelper

import android.view.KeyEvent.KEYCODE_META_RIGHT

import androidx.test.platform.app.InstrumentationRegistry

import androidx.test.uiautomator.UiDevice

import com.android.launcher3.tapl.LauncherInstrumentation

import com.android.server.wm.flicker.helpers.DesktopModeAppHelper

import com.android.server.wm.flicker.helpers.KeyEventHelper

import com.android.server.wm.flicker.helpers.SimpleAppHelper

import com.android.wm.shell.Utils

import com.android.wm.shell.shared.desktopmode.DesktopState

import org.junit.After

import org.junit.Assume

import org.junit.Before

import org.junit.Ignore

import org.junit.Rule

import org.junit.Test

@Ignore("Test Base Class")

abstract class EnterSplitScreenWithAppHandleMenu(val rotation: Rotation = Rotation.ROTATION_0) : TestScenarioBase() {

    private val instrumentation: Instrumentation = InstrumentationRegistry.getInstrumentation()

    private val tapl = LauncherInstrumentation()

    private val wmHelper = WindowManagerStateHelper(instrumentation)

    private val device = UiDevice.getInstance(instrumentation)

    private val keyEventHelper = KeyEventHelper(InstrumentationRegistry.getInstrumentation())

    private val simpleAppHelper = SimpleAppHelper(instrumentation)

    val calculatorApp = CalculatorAppHelper(instrumentation)

    val testApp = DesktopModeAppHelper(simpleAppHelper)

    @Rule

    @JvmField

    val testSetupRule = Utils.testSetupRule(NavBar.MODE_GESTURAL, rotation)

    @Before

    fun setup() {

        Assume.assumeTrue(

            DesktopState.fromContext(instrumentation.context)

                .isDesktopModeSupportedOnDisplay(DEFAULT_DISPLAY)

        )

        tapl.setEnableRotation(true)

        tapl.setExpectedRotation(rotation.value)

        tapl.enableTransientTaskbar(false)

        ChangeDisplayOrientationRule.setRotation(rotation)

    }

    @Test

    open fun enterSplitScreenFromAppHandle() {

        // Launch app in order to enter split screen

        simpleAppHelper.launchViaIntent(wmHelper)

        testApp.enterSplitScreenFromAppHandleMenu(wmHelper, device)

        // Open allApps via keyboard shortcut

        keyEventHelper.press(KEYCODE_META_RIGHT)

        tapl.allApps

            .getAppIcon(calculatorApp.appName)

            .launch(calculatorApp.packageName)

    }

    @After

    fun teardown() {

        testApp.exit(wmHelper)

        calculatorApp.exit(wmHelper)

    }

}