libs/WindowManager/Shell/tests/flicker/src/com/android/wm/shell/flicker/legacysplitscreen/SplitScreenTestBase.kt

/*
 * Copyright (C) 2020 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.legacysplitscreen

import android.support.test.launcherhelper.LauncherStrategyFactory
import com.android.server.wm.flicker.dsl.FlickerBuilder
import com.android.server.wm.flicker.helpers.exitSplitScreen
import com.android.server.wm.flicker.helpers.isInSplitScreen
import com.android.server.wm.flicker.helpers.openQuickStepAndClearRecentAppsFromOverview
import com.android.server.wm.flicker.helpers.wakeUpAndGoToHomeScreen
import com.android.server.wm.flicker.navBarLayerIsAlwaysVisible
import com.android.server.wm.flicker.navBarWindowIsAlwaysVisible
import com.android.server.wm.flicker.statusBarLayerIsAlwaysVisible
import com.android.server.wm.flicker.statusBarWindowIsAlwaysVisible
import com.android.wm.shell.flicker.NonRotationTestBase
import com.android.wm.shell.flicker.helpers.SplitScreenHelper

abstract class SplitScreenTestBase(
    rotationName: String,
    rotation: Int
) : NonRotationTestBase(rotationName, rotation) {
    protected val splitScreenApp = SplitScreenHelper.getPrimary(instrumentation)
    protected val secondaryApp = SplitScreenHelper.getSecondary(instrumentation)
    protected val nonResizeableApp = SplitScreenHelper.getNonResizeable(instrumentation)
    protected val LAUNCHER_PACKAGE_NAME = LauncherStrategyFactory.getInstance(instrumentation)
            .launcherStrategy.supportedLauncherPackage
    protected val LIVE_WALLPAPER_PACKAGE_NAME =
            "com.breel.wallpapers18.soundviz.wallpaper.variations.SoundVizWallpaperV2"
    protected val LETTER_BOX_NAME = "Letterbox"
    protected val TOAST_NAME = "Toast"

    protected val transitionSetup: FlickerBuilder
        get() = FlickerBuilder(instrumentation).apply {
                setup {
                    eachRun {
                        uiDevice.wakeUpAndGoToHomeScreen()
                        uiDevice.openQuickStepAndClearRecentAppsFromOverview()
                    }
                }
                teardown {
                    eachRun {
                        if (uiDevice.isInSplitScreen()) {
                            uiDevice.exitSplitScreen()
                        }
                        splitScreenApp.exit()
                        nonResizeableApp.exit()
                    }
                }
                assertions {
                    layersTrace {
                        navBarLayerIsAlwaysVisible()
                        statusBarLayerIsAlwaysVisible()
                    }
                    windowManagerTrace {
                        navBarWindowIsAlwaysVisible()
                        statusBarWindowIsAlwaysVisible()
                    }
                }
            }
}