For #2130 - Fix fileDownloadTest

mozilla-mobile · Dec 5, 2022 · 511886f · 511886f
1 parent 3ef62ea
commit 511886f
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/app/src/androidTest/java/org/mozilla/reference/browser/ui/robots/NotificationRobot.kt b/app/src/androidTest/java/org/mozilla/reference/browser/ui/robots/NotificationRobot.kt
@@ -29,7 +29,7 @@ class NotificationRobot {
         val downloadFilename = mDevice.findObject(UiSelector().text(fileName))
 
         while (!notificationFound) {
-            notificationTray.swipeUp(2)
+            scrollToEnd()
             notificationFound = mDevice.findObject(notification).waitForExists(waitingTime)
         }
         assertTrue(notificationFound)
@@ -95,3 +95,11 @@ private val systemMediaNotification =
             .className("android.view.ViewGroup")
             .packageName("com.android.systemui")
     )
+
+private fun notificationTray() = UiScrollable(
+    UiSelector().resourceId("com.android.systemui:id/notification_stack_scroller")
+).setAsVerticalList()
+
+private fun scrollToEnd() {
+    notificationTray().scrollToEnd(1)
+}