131 lines
4.7 KiB
Kotlin
131 lines
4.7 KiB
Kotlin
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
@file:Suppress("TooManyFunctions")
|
|
|
|
package org.mozilla.fenix.ui.robots
|
|
|
|
import androidx.test.espresso.Espresso.onView
|
|
import androidx.test.espresso.assertion.ViewAssertions
|
|
import androidx.test.espresso.matcher.ViewMatchers
|
|
import androidx.test.platform.app.InstrumentationRegistry
|
|
import androidx.test.uiautomator.By
|
|
import androidx.test.uiautomator.UiDevice
|
|
import androidx.test.uiautomator.Until
|
|
import org.mozilla.fenix.R
|
|
import org.mozilla.fenix.helpers.TestAssetHelper
|
|
import org.mozilla.fenix.helpers.click
|
|
import org.mozilla.fenix.helpers.ext.waitNotNull
|
|
import org.mozilla.fenix.helpers.isChecked
|
|
|
|
/**
|
|
* Implementation of Robot Pattern for Enhanced Tracking Protection UI.
|
|
*/
|
|
class EnhancedTrackingProtectionRobot {
|
|
|
|
val mDevice = UiDevice.getInstance(InstrumentationRegistry.getInstrumentation())!!
|
|
|
|
fun verifyEnhancedTrackingProtectionNotice() = assertEnhancedTrackingProtectionNotice()
|
|
|
|
fun verifyEnhancedTrackingProtectionShield() = assertEnhancedTrackingProtectionShield()
|
|
|
|
fun verifyEnhancedTrackingProtectionSheetStatus(status: String, state: Boolean) =
|
|
assertEnhancedTrackingProtectionSheetStatus(status, state)
|
|
|
|
fun verifyEnhancedTrackingProtectionDetailsStatus(status: String) =
|
|
assertEnhancedTrackingProtectionDetailsStatus(status)
|
|
|
|
class Transition {
|
|
fun openEnhancedTrackingProtectionSheet(interact: EnhancedTrackingProtectionRobot.() -> Unit): Transition {
|
|
openEnhancedTrackingProtectionSheet().click()
|
|
|
|
EnhancedTrackingProtectionRobot().interact()
|
|
return Transition()
|
|
}
|
|
|
|
fun closeNotificationPopup(interact: BrowserRobot.() -> Unit): Transition {
|
|
closeButton().click()
|
|
|
|
BrowserRobot().interact()
|
|
return Transition()
|
|
}
|
|
|
|
fun closeEnhancedTrackingProtectionSheet(interact: BrowserRobot.() -> Unit): Transition {
|
|
// Back out of the Enhanced Tracking Protection sheet
|
|
mDevice.pressBack()
|
|
|
|
BrowserRobot().interact()
|
|
return Transition()
|
|
}
|
|
|
|
fun disableEnhancedTrackingProtectionFromSheet(interact: EnhancedTrackingProtectionRobot.() -> Unit): Transition {
|
|
disableEnhancedTrackingProtection().click()
|
|
|
|
EnhancedTrackingProtectionRobot().interact()
|
|
return Transition()
|
|
}
|
|
|
|
fun openProtectionSettings(interact: SettingsSubMenuEnhancedTrackingProtectionRobot.() -> Unit): Transition {
|
|
openEnhancedTrackingProtectionSettings().click()
|
|
|
|
SettingsSubMenuEnhancedTrackingProtectionRobot().interact()
|
|
return Transition()
|
|
}
|
|
|
|
fun openDetails(interact: EnhancedTrackingProtectionRobot.() -> Unit): Transition {
|
|
openEnhancedTrackingProtectionDetails().click()
|
|
|
|
EnhancedTrackingProtectionRobot().interact()
|
|
return Transition()
|
|
}
|
|
}
|
|
}
|
|
|
|
fun enhancedTrackingProtection(interact: EnhancedTrackingProtectionRobot.() -> Unit): EnhancedTrackingProtectionRobot.Transition {
|
|
EnhancedTrackingProtectionRobot().interact()
|
|
return EnhancedTrackingProtectionRobot.Transition()
|
|
}
|
|
|
|
private fun assertEnhancedTrackingProtectionNotice() {
|
|
mDevice.waitNotNull(
|
|
Until.findObject(By.res("org.mozilla.fenix.debug:id/onboarding_message")),
|
|
TestAssetHelper.waitingTime
|
|
)
|
|
}
|
|
|
|
private fun assertEnhancedTrackingProtectionShield() {
|
|
mDevice.waitNotNull(
|
|
Until.findObjects(By.descContains("Tracking Protection has blocked trackers"))
|
|
)
|
|
}
|
|
|
|
private fun assertEnhancedTrackingProtectionSheetStatus(status: String, state: Boolean) {
|
|
mDevice.waitNotNull(Until.findObjects(By.textContains(status)))
|
|
onView(ViewMatchers.withResourceName("switch_widget")).check(
|
|
ViewAssertions.matches(
|
|
isChecked(
|
|
state
|
|
)
|
|
)
|
|
)
|
|
}
|
|
|
|
private fun assertEnhancedTrackingProtectionDetailsStatus(status: String) {
|
|
mDevice.waitNotNull(Until.findObjects(By.textContains(status)))
|
|
}
|
|
|
|
private fun closeButton() = onView(ViewMatchers.withId(R.id.close_onboarding))
|
|
|
|
private fun openEnhancedTrackingProtectionSheet() =
|
|
onView(ViewMatchers.withId(R.id.mozac_browser_toolbar_tracking_protection_indicator))
|
|
|
|
private fun disableEnhancedTrackingProtection() =
|
|
onView(ViewMatchers.withResourceName("switch_widget"))
|
|
|
|
private fun openEnhancedTrackingProtectionSettings() =
|
|
onView(ViewMatchers.withId(R.id.protection_settings))
|
|
|
|
private fun openEnhancedTrackingProtectionDetails() =
|
|
onView(ViewMatchers.withId(R.id.tracking_content))
|