Bug 1840581 - Add condition before switching to private browsing

11 months ago · 8fa7e645f0
parent 53cc9846d8
commit 8fa7e645f0
1 changed files with 12 additions and 5 deletions
--- a/app/src/androidTest/java/org/mozilla/fenix/ui/robots/HomeScreenRobot.kt
+++ b/app/src/androidTest/java/org/mozilla/fenix/ui/robots/HomeScreenRobot.kt
@ -625,11 +625,18 @@ class HomeScreenRobot {
        }

        fun togglePrivateBrowsingMode() {
-            mDevice.findObject(UiSelector().resourceId("$packageName:id/privateBrowsingButton"))
-                .waitForExists(
-                    waitingTime,
-                )
-            privateBrowsingButton.click()
+            if (
+                !itemWithResIdAndDescription(
+                    "$packageName:id/privateBrowsingButton",
+                    "Disable private browsing",
+                ).exists()
+            ) {
+                mDevice.findObject(UiSelector().resourceId("$packageName:id/privateBrowsingButton"))
+                    .waitForExists(
+                        waitingTime,
+                    )
+                privateBrowsingButton.click()
+            }
        }

        fun triggerPrivateBrowsingShortcutPrompt(interact: AddToHomeScreenRobot.() -> Unit): AddToHomeScreenRobot.Transition {