1 /*
2  * Copyright (C) 2023 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package com.android.wm.shell.flicker
18 
19 import android.app.Instrumentation
20 import android.tools.flicker.junit.FlickerBuilderProvider
21 import android.tools.flicker.legacy.FlickerBuilder
22 import android.tools.flicker.legacy.LegacyFlickerTest
23 import androidx.test.platform.app.InstrumentationRegistry
24 import com.android.launcher3.tapl.LauncherInstrumentation
25 
26 abstract class BaseBenchmarkTest
27 @JvmOverloads
28 constructor(
29     protected open val flicker: LegacyFlickerTest,
30     protected val instrumentation: Instrumentation = InstrumentationRegistry.getInstrumentation(),
31     protected val tapl: LauncherInstrumentation = LauncherInstrumentation()
32 ) {
33     /** Specification of the test transition to execute */
34     abstract val transition: FlickerBuilder.() -> Unit
35 
36     /**
37      * Entry point for the test runner. It will use this method to initialize and cache flicker
38      * executions
39      */
40     @FlickerBuilderProvider
buildFlickernull41     open fun buildFlicker(): FlickerBuilder {
42         return FlickerBuilder(instrumentation).apply {
43             setup { flicker.scenario.setIsTablet(tapl.isTablet) }
44             transition()
45         }
46     }
47 }
48