From d197b20357c711303c440fee815e92957ad2d5f2 Mon Sep 17 00:00:00 2001
From: Gijs Kruitbosch <gijskruitbosch@gmail.com>
Date: Tue, 19 Oct 2021 10:35:46 +0000
Subject: [PATCH] Bug 1735368 - fix browser_auto_close_window.js so it passes
 with download improvements enabled and when asking for all downloads,
 r=sfoster

Differential Revision: https://phabricator.services.mozilla.com/D128628
---
 .../mochitest/browser_auto_close_window.js      | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/uriloader/exthandler/tests/mochitest/browser_auto_close_window.js b/uriloader/exthandler/tests/mochitest/browser_auto_close_window.js
index 2e1d17e13992..f1059dac956d 100644
--- a/uriloader/exthandler/tests/mochitest/browser_auto_close_window.js
+++ b/uriloader/exthandler/tests/mochitest/browser_auto_close_window.js
@@ -53,6 +53,23 @@ add_task(async function setup() {
     HELPERAPP_DIALOG_CONTRACT_ID,
     mockHelperAppService
   );
+
+  // Ensure we always prompt for these downloads.
+  const HandlerService = Cc[
+    "@mozilla.org/uriloader/handler-service;1"
+  ].getService(Ci.nsIHandlerService);
+
+  const MIMEService = Cc["@mozilla.org/mime;1"].getService(Ci.nsIMIMEService);
+  const mimeInfo = MIMEService.getFromTypeAndExtension(
+    "application/octet-stream",
+    "bin"
+  );
+  mimeInfo.alwaysAskBeforeHandling = true;
+  HandlerService.store(mimeInfo);
+
+  registerCleanupFunction(() => {
+    HandlerService.remove(mimeInfo);
+  });
 });
 
 add_task(async function simple_navigation() {