From 5675734d97ffde493ae1c7becb1f5b2564ab26fa Mon Sep 17 00:00:00 2001 From: Vadim Tryshev Date: Thu, 4 Jan 2024 14:26:07 -0800 Subject: [PATCH] Waiting for Split Select state after tapping split menu item Bug: 313926097 Flag: N/A Test: presubmit Change-Id: I54edb4e2b79c5dae0acedecae0f0b858a2ed0ddb --- .../tapl/com/android/launcher3/tapl/OverviewTaskMenu.java | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/tapl/com/android/launcher3/tapl/OverviewTaskMenu.java b/tests/tapl/com/android/launcher3/tapl/OverviewTaskMenu.java index c6fa31d260..902ad5b568 100644 --- a/tests/tapl/com/android/launcher3/tapl/OverviewTaskMenu.java +++ b/tests/tapl/com/android/launcher3/tapl/OverviewTaskMenu.java @@ -17,6 +17,7 @@ package com.android.launcher3.tapl; import static com.android.launcher3.testing.shared.TestProtocol.OVERVIEW_MODAL_TASK_STATE_ORDINAL; +import static com.android.launcher3.testing.shared.TestProtocol.OVERVIEW_SPLIT_SELECT_ORDINAL; import androidx.annotation.NonNull; import androidx.test.uiautomator.By; @@ -42,8 +43,11 @@ public class OverviewTaskMenu { try (LauncherInstrumentation.Closable e = mLauncher.eventsCheck(); LauncherInstrumentation.Closable c = mLauncher.addContextLayer( "tap split menu item")) { - mLauncher.clickLauncherObject( - mLauncher.findObjectInContainer(mMenu, By.textStartsWith("Split"))); + mLauncher.runToState(() -> mLauncher.clickLauncherObject( + mLauncher.findObjectInContainer(mMenu, By.textStartsWith("Split"))), + OVERVIEW_SPLIT_SELECT_ORDINAL, + "tapping split menu item" + ); try (LauncherInstrumentation.Closable c1 = mLauncher.addContextLayer( "tapped split menu item")) {