From b6f9bbf560d5d295b333cc7f9a16a3cc4b9c0d49 Mon Sep 17 00:00:00 2001
From: Jonathan Almeida <jalmeida@mozilla.com>
Date: Fri, 16 Apr 2021 01:03:22 -0400
Subject: [PATCH] Attempt at fixing openAllTabsInCollectionTest test

---
 .../org/mozilla/fenix/ui/robots/TabDrawerRobot.kt   | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/app/src/androidTest/java/org/mozilla/fenix/ui/robots/TabDrawerRobot.kt b/app/src/androidTest/java/org/mozilla/fenix/ui/robots/TabDrawerRobot.kt
index 86069d50e..f129845f8 100644
--- a/app/src/androidTest/java/org/mozilla/fenix/ui/robots/TabDrawerRobot.kt
+++ b/app/src/androidTest/java/org/mozilla/fenix/ui/robots/TabDrawerRobot.kt
@@ -162,7 +162,14 @@ class TabDrawerRobot {
 
     fun clickSelectTabs() {
         threeDotMenu().click()
-        onView(withText("Select tabs")).click()
+
+        mDevice.waitNotNull(
+            Until.findObject(text("Select tabs")),
+            waitingTime
+        )
+
+        val selectTabsButton = mDevice.findObject(text("Select tabs"))
+        selectTabsButton.click()
     }
 
     fun clickAddNewCollection() = addNewCollectionButton().click()
@@ -201,13 +208,13 @@ class TabDrawerRobot {
         }
 
         fun openTabDrawer(interact: TabDrawerRobot.() -> Unit): TabDrawerRobot.Transition {
-            mDevice.findObject(UiSelector().resourceId("org.mozilla.fenix.debug:id/tab_button"))
+            mDevice.findObject(UiSelector().resourceId("$packageName:id/tab_button"))
                 .waitForExists(waitingTime)
 
             tabsCounter().click()
 
             org.mozilla.fenix.ui.robots.mDevice.waitNotNull(
-                Until.findObject(By.res("org.mozilla.fenix.debug:id/tab_layout")),
+                Until.findObject(By.res("$packageName:id/tab_layout")),
                 waitingTime
             )