Add base scenario test for enter desktop mode with app handle menu.

/*

 * Copyright (C) 2024 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.service.desktopmode.scenarios

import android.app.Instrumentation

import android.tools.traces.parsers.WindowManagerStateHelper

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.SimpleAppHelper

import com.android.window.flags.Flags

import org.junit.After

import org.junit.Assume

import org.junit.Before

import org.junit.Ignore

import org.junit.Test

/** Base test class for enter desktop with app handle menu CUJ. */

@Ignore("Base Test Class")

abstract class EnterDesktopWithAppHandleMenu {

    private val instrumentation: Instrumentation = InstrumentationRegistry.getInstrumentation()

    private val tapl = LauncherInstrumentation()

    private val wmHelper = WindowManagerStateHelper(instrumentation)

    private val device = UiDevice.getInstance(instrumentation)

    private val simpleAppHelper = SimpleAppHelper(instrumentation)

    private val testApp = DesktopModeAppHelper(simpleAppHelper)

    @Before

    fun setup() {

        Assume.assumeTrue(Flags.enableDesktopWindowingMode() && tapl.isTablet)

    }

    @Test

    open fun enterDesktopWithAppHandleMenu() {

        simpleAppHelper.launchViaIntent(wmHelper)

        testApp.enterDesktopModeFromAppHandleMenu(wmHelper, device)

    }

    @After

    fun teardown() {

        testApp.exit(wmHelper)

    }

}

package com.android.server.wm.flicker.helpers

import android.graphics.Rect

import android.platform.uiautomator_helpers.DeviceHelpers

import android.tools.device.apphelpers.IStandardAppHelper

import android.tools.helpers.SYSTEMUI_PACKAGE

import android.tools.traces.parsers.WindowManagerStateHelper

import androidx.test.uiautomator.UiDevice

import androidx.test.uiautomator.UiObject2

import androidx.test.uiautomator.Until

import java.time.Duration

/**

 * Wrapper class around App helper classes. This class adds functionality to the apps that the

        RIGHT_BOTTOM

    }

    private val TIMEOUT_MS = 3_000L

    private val CAPTION = "desktop_mode_caption"

    private val CAPTION_HANDLE = "caption_handle"

    private val MAXIMIZE_BUTTON = "maximize_window"

    private val MAXIMIZE_BUTTON_VIEW = "maximize_button_view"

    private val CLOSE_BUTTON = "close_window"

    private val caption: BySelector

        get() = By.res(SYSTEMUI_PACKAGE, CAPTION)

    /** Wait for an app moved to desktop to finish its transition. */

    private fun waitForAppToMoveToDesktop(wmHelper: WindowManagerStateHelper) {

        wmHelper

        )

            error("expected a freeform window with caption but window is not in freeform mode")

        val captions =

            device.wait(Until.findObjects(caption), TIMEOUT_MS)

            device.wait(Until.findObjects(caption), TIMEOUT.toMillis())

                ?: error("Unable to find view $caption\n")

        return captions.find {

        device.drag(startX, startY, endX, endY, 100)

    }

    fun enterDesktopModeFromAppHandleMenu(

        wmHelper: WindowManagerStateHelper,

        device: UiDevice) {

        val windowRect = wmHelper.getWindowRegion(innerHelper).bounds

        val startX = windowRect.centerX()

        // Click a little under the top to prevent opening the notification shade.

        val startY = 10

        // Click on the app handle coordinates.

        device.click(startX, startY)

        wmHelper.StateSyncBuilder().withAppTransitionIdle().waitForAndVerify()

        val pill = getAppHandlePillForWindow()

        val desktopModeButton =

            pill

                ?.children

                ?.find { it.resourceName.endsWith(DESKTOP_MODE_BUTTON) }

        desktopModeButton?.click()

        wmHelper.StateSyncBuilder().withAppTransitionIdle().waitForAndVerify()

    }

    private fun getAppHandlePillForWindow(): UiObject2? {

        val pillContainer: BySelector = By.res(SYSTEMUI_PACKAGE, PILL_CONTAINER)

        return DeviceHelpers.waitForObj(pillContainer, TIMEOUT)

    }

    /** Wait for transition to full screen to finish. */

    private fun waitForTransitionToFullscreen(wmHelper: WindowManagerStateHelper) {

        wmHelper

            .withAppTransitionIdle()

            .waitForAndVerify()

    }

    private companion object {

        val TIMEOUT = Duration.ofSeconds(3)

        val CAPTION = "desktop_mode_caption"

        val MAXIMIZE_BUTTON_VIEW = "maximize_button_view"

        val CLOSE_BUTTON = "close_window"

        val PILL_CONTAINER = "windowing_pill"

        val DESKTOP_MODE_BUTTON = "desktop_button"

        val caption: BySelector

            get() = By.res(SYSTEMUI_PACKAGE, CAPTION)

    }

}