| /* |
| * 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.server.wm.parser |
| |
| import androidx.test.platform.app.InstrumentationRegistry |
| import com.android.server.wm.InitRule |
| import com.android.server.wm.traces.common.NullableDeviceStateDump |
| import com.android.server.wm.traces.parser.FLAG_STATE_DUMP_FLAG_LAYERS |
| import com.android.server.wm.traces.parser.FLAG_STATE_DUMP_FLAG_WM |
| import com.android.server.wm.traces.parser.WmStateDumpFlags |
| import com.android.server.wm.traces.parser.getCurrentState |
| import com.android.server.wm.traces.parser.getCurrentStateDumpNullable |
| import com.google.common.truth.Truth |
| import org.junit.ClassRule |
| import org.junit.FixMethodOrder |
| import org.junit.Test |
| import org.junit.runners.MethodSorters |
| |
| /** |
| * Contains [com.android.server.wm.traces.parser.Extensions] tests. |
| * |
| * To run this test: `atest FlickerLibTest:UiDeviceExtensionsTest` |
| */ |
| @FixMethodOrder(MethodSorters.NAME_ASCENDING) |
| class UiDeviceExtensionsTest { |
| private fun getCurrState( |
| @WmStateDumpFlags dumpFlags: Int = FLAG_STATE_DUMP_FLAG_WM.or(FLAG_STATE_DUMP_FLAG_LAYERS) |
| ): Pair<ByteArray, ByteArray> { |
| val instrumentation = InstrumentationRegistry.getInstrumentation() |
| return getCurrentState(instrumentation.uiAutomation, dumpFlags) |
| } |
| |
| private fun getCurrStateDump( |
| @WmStateDumpFlags dumpFlags: Int = FLAG_STATE_DUMP_FLAG_WM.or(FLAG_STATE_DUMP_FLAG_LAYERS) |
| ): NullableDeviceStateDump { |
| val instrumentation = InstrumentationRegistry.getInstrumentation() |
| return getCurrentStateDumpNullable( |
| instrumentation.uiAutomation, |
| dumpFlags, |
| clearCacheAfterParsing = false |
| ) |
| } |
| |
| @Test |
| fun canFetchCurrentDeviceState() { |
| val currState = this.getCurrState() |
| Truth.assertThat(currState.first).isNotEmpty() |
| Truth.assertThat(currState.second).isNotEmpty() |
| } |
| |
| @Test |
| fun canFetchCurrentDeviceStateOnlyWm() { |
| val currStateDump = this.getCurrState(FLAG_STATE_DUMP_FLAG_WM) |
| Truth.assertThat(currStateDump.first).isNotEmpty() |
| Truth.assertThat(currStateDump.second).isEmpty() |
| val currState = this.getCurrStateDump(FLAG_STATE_DUMP_FLAG_WM) |
| Truth.assertThat(currState.wmState).isNotNull() |
| Truth.assertThat(currState.layerState).isNull() |
| } |
| |
| @Test |
| fun canFetchCurrentDeviceStateOnlyLayers() { |
| val currStateDump = this.getCurrState(FLAG_STATE_DUMP_FLAG_LAYERS) |
| Truth.assertThat(currStateDump.first).isEmpty() |
| Truth.assertThat(currStateDump.second).isNotEmpty() |
| val currState = this.getCurrStateDump(FLAG_STATE_DUMP_FLAG_LAYERS) |
| Truth.assertThat(currState.wmState).isNull() |
| Truth.assertThat(currState.layerState).isNotNull() |
| } |
| |
| @Test |
| fun canParseCurrentDeviceState() { |
| val currState = this.getCurrStateDump() |
| Truth.assertThat(currState.wmState?.asTrace()?.entries ?: emptyArray()).asList().hasSize(1) |
| Truth.assertThat(currState.wmState?.asTrace()?.entries?.first()?.windowStates).isNotEmpty() |
| Truth.assertThat(currState.layerState?.asTrace()?.entries ?: emptyArray()) |
| .asList() |
| .hasSize(1) |
| Truth.assertThat(currState.layerState?.asTrace()?.entries?.first()?.flattenedLayers) |
| .isNotEmpty() |
| } |
| |
| companion object { |
| @ClassRule @JvmField val initRule = InitRule() |
| } |
| } |