diff --git a/app/src/androidTest/java/org/mozilla/fenix/screenshots/DefaultHomeScreenTest.kt b/app/src/androidTest/java/org/mozilla/fenix/screenshots/DefaultHomeScreenTest.kt index 0fa99cc351..d0f32270df 100644 --- a/app/src/androidTest/java/org/mozilla/fenix/screenshots/DefaultHomeScreenTest.kt +++ b/app/src/androidTest/java/org/mozilla/fenix/screenshots/DefaultHomeScreenTest.kt @@ -44,10 +44,14 @@ class DefaultHomeScreenTest : ScreenshotTest() { SystemClock.sleep(TestAssetHelper.waitingTimeShort) Screengrab.screenshot("HomeScreenRobot_home-screen") }.openThreeDotMenu { - }.openSettings { } + }.openSettings { + }.openPrivateBrowsingSubMenu { + clickPrivateModeScreenshotsSwitch() + } // To get private screenshot, // dismiss onboarding going to settings and back mDevice.pressBack() + mDevice.pressBack() homeScreen { togglePrivateBrowsingModeOnOff() Screengrab.screenshot("HomeScreenRobot_private-browsing-menu") diff --git a/app/src/androidTest/java/org/mozilla/fenix/ui/robots/SettingsSubMenuPrivateBrowsingRobot.kt b/app/src/androidTest/java/org/mozilla/fenix/ui/robots/SettingsSubMenuPrivateBrowsingRobot.kt index 5ce0944af7..c8ef29dd09 100644 --- a/app/src/androidTest/java/org/mozilla/fenix/ui/robots/SettingsSubMenuPrivateBrowsingRobot.kt +++ b/app/src/androidTest/java/org/mozilla/fenix/ui/robots/SettingsSubMenuPrivateBrowsingRobot.kt @@ -46,6 +46,8 @@ class SettingsSubMenuPrivateBrowsingRobot { fun verifyPrivateBrowsingShortcutIcon() = assertPrivateBrowsingShortcutIcon() + fun clickPrivateModeScreenshotsSwitch() = screenshotsInPrivateModeSwitch().click() + fun clickOpenLinksInPrivateTabSwitch() = openLinksInPrivateTabSwitch().click() fun addPrivateShortcutToHomescreen() { @@ -92,6 +94,9 @@ private fun assertNavigationToolBarHeader() { private fun openLinksInPrivateTabSwitch() = onView(withText("Open links in a private tab")) +private fun screenshotsInPrivateModeSwitch() = + onView(withText("Allow screenshots in private browsing")) + private fun addPrivateBrowsingShortcutButton() = onView(withText("Add private browsing shortcut")) private fun goBackButton() = onView(withContentDescription("Navigate up"))