Bug 1844305 - Add snackbar verification in UI tests when removing tabs from a collection

app/src/androidTest/java/org/mozilla/fenix/ui/CollectionTest.kt

@@ -353,6 +353,31 @@ class CollectionTest {
         }
     }
 
+    @Test
+    fun undoTabRemovalFromCollectionTest() {
+        val webPage = getGenericAsset(mockWebServer, 1)
+
+        navigationToolbar {
+        }.enterURLAndEnterToBrowser(webPage.url) {
+        }.openTabDrawer {
+            createCollection(webPage.title, collectionName = collectionName)
+            closeTab()
+        }
+
+        homeScreen {
+            verifyCollectionIsDisplayed(collectionName)
+        }.expandCollection(collectionName) {
+            verifyTabSavedInCollection(webPage.title, true)
+            removeTabFromCollection(webPage.title)
+        }
+        homeScreen {
+            verifySnackBarText("Collection deleted")
+            clickSnackbarButton("UNDO")
+            verifyCollectionIsDisplayed(collectionName, true)
+            verifyCollectionIsDisplayed(collectionName, true)
+        }
+    }
+
     @Test
     fun swipeLeftToRemoveTabFromCollectionTest() {
         val testPage = getGenericAsset(mockWebServer, 1)
@@ -375,6 +400,8 @@ class CollectionTest {
             verifyTabSavedInCollection(testPage.title, false)
         }
         homeScreen {
+            verifySnackBarText("Collection deleted")
+            verifySnackBarText("UNDO")
             verifyCollectionIsDisplayed(collectionName, false)
         }
     }
@@ -401,6 +428,8 @@ class CollectionTest {
             verifyTabSavedInCollection(testPage.title, false)
         }
         homeScreen {
+            verifySnackBarText("Collection deleted")
+            verifySnackBarText("UNDO")
             verifyCollectionIsDisplayed(collectionName, false)
         }
     }

app/src/androidTest/java/org/mozilla/fenix/ui/ComposeCollectionTest.kt

@@ -357,6 +357,31 @@ class ComposeCollectionTest {
         }
     }
 
+    @Test
+    fun undoTabRemovalFromCollectionTest() {
+        val webPage = getGenericAsset(mockWebServer, 1)
+
+        navigationToolbar {
+        }.enterURLAndEnterToBrowser(webPage.url) {
+        }.openComposeTabDrawer(composeTestRule) {
+            createCollection(webPage.title, collectionName = collectionName)
+            closeTab()
+        }
+
+        homeScreen {
+            verifyCollectionIsDisplayed(collectionName)
+        }.expandCollection(collectionName) {
+            verifyTabSavedInCollection(webPage.title, true)
+            removeTabFromCollection(webPage.title)
+        }
+        homeScreen {
+            verifySnackBarText("Collection deleted")
+            clickSnackbarButton("UNDO")
+            verifyCollectionIsDisplayed(collectionName, true)
+            verifyCollectionIsDisplayed(collectionName, true)
+        }
+    }
+
     @Test
     fun swipeLeftToRemoveTabFromCollectionTest() {
         val testPage = getGenericAsset(mockWebServer, 1)
@@ -379,6 +404,8 @@ class ComposeCollectionTest {
             verifyTabSavedInCollection(testPage.title, false)
         }
         homeScreen {
+            verifySnackBarText("Collection deleted")
+            verifySnackBarText("UNDO")
             verifyCollectionIsDisplayed(collectionName, false)
         }
     }
@@ -405,6 +432,8 @@ class ComposeCollectionTest {
             verifyTabSavedInCollection(testPage.title, false)
         }
         homeScreen {
+            verifySnackBarText("Collection deleted")
+            verifySnackBarText("UNDO")
             verifyCollectionIsDisplayed(collectionName, false)
         }
     }