From c9d3af54c68e0be8698a80c386b731b2b1945f68 Mon Sep 17 00:00:00 2001
From: AndiAJ <andiaj@users.noreply.github.com>
Date: Tue, 12 Jul 2022 11:29:04 +0300
Subject: [PATCH] For #25992 #25832 and #25710 fix flaky PWA UI tests

---
 app/src/androidTest/java/org/mozilla/fenix/ui/PwaTest.kt | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/app/src/androidTest/java/org/mozilla/fenix/ui/PwaTest.kt b/app/src/androidTest/java/org/mozilla/fenix/ui/PwaTest.kt
index 9460e7a08..5dd974ac3 100644
--- a/app/src/androidTest/java/org/mozilla/fenix/ui/PwaTest.kt
+++ b/app/src/androidTest/java/org/mozilla/fenix/ui/PwaTest.kt
@@ -44,6 +44,8 @@ class PwaTest {
 
         navigationToolbar {
         }.enterURLAndEnterToBrowser(externalLinksPWAPage.toUri()) {
+            waitForPageToLoad()
+            verifyNotificationDotOnMainMenu()
         }.openThreeDotMenu {
         }.clickInstall {
             clickAddAutomaticallyButton()
@@ -63,6 +65,7 @@ class PwaTest {
         navigationToolbar {
         }.enterURLAndEnterToBrowser(externalLinksPWAPage.toUri()) {
             waitForPageToLoad()
+            verifyNotificationDotOnMainMenu()
         }.openThreeDotMenu {
         }.clickInstall {
             clickAddAutomaticallyButton()
@@ -79,6 +82,7 @@ class PwaTest {
         navigationToolbar {
         }.enterURLAndEnterToBrowser(externalLinksPWAPage.toUri()) {
             waitForPageToLoad()
+            verifyNotificationDotOnMainMenu()
         }.openThreeDotMenu {
         }.clickInstall {
             clickAddAutomaticallyButton()