From a16e6ce1026a95413bd2e12352ba9c7e6181ca0a Mon Sep 17 00:00:00 2001
From: AndiAJ <andiaj@users.noreply.github.com>
Date: Thu, 27 Oct 2022 16:50:08 +0300
Subject: [PATCH] For #27522 new delete all search engines UI test

---
 .../mozilla/fenix/ui/SettingsSearchTest.kt    | 27 +++++++++++++++++++
 .../ui/robots/SettingsSubMenuSearchRobot.kt   | 18 +++++++++++++
 2 files changed, 45 insertions(+)

diff --git a/app/src/androidTest/java/org/mozilla/fenix/ui/SettingsSearchTest.kt b/app/src/androidTest/java/org/mozilla/fenix/ui/SettingsSearchTest.kt
index 4734a16c5..7ecec9759 100644
--- a/app/src/androidTest/java/org/mozilla/fenix/ui/SettingsSearchTest.kt
+++ b/app/src/androidTest/java/org/mozilla/fenix/ui/SettingsSearchTest.kt
@@ -343,4 +343,31 @@ class SettingsSearchTest {
             verifyDefaultSearchEngine("Bing")
         }
     }
+
+    // Expected for en-us defaults
+    @Test
+    fun deleteAllSearchEnginesTest() {
+        homeScreen {
+        }.openThreeDotMenu {
+        }.openSettings {
+        }.openSearchSubMenu {
+            deleteMultipleSearchEngines(
+                "Google",
+                "Bing",
+                "Amazon.com",
+                "DuckDuckGo",
+                "eBay",
+            )
+            verifyDefaultSearchEngine("Wikipedia")
+            verifyThreeDotButtonIsNotDisplayed("Wikipedia")
+            openAddSearchEngineMenu()
+            verifyAddSearchEngineListContains(
+                "Google",
+                "Bing",
+                "Amazon.com",
+                "DuckDuckGo",
+                "eBay",
+            )
+        }
+    }
 }
diff --git a/app/src/androidTest/java/org/mozilla/fenix/ui/robots/SettingsSubMenuSearchRobot.kt b/app/src/androidTest/java/org/mozilla/fenix/ui/robots/SettingsSubMenuSearchRobot.kt
index acd649e27..cee8dcd7f 100644
--- a/app/src/androidTest/java/org/mozilla/fenix/ui/robots/SettingsSubMenuSearchRobot.kt
+++ b/app/src/androidTest/java/org/mozilla/fenix/ui/robots/SettingsSubMenuSearchRobot.kt
@@ -176,6 +176,14 @@ class SettingsSubMenuSearchRobot {
 
     fun verifyDefaultSearchEngine(searchEngineName: String) = assertDefaultSearchEngine(searchEngineName)
 
+    fun verifyThreeDotButtonIsNotDisplayed(searchEngineName: String) = assertThreeDotButtonIsNotDisplayed(searchEngineName)
+
+    fun verifyAddSearchEngineListContains(vararg searchEngines: String) {
+        for (searchEngine in searchEngines) {
+            assertEngineListContains(searchEngine)
+        }
+    }
+
     fun saveNewSearchEngine() {
         addSearchEngineSaveButton().click()
         assertTrue(
@@ -252,6 +260,13 @@ class SettingsSubMenuSearchRobot {
         threeDotMenu(searchEngineName).click()
     }
 
+    fun deleteMultipleSearchEngines(vararg searchEngines: String) {
+        for (searchEngine in searchEngines) {
+            openEngineOverflowMenu(searchEngine)
+            clickDeleteSearchEngine()
+        }
+    }
+
     fun clickEdit() = onView(withText("Edit")).click()
 
     fun clickDeleteSearchEngine() =
@@ -367,6 +382,9 @@ private fun assertEngineListDoesNotContain(searchEngineName: String) {
     onView(withId(R.id.search_engine_group)).check(matches(not(hasDescendant(withText(searchEngineName)))))
 }
 
+private fun assertThreeDotButtonIsNotDisplayed(searchEngineName: String) =
+    threeDotMenu(searchEngineName).check(matches(not(isDisplayed())))
+
 private fun threeDotMenu(searchEngineName: String) =
     onView(
         allOf(