Merge "Add new test transition SplitScreenRotateOneLaunchedAppTest"

/*

 * 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.splitscreen

import android.view.Surface

import androidx.test.filters.RequiresDevice

import androidx.test.platform.app.InstrumentationRegistry

import com.android.server.wm.flicker.Flicker

import com.android.server.wm.flicker.FlickerTestRunner

import com.android.server.wm.flicker.FlickerTestRunnerFactory

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

import com.android.server.wm.flicker.startRotation

import com.android.server.wm.flicker.endRotation

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

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

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

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

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

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

import com.android.server.wm.flicker.repetitions

import org.junit.FixMethodOrder

import org.junit.runner.RunWith

import org.junit.runners.MethodSorters

import org.junit.runners.Parameterized

/**

 * Test open app to split screen.

 * To run this test: `atest WMShellFlickerTests:SplitScreenRotateOneLaunchedAppTest`

 */

@RequiresDevice

@RunWith(Parameterized::class)

@FixMethodOrder(MethodSorters.NAME_ASCENDING)

class SplitScreenRotateOneLaunchedAppTest(

    testName: String,

    flickerSpec: Flicker

) : FlickerTestRunner(testName, flickerSpec) {

    companion object {

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

        @JvmStatic

        fun getParams(): Collection<Array<Any>> {

            val instrumentation = InstrumentationRegistry.getInstrumentation()

            val testApp = StandardAppHelper(instrumentation,

                    "com.android.wm.shell.flicker.testapp", "SimpleApp")

            return FlickerTestRunnerFactory(instrumentation, repetitions = 3)

                    .buildTest { configuration ->

                        withTestName {

                            buildTestTag("splitScreenRotateOneApp", testApp, configuration)

                        }

                        repeat { configuration.repetitions }

                        setup {

                            test {

                                device.wakeUpAndGoToHomeScreen()

                                testApp.open()

                                device.launchSplitScreen()

                                device.waitForIdle()

                            }

                            eachRun {

                                this.setRotation(configuration.startRotation)

                            }

                        }

                        teardown {

                            eachRun {

                                setRotation(Surface.ROTATION_0)

                            }

                            test {

                                testApp.exit()

                                if (device.isInSplitScreen()) {

                                    device.exitSplitScreen()

                                }

                            }

                        }

                        transitions {

                            this.setRotation(configuration.endRotation)

                        }

                    }

        }

    }

}

 No newline at end of file