1 /* 2 * Copyright (C) 2024 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 android.platform.systemui_tapl.ui 18 19 import android.platform.systemui_tapl.utils.DeviceUtils.sysuiResSelector 20 import android.platform.uiautomatorhelpers.DeviceHelpers.assertVisible 21 import android.platform.uiautomatorhelpers.DeviceHelpers.waitForObj 22 import com.google.common.truth.Truth.assertThat 23 import java.time.Duration 24 25 /** System UI test automation object representing a bubble overflow panel. */ 26 class BubbleOverflow internal constructor() { 27 init { 28 BUBBLE_OVERFLOW_CONTAINER.assertVisible(timeout = FIND_OBJECT_TIMEOUT) 29 } 30 31 /** Asserts that the panel is empty. */ verifyIsEmptynull32 fun verifyIsEmpty() { 33 BUBBLE_OVERFLOW_EMPTY_STATE_ID.assertVisible(timeout = FIND_OBJECT_TIMEOUT) 34 } 35 36 /** Asserts that the panel has at least one bubble. */ verifyHasBubblesnull37 fun verifyHasBubbles() { 38 val recycler = waitForObj(BUBBLE_OVERFLOW_RECYCLER_ID, timeout = FIND_OBJECT_TIMEOUT) 39 val overflowBubbleViews = recycler.findObjects(BUBBLE_OVERFLOW_VIEW) 40 assertThat(overflowBubbleViews.size).isAtLeast(1) 41 } 42 43 private companion object { 44 val FIND_OBJECT_TIMEOUT = Duration.ofSeconds(20) 45 val BUBBLE_OVERFLOW_RECYCLER_ID = sysuiResSelector("bubble_overflow_recycler") 46 val BUBBLE_OVERFLOW_EMPTY_STATE_ID = sysuiResSelector("bubble_overflow_empty_state") 47 val BUBBLE_OVERFLOW_VIEW = sysuiResSelector("bubble_overflow_view") 48 val BUBBLE_OVERFLOW_CONTAINER = sysuiResSelector("bubble_overflow_container") 49 } 50 } 51