fenix/app/src/androidTest/java/org/mozilla/fenix/ui/robots/SettingsSubMenuDeleteBrowsingDataRobot.kt

/* 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/. */

package org.mozilla.fenix.ui.robots

import androidx.test.espresso.Espresso.onView
import androidx.test.espresso.assertion.ViewAssertions.matches
import androidx.test.espresso.matcher.RootMatchers.isDialog
import androidx.test.espresso.matcher.ViewMatchers.Visibility
import androidx.test.espresso.matcher.ViewMatchers.hasSibling
import androidx.test.espresso.matcher.ViewMatchers.withChild
import androidx.test.espresso.matcher.ViewMatchers.withContentDescription
import androidx.test.espresso.matcher.ViewMatchers.withEffectiveVisibility
import androidx.test.espresso.matcher.ViewMatchers.withId
import androidx.test.espresso.matcher.ViewMatchers.withText
import androidx.test.uiautomator.By
import androidx.test.uiautomator.UiSelector
import androidx.test.uiautomator.Until
import org.hamcrest.CoreMatchers.allOf
import org.junit.Assert.assertTrue
import org.mozilla.fenix.R
import org.mozilla.fenix.helpers.TestAssetHelper.waitingTime
import org.mozilla.fenix.helpers.TestHelper.appName
import org.mozilla.fenix.helpers.TestHelper.mDevice
import org.mozilla.fenix.helpers.assertIsChecked
import org.mozilla.fenix.helpers.click

/**
 * Implementation of Robot Pattern for the settings Delete Browsing Data sub menu.
 */
class SettingsSubMenuDeleteBrowsingDataRobot {

    fun verifyNavigationToolBarHeader() = assertNavigationToolBarHeader()
    fun verifyDeleteBrowsingDataButton() = assertDeleteBrowsingDataButton()
    fun verifyMessageInDialogBox() = assertMessageInDialogBox()
    fun verifyDeleteButtonInDialogBox() = assertDeleteButtonInDialogBox()
    fun verifyCancelButtonInDialogBox() = assertCancelButtonInDialogBox()
    fun verifyAllOptionsAndCheckBoxes() = assertAllOptionsAndCheckBoxes()
    fun verifyAllCheckBoxesAreChecked() = assertAllCheckBoxesAreChecked()
    fun verifyOpenTabsCheckBox(status: Boolean) = assertOpenTabsCheckBox(status)
    fun verifyBrowsingHistoryDetails(status: Boolean) = assertBrowsingHistoryCheckBox(status)
    fun verifyCookiesCheckBox(status: Boolean) = assertCookiesCheckBox(status)
    fun verifyCachedFilesCheckBox(status: Boolean) = assertCachedFilesCheckBox(status)
    fun verifySitePermissionsCheckBox(status: Boolean) = assertSitePermissionsCheckBox(status)
    fun verifyDownloadsCheckBox(status: Boolean) = assertDownloadsCheckBox(status)
    fun verifyOpenTabsDetails(tabNumber: String) = assertOpenTabsDescription(tabNumber)
    fun verifyBrowsingHistoryDetails(addresses: String) = assertBrowsingHistoryDescription(addresses)

    fun verifyDialogElements() {
        verifyMessageInDialogBox()
        verifyDeleteButtonInDialogBox()
        verifyCancelButtonInDialogBox()
    }

    fun switchOpenTabsCheckBox() = clickOpenTabsCheckBox()
    fun switchBrowsingHistoryCheckBox() = clickBrowsingHistoryCheckBox()
    fun switchCookiesCheckBox() = clickCookiesCheckBox()
    fun switchCachedFilesCheckBox() = clickCachedFilesCheckBox()
    fun switchSitePermissionsCheckBox() = clickSitePermissionsCheckBox()
    fun switchDownloadsCheckBox() = clickDownloadsCheckBox()
    fun clickDeleteBrowsingDataButton() = deleteBrowsingDataButton().click()
    fun clickDialogCancelButton() = dialogCancelButton().click()

    fun selectOnlyOpenTabsCheckBox() {
        clickBrowsingHistoryCheckBox()
        assertBrowsingHistoryCheckBox(false)

        clickCookiesCheckBox()
        assertCookiesCheckBox(false)

        clickCachedFilesCheckBox()
        assertCachedFilesCheckBox(false)

        clickSitePermissionsCheckBox()
        assertSitePermissionsCheckBox(false)

        clickDownloadsCheckBox()
        assertDownloadsCheckBox(false)

        assertOpenTabsCheckBox(true)
    }

