From 26a029d407be480d791972afb5975cf62c9360a6 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Fri, 19 Apr 2024 02:47:55 +0200
Subject: Adding upstream version 124.0.1.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 .../test/xpcshell/test_ext_downloads_eventpage.js  | 162 +++++++++++++++++++++
 1 file changed, 162 insertions(+)
 create mode 100644 toolkit/components/extensions/test/xpcshell/test_ext_downloads_eventpage.js

(limited to 'toolkit/components/extensions/test/xpcshell/test_ext_downloads_eventpage.js')

diff --git a/toolkit/components/extensions/test/xpcshell/test_ext_downloads_eventpage.js b/toolkit/components/extensions/test/xpcshell/test_ext_downloads_eventpage.js
new file mode 100644
index 0000000000..9c71c63e96
--- /dev/null
+++ b/toolkit/components/extensions/test/xpcshell/test_ext_downloads_eventpage.js
@@ -0,0 +1,162 @@
+"use strict";
+
+AddonTestUtils.init(this);
+AddonTestUtils.overrideCertDB();
+AddonTestUtils.createAppInfo(
+  "xpcshell@tests.mozilla.org",
+  "XPCShell",
+  "42",
+  "42"
+);
+
+const server = createHttpServer();
+server.registerDirectory("/data/", do_get_file("data"));
+
+const BASE = `http://localhost:${server.identity.primaryPort}/data`;
+const TXT_FILE = "file_download.txt";
+const TXT_URL = BASE + "/" + TXT_FILE;
+
+add_task(function setup() {
+  let downloadDir = FileUtils.getDir("TmpD", ["downloads"]);
+  downloadDir.createUnique(
+    Ci.nsIFile.DIRECTORY_TYPE,
+    FileUtils.PERMS_DIRECTORY
+  );
+  info(`Using download directory ${downloadDir.path}`);
+
+  Services.prefs.setIntPref("browser.download.folderList", 2);
+  Services.prefs.setComplexValue(
+    "browser.download.dir",
+    Ci.nsIFile,
+    downloadDir
+  );
+
+  registerCleanupFunction(() => {
+    Services.prefs.clearUserPref("browser.download.folderList");
+    Services.prefs.clearUserPref("browser.download.dir");
+
+    let entries = downloadDir.directoryEntries;
+    while (entries.hasMoreElements()) {
+      let entry = entries.nextFile;
+      ok(false, `Leftover file ${entry.path} in download directory`);
+      entry.remove(false);
+    }
+
+    downloadDir.remove(false);
+  });
+});
+
+add_task(
+  { pref_set: [["extensions.eventPages.enabled", true]] },
+  async function test_downloads_event_page() {
+    await AddonTestUtils.promiseStartupManager();
+
+    // A simple download driving extension
+    let dl_extension = ExtensionTestUtils.loadExtension({
+      manifest: {
+        browser_specific_settings: { gecko: { id: "downloader@mochitest" } },
+        permissions: ["downloads"],
+        background: { persistent: false },
+      },
+      background() {
+        let downloadId;
+        browser.downloads.onChanged.addListener(async info => {
+          if (info.state && info.state.current === "complete") {
+            browser.test.sendMessage("downloadComplete");
+          }
+        });
+        browser.test.onMessage.addListener(async (msg, opts) => {
+          if (msg == "download") {
+            downloadId = await browser.downloads.download(opts);
+          }
+          if (msg == "erase") {
+            await browser.downloads.removeFile(downloadId);
+            await browser.downloads.erase({ id: downloadId });
+          }
+        });
+      },
+    });
+    await dl_extension.startup();
+
+    let extension = ExtensionTestUtils.loadExtension({
+      useAddonManager: "permanent",
+      manifest: {
+        permissions: ["downloads"],
+        background: { persistent: false },
+      },
+      background() {
+        browser.downloads.onChanged.addListener(() => {
+          browser.test.sendMessage("onChanged");
+        });
+        browser.downloads.onCreated.addListener(() => {
+          browser.test.sendMessage("onCreated");
+        });
+        browser.downloads.onErased.addListener(() => {
+          browser.test.sendMessage("onErased");
+        });
+        browser.test.sendMessage("ready");
+      },
+    });
+
+    // onDeterminingFilename is never persisted, it is an empty event handler.
+    const EVENTS = ["onChanged", "onCreated", "onErased"];
+
+    await extension.startup();
+    await extension.awaitMessage("ready");
+    for (let event of EVENTS) {
+      assertPersistentListeners(extension, "downloads", event, {
+        primed: false,
+      });
+    }
+
+    await extension.terminateBackground({ disableResetIdleForTest: true });
+    ok(
+      !extension.extension.backgroundContext,
+      "Background Extension context should have been destroyed"
+    );
+
+    for (let event of EVENTS) {
+      assertPersistentListeners(extension, "downloads", event, {
+        primed: true,
+      });
+    }
+
+    // test download events waken background
+    dl_extension.sendMessage("download", {
+      url: TXT_URL,
+      filename: TXT_FILE,
+    });
+    await extension.awaitMessage("ready");
+    await extension.awaitMessage("onCreated");
+    for (let event of EVENTS) {
+      assertPersistentListeners(extension, "downloads", event, {
+        primed: false,
+      });
+    }
+    await extension.awaitMessage("onChanged");
+
+    await extension.terminateBackground({ disableResetIdleForTest: true });
+    ok(
+      !extension.extension.backgroundContext,
+      "Background Extension context should have been destroyed"
+    );
+
+    await dl_extension.awaitMessage("downloadComplete");
+    dl_extension.sendMessage("erase");
+    await extension.awaitMessage("ready");
+    await extension.awaitMessage("onErased");
+    await dl_extension.unload();
+
+    // check primed listeners after startup
+    await AddonTestUtils.promiseRestartManager();
+    await extension.awaitStartup();
+
+    for (let event of EVENTS) {
+      assertPersistentListeners(extension, "downloads", event, {
+        primed: true,
+      });
+    }
+
+    await extension.unload();
+  }
+);
-- 
cgit v1.2.3