1 files changed, 168 insertions, 0 deletions
diff --git a/dom/base/test/browser_object_attachment.js b/dom/base/test/browser_object_attachment.js
new file mode 100644
index 0000000000..b4432862f0
--- /dev/null
+++ b/dom/base/test/browser_object_attachment.js
@@ -0,0 +1,168 @@
+ChromeUtils.defineESModuleGetters(this, {
+  Downloads: "resource://gre/modules/Downloads.sys.mjs",
+});
+
+const httpsTestRoot = getRootDirectory(gTestPath).replace(
+  "chrome://mochitests/content",
+  "https://example.com"
+);
+
+add_task(async function test_pdf_object_attachment() {
+  await SpecialPowers.pushPrefEnv({
+    set: [["dom.navigation.object_embed.allow_retargeting", false]],
+  });
+
+  await BrowserTestUtils.withNewTab(
+    `${httpsTestRoot}/file_pdf_object_attachment.html`,
+    async browser => {
+      is(
+        browser.browsingContext.children.length,
+        1,
+        "Should have a child frame"
+      );
+      await SpecialPowers.spawn(browser, [], async () => {
+        let obj = content.document.querySelector("object");
+        is(
+          obj.displayedType,
+          Ci.nsIObjectLoadingContent.TYPE_DOCUMENT,
+          "should be displaying TYPE_DOCUMENT"
+        );
+      });
+    }
+  );
+});
+
+add_task(async function test_img_object_attachment() {
+  await SpecialPowers.pushPrefEnv({
+    set: [["dom.navigation.object_embed.allow_retargeting", false]],
+  });
+
+  await BrowserTestUtils.withNewTab(
+    `${httpsTestRoot}/file_img_object_attachment.html`,
+    async browser => {
+      is(
+        browser.browsingContext.children.length,
+        1,
+        "Should have a child frame"
+      );
+      await SpecialPowers.spawn(browser, [], async () => {
+        let obj = content.document.querySelector("object");
+        is(
+          obj.displayedType,
+          Ci.nsIObjectLoadingContent.TYPE_DOCUMENT,
+          "should be displaying TYPE_DOCUMENT"
+        );
+      });
+    }
+  );
+});
+
+async function waitForDownload() {
+  // Get the downloads list and add a view to listen for a download to be added.
+  let downloadList = await Downloads.getList(Downloads.ALL);
+
+  // Wait for a single download
+  let downloadView;
+  let finishedAllDownloads = new Promise(resolve => {
+    downloadView = {
+      onDownloadAdded(aDownload) {
+        info("download added");
+        resolve(aDownload);
+      },
+    };
+  });
+  await downloadList.addView(downloadView);
+  let download = await finishedAllDownloads;
+  await downloadList.removeView(downloadView);
+
+  // Clean up the download from the list.
+  await downloadList.remove(download);
+  await download.finalize(true);
+
+  // Return the download
+  return download;
+}
+
+add_task(async function test_pdf_object_attachment_download() {
+  await SpecialPowers.pushPrefEnv({
+    set: [["dom.navigation.object_embed.allow_retargeting", true]],
+  });
+
+  // Set the behaviour to save pdfs to disk and not handle internally, so we
+  // don't end up with extra tabs after the test.
+  var gMimeSvc = Cc["@mozilla.org/mime;1"].getService(Ci.nsIMIMEService);
+  var gHandlerSvc = Cc["@mozilla.org/uriloader/handler-service;1"].getService(
+    Ci.nsIHandlerService
+  );
+  const mimeInfo = gMimeSvc.getFromTypeAndExtension("application/pdf", "pdf");
+  let previousAction = mimeInfo.preferredAction;
+  mimeInfo.preferredAction = Ci.nsIHandlerInfo.saveToDisk;
+  gHandlerSvc.store(mimeInfo);
+  registerCleanupFunction(() => {
+    mimeInfo.preferredAction = previousAction;
+    gHandlerSvc.store(mimeInfo);
+  });
+
+  // Start listening for the download before opening the new tab.
+  let downloadPromise = waitForDownload();
+  await BrowserTestUtils.withNewTab(
+    `${httpsTestRoot}/file_pdf_object_attachment.html`,
+    async browser => {
+      let download = await downloadPromise;
+      is(
+        download.source.url,
+        `${httpsTestRoot}/file_pdf_attachment.pdf`,
+        "download should be the pdf"
+      );
+
+      await SpecialPowers.spawn(browser, [], async () => {
+        let obj = content.document.querySelector("object");
+        is(
+          obj.displayedType,
+          Ci.nsIObjectLoadingContent.TYPE_FALLBACK,
+          "should be displaying TYPE_FALLBACK"
+        );
+      });
+    }
+  );
+});
+
+add_task(async function test_img_object_attachment_download() {
+  // NOTE: This is testing our current behaviour here as of bug 1868001 (which
+  // is to download an image with `Content-Disposition: attachment` embedded
+  // within an object or embed element).
+  //
+  // Other browsers ignore the `Content-Disposition: attachment` header when
+  // loading images within object or embed element as-of december 2023, as
+  // we did prior to the changes in bug 1595491.
+  //
+  // If this turns out to be a web-compat issue, we may want to introduce
+  // special handling to ignore content-disposition when loading images within
+  // an object or embed element.
+  await SpecialPowers.pushPrefEnv({
+    set: [["dom.navigation.object_embed.allow_retargeting", true]],
+  });
+
+  // Start listening for the download before opening the new tab.
+  let downloadPromise = waitForDownload();
+  await BrowserTestUtils.withNewTab(
+    `${httpsTestRoot}/file_img_object_attachment.html`,
+    async browser => {
+      let download = await downloadPromise;
+      is(
+        download.source.url,
+        `${httpsTestRoot}/file_img_attachment.jpg`,
+        "download should be the jpg"
+      );
+
+      await SpecialPowers.spawn(browser, [], async () => {
+        let obj = content.document.querySelector("object");
+        is(
+          obj.displayedType,
+          Ci.nsIObjectLoadingContent.TYPE_FALLBACK,
+          "should be displaying TYPE_FALLBACK"
+        );
+      });
+    }
+  );
+});