    fun selectOnlyBrowsingHistoryCheckBox() {
        clickOpenTabsCheckBox()
        assertOpenTabsCheckBox(false)

        clickCookiesCheckBox()
        assertCookiesCheckBox(false)

        clickCachedFilesCheckBox()
        assertCachedFilesCheckBox(false)

        clickSitePermissionsCheckBox()
        assertSitePermissionsCheckBox(false)

        clickDownloadsCheckBox()
        assertDownloadsCheckBox(false)

        assertBrowsingHistoryCheckBox(true)
    }

    fun selectOnlyCookiesCheckBox() {
        clickOpenTabsCheckBox()
        assertOpenTabsCheckBox(false)

        assertCookiesCheckBox(true)

        clickCachedFilesCheckBox()
        assertCachedFilesCheckBox(false)

        clickSitePermissionsCheckBox()
        assertSitePermissionsCheckBox(false)

        clickDownloadsCheckBox()
        assertDownloadsCheckBox(false)

        clickBrowsingHistoryCheckBox()
        assertBrowsingHistoryCheckBox(false)
    }

    fun selectOnlyCachedFilesCheckBox() {
        clickOpenTabsCheckBox()
        assertOpenTabsCheckBox(false)

        clickBrowsingHistoryCheckBox()
        assertBrowsingHistoryCheckBox(false)

        clickCookiesCheckBox()
        assertCookiesCheckBox(false)

        assertCachedFilesCheckBox(true)

        clickSitePermissionsCheckBox()
        assertSitePermissionsCheckBox(false)

        clickDownloadsCheckBox()
        assertDownloadsCheckBox(false)
    }

    fun clickCancelButtonInDialogBoxAndVerifyContentsInDialogBox() {
        mDevice.wait(
            Until.findObject(By.text("Delete browsing data")),
            waitingTime,
        )
        clickDeleteBrowsingDataButton()
        verifyDialogElements()
        cancelButton().click()
    }
    fun confirmDeletionAndAssertSnackbar() {
        dialogDeleteButton().click()
        assertDeleteBrowsingDataSnackbar()
    }

    fun verifyDeleteBrowsingDataSubMenuItems() {
        verifyDeleteBrowsingDataButton()
        clickCancelButtonInDialogBoxAndVerifyContentsInDialogBox()
        verifyAllOptionsAndCheckBoxes()
        verifyAllCheckBoxesAreChecked()
    }

    class Transition {
        fun goBack(interact: SettingsRobot.() -> Unit): SettingsRobot.Transition {
            goBackButton().click()

            SettingsRobot().interact()
            return SettingsRobot.Transition()
        }
    }
}

private fun goBackButton() =
    onView(allOf(withContentDescription("Navigate up")))

private fun navigationToolBarHeader() =
    onView(
        allOf(
            withId(R.id.navigationToolbar),
            withChild(withText(R.string.preferences_delete_browsing_data)),
        ),
    )

private fun deleteBrowsingDataButton() = onView(withId(R.id.delete_data))

private fun assertNavigationToolBarHeader() =
    navigationToolBarHeader().check((matches(withEffectiveVisibility(Visibility.VISIBLE))))

private fun assertDeleteBrowsingDataButton() =
    deleteBrowsingDataButton().check((matches(withEffectiveVisibility(Visibility.VISIBLE))))

private fun cancelButton() =
    mDevice.findObject(UiSelector().textStartsWith("CANCEL"))

private fun dialogDeleteButton() = onView(withText("Delete")).inRoot(isDialog())

private fun dialogCancelButton() = onView(withText("Cancel")).inRoot(isDialog())

private fun openTabsSubsection() = onView(withText(R.string.preferences_delete_browsing_data_tabs_title_2))

private fun openTabsDescription(tabNumber: String) = onView(withText("$tabNumber tabs"))

private fun openTabsCheckBox() = onView(allOf(withId(R.id.checkbox), hasSibling(withText("Open tabs"))))

private fun browsingHistorySubsection() =
    onView(withText(R.string.preferences_delete_browsing_data_browsing_data_title))

private fun browsingHistoryDescription(addresses: String) = mDevice.findObject(UiSelector().textContains("$addresses addresses"))

private fun browsingHistoryCheckBox() =
    onView(allOf(withId(R.id.checkbox), hasSibling(withText("Browsing history and site data"))))

private fun cookiesSubsection() =
    onView(withText(R.string.preferences_delete_browsing_data_cookies))

private fun cookiesDescription() = onView(withText(R.string.preferences_delete_browsing_data_cookies_subtitle))

private fun cookiesCheckBox() =
    onView(allOf(withId(R.id.checkbox), hasSibling(withText("Cookies"))))

private fun cachedFilesSubsection() =
    onView(withText(R.string.preferences_delete_browsing_data_cached_files))

private fun cachedFilesDescription() =
    onView(withText(R.string.preferences_delete_browsing_data_cached_files_subtitle))

private fun cachedFilesCheckBox() =
    onView(allOf(withId(R.id.checkbox), hasSibling(withText("Cached images and files"))))

private fun sitePermissionsSubsection() =
    onView(withText(R.string.preferences_delete_browsing_data_site_permissions))

private fun sitePermissionsCheckBox() =
    onView(allOf(withId(R.id.checkbox), hasSibling(withText("Site permissions"))))

private fun downloadsSubsection() =
    onView(withText(R.string.preferences_delete_browsing_data_downloads))

private fun downloadsCheckBox() =
    onView(allOf(withId(R.id.checkbox), hasSibling(withText("Downloads"))))

private fun dialogMessage() =
    onView(withText("$appName will delete the selected browsing data."))
        .inRoot(isDialog())

private fun assertMessageInDialogBox() =
    dialogMessage().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))

private fun assertDeleteButtonInDialogBox() =
    dialogDeleteButton().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))

private fun assertCancelButtonInDialogBox() =
    dialogCancelButton().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))

private fun assertAllOptionsAndCheckBoxes() {
    openTabsSubsection().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    openTabsDescription("0").check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    openTabsCheckBox().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    browsingHistorySubsection().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    assertTrue(browsingHistoryDescription("0").waitForExists(waitingTime))
    browsingHistoryCheckBox().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    cookiesSubsection().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    cookiesDescription().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    cookiesCheckBox().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    cachedFilesSubsection().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    cachedFilesDescription().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    cachedFilesCheckBox().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    sitePermissionsSubsection().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    sitePermissionsCheckBox().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    downloadsSubsection().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
    downloadsCheckBox().check(matches(withEffectiveVisibility(Visibility.VISIBLE)))
}

private fun assertAllCheckBoxesAreChecked() {
    openTabsCheckBox().assertIsChecked(true)
    browsingHistoryCheckBox().assertIsChecked(true)
    cookiesCheckBox().assertIsChecked(true)
    cachedFilesCheckBox().assertIsChecked(true)
    sitePermissionsCheckBox().assertIsChecked(true)
    downloadsCheckBox().assertIsChecked(true)
}

private fun assertOpenTabsDescription(tabNumber: String) =
    openTabsDescription(tabNumber).check(matches(withEffectiveVisibility(Visibility.VISIBLE)))

private fun assertBrowsingHistoryDescription(addresses: String) =
    assertTrue(browsingHistoryDescription(addresses).waitForExists(waitingTime))

private fun assertDeleteBrowsingDataSnackbar() {
    assertTrue(
        mDevice.findObject(
            UiSelector().text("Browsing data deleted"),
        ).waitUntilGone(waitingTime),
    )
}

private fun clickOpenTabsCheckBox() = openTabsCheckBox().click()
private fun assertOpenTabsCheckBox(status: Boolean) = openTabsCheckBox().assertIsChecked(status)
private fun clickBrowsingHistoryCheckBox() = browsingHistoryCheckBox().click()
private fun assertBrowsingHistoryCheckBox(status: Boolean) = browsingHistoryCheckBox().assertIsChecked(status)
private fun clickCookiesCheckBox() = cookiesCheckBox().click()
private fun assertCookiesCheckBox(status: Boolean) = cookiesCheckBox().assertIsChecked(status)
private fun clickCachedFilesCheckBox() = cachedFilesCheckBox().click()
private fun assertCachedFilesCheckBox(status: Boolean) = cachedFilesCheckBox().assertIsChecked(status)
private fun clickSitePermissionsCheckBox() = sitePermissionsCheckBox().click()
private fun assertSitePermissionsCheckBox(status: Boolean) = sitePermissionsCheckBox().assertIsChecked(status)
private fun clickDownloadsCheckBox() = downloadsCheckBox().click()
private fun assertDownloadsCheckBox(status: Boolean) = downloadsCheckBox().assertIsChecked(status)