Bug 1847881 - Fix UI test failure defaultSearchCodeGoogleUS (#3178)

Co-authored-by: oana.horvath <ohorvath@mozilla.com>
9 months ago · 37479c142a
parent 8f6981d318
commit 37479c142a
2 changed files with 14 additions and 2 deletions
--- a/app/src/androidTest/java/org/mozilla/fenix/ui/ComposeSearchTest.kt
+++ b/app/src/androidTest/java/org/mozilla/fenix/ui/ComposeSearchTest.kt
@ -557,7 +557,13 @@ class ComposeSearchTest {
        }.openThreeDotMenu {
        }.openHistory {
            // Full URL no longer visible in the nav bar, so we'll check the history record
-            verifyHistoryItemExists(shouldExist = true, Constants.searchEngineCodes["Google"]!!)
+            // A search group is sometimes created when searching with Google (probably redirects)
+            try {
+                verifyHistoryItemExists(shouldExist = true, Constants.searchEngineCodes["Google"]!!)
+            } catch (e: AssertionError) {
+                openSearchGroup(queryString)
+                verifyHistoryItemExists(shouldExist = true, Constants.searchEngineCodes["Google"]!!)
+            }
        }
    }

--- a/app/src/androidTest/java/org/mozilla/fenix/ui/SearchTest.kt
+++ b/app/src/androidTest/java/org/mozilla/fenix/ui/SearchTest.kt
@ -565,7 +565,13 @@ class SearchTest {
        }.openThreeDotMenu {
        }.openHistory {
            // Full URL no longer visible in the nav bar, so we'll check the history record
-            verifyHistoryItemExists(shouldExist = true, searchEngineCodes["Google"]!!)
+            // A search group is sometimes created when searching with Google (probably redirects)
+            try {
+                verifyHistoryItemExists(shouldExist = true, searchEngineCodes["Google"]!!)
+            } catch (e: AssertionError) {
+                openSearchGroup(queryString)
+                verifyHistoryItemExists(shouldExist = true, searchEngineCodes["Google"]!!)
+            }
        }
    }