From 6a38fc03aa0d0355d702457fbb15441889724b9e Mon Sep 17 00:00:00 2001
From: AndiAJ <andiaj@users.noreply.github.com>
Date: Fri, 27 Oct 2023 16:21:52 +0300
Subject: [PATCH] Bug 1862383 - New high priority download UI tests

---
 .../java/org/mozilla/fenix/ui/DownloadTest.kt | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/app/src/androidTest/java/org/mozilla/fenix/ui/DownloadTest.kt b/app/src/androidTest/java/org/mozilla/fenix/ui/DownloadTest.kt
index 72a5c68a0..d14f481d1 100644
--- a/app/src/androidTest/java/org/mozilla/fenix/ui/DownloadTest.kt
+++ b/app/src/androidTest/java/org/mozilla/fenix/ui/DownloadTest.kt
@@ -471,4 +471,29 @@ class DownloadTest {
         }
         deleteDownloadedFileOnStorage(downloadFile)
     }
+
+    // TestRail link: https://testrail.stage.mozaws.net/index.php?/cases/view/244125
+    @Test
+    fun restartDownloadFromAppNotificationAfterConnectionIsInterruptedTest() {
+        downloadFile = "1GB.zip"
+
+        navigationToolbar {
+        }.enterURLAndEnterToBrowser(downloadTestPage.toUri()) {
+            waitForPageToLoad()
+        }.clickDownloadLink(downloadFile) {
+            verifyDownloadPrompt(downloadFile)
+            setNetworkEnabled(false)
+        }.clickDownload {
+            verifyDownloadFailedPrompt(downloadFile)
+            setNetworkEnabled(true)
+            clickTryAgainButton()
+        }
+        browserScreen {
+        }.openNotificationShade {
+            verifySystemNotificationExists("Firefox Fenix")
+            expandNotificationMessage()
+            clickDownloadNotificationControlButton("CANCEL")
+        }
+        deleteDownloadedFileOnStorage(downloadFile)
+    }
 }