commit 5675734d97ffde493ae1c7becb1f5b2564ab26fa
author Vadim Tryshev <vadimt@google.com> Thu Jan 04 14:26:07 2024 -0800
committer Vadim Tryshev <vadimt@google.com> Thu Jan 04 14:58:05 2024 -0800
tree ce721414e5a79ac0c425a7e63ab18b4c43cb4da5
parent f0b1ddc038c6a0c0ef1cc360393866efce837b77

Waiting for Split Select state after tapping split menu item

Bug: 313926097
Flag: N/A
Test: presubmit
Change-Id: I54edb4e2b79c5dae0acedecae0f0b858a2ed0ddb

tests/tapl/com/android/launcher3/tapl/OverviewTaskMenu.java

diff --git a/tests/tapl/com/android/launcher3/tapl/OverviewTaskMenu.java b/tests/tapl/com/android/launcher3/tapl/OverviewTaskMenu.java
index c6fa31d..902ad5b 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 @@
         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")) {