1 files changed, 198 insertions, 0 deletions
diff --git a/toolkit/components/pdfjs/test/browser_pdfjs_octet_stream.js b/toolkit/components/pdfjs/test/browser_pdfjs_octet_stream.js
new file mode 100644
index 0000000000..f208402f0e
--- /dev/null
+++ b/toolkit/components/pdfjs/test/browser_pdfjs_octet_stream.js
@@ -0,0 +1,198 @@
+/* Any copyright is dedicated to the Public Domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+const TESTROOT = getRootDirectory(gTestPath).replace(
+  "chrome://mochitests/content/",
+  "http://mochi.test:8888/"
+);
+
+// Get a ref to the pdf we want to open.
+const PDF_URL = TESTROOT + "file_pdfjs_object_stream.pdf";
+
+var gMIMEService = Cc["@mozilla.org/mime;1"].getService(Ci.nsIMIMEService);
+
+/**
+ * Check that if we open a PDF with octet-stream mimetype, it can load
+ * PDF.js .
+ */
+add_task(async function test_octet_stream_opens_pdfjs() {
+  await SpecialPowers.pushPrefEnv({ set: [["pdfjs.handleOctetStream", true]] });
+  let handlerInfo = gMIMEService.getFromTypeAndExtension(
+    "application/pdf",
+    "pdf"
+  );
+
+  // Make sure pdf.js is the default handler.
+  is(
+    handlerInfo.alwaysAskBeforeHandling,
+    false,
+    "pdf handler defaults to always-ask is false"
+  );
+  is(
+    handlerInfo.preferredAction,
+    Ci.nsIHandlerInfo.handleInternally,
+    "pdf handler defaults to internal"
+  );
+
+  await BrowserTestUtils.withNewTab(
+    { gBrowser, url: "about:blank" },
+    async function(newTabBrowser) {
+      await waitForPdfJS(newTabBrowser, PDF_URL);
+      is(newTabBrowser.currentURI.spec, PDF_URL, "Should load pdfjs");
+    }
+  );
+});
+
+/**
+ * Check that when the improvements_to_download_panel pref is false and
+ * if the octet-stream thing is in a frame, we don't load it inside PDF.js
+ */
+add_task(async function test_octet_stream_in_frame_downloads() {
+  await SpecialPowers.pushPrefEnv({
+    set: [
+      ["pdfjs.handleOctetStream", true],
+      ["browser.download.improvements_to_download_panel", false],
+    ],
+  });
+
+  let handlerInfo = gMIMEService.getFromTypeAndExtension(
+    "application/pdf",
+    "pdf"
+  );
+
+  // Make sure pdf.js is the default handler.
+  is(
+    handlerInfo.alwaysAskBeforeHandling,
+    false,
+    "pdf handler defaults to always-ask is false"
+  );
+  is(
+    handlerInfo.preferredAction,
+    Ci.nsIHandlerInfo.handleInternally,
+    "pdf handler defaults to internal"
+  );
+
+  let dialogPromise = BrowserTestUtils.domWindowOpenedAndLoaded();
+  await BrowserTestUtils.withNewTab(
+    { gBrowser, url: `data:text/html,<iframe src='${PDF_URL}'>` },
+    async function(newTabBrowser) {
+      let dialogWin = await dialogPromise;
+      ok(dialogWin, "Should have a dialog asking what to do.");
+      dialogWin.close(); // This is going to cancel the dialog.
+    }
+  );
+});
+
+/**
+ * Check that when the improvements_to_download_panel pref is true and
+ * if the octet-stream thing is in a frame, we don't load it inside PDF.js
+ */
+add_task(
+  async function test_octet_stream_in_frame_downloads_improvements_to_download_panel() {
+    await SpecialPowers.pushPrefEnv({
+      set: [
+        ["pdfjs.handleOctetStream", true],
+        ["browser.download.improvements_to_download_panel", true],
+      ],
+    });
+
+    let handlerInfo = gMIMEService.getFromTypeAndExtension(
+      "application/pdf",
+      "pdf"
+    );
+
+    // Make sure pdf.js is the default handler.
+    is(
+      handlerInfo.alwaysAskBeforeHandling,
+      false,
+      "pdf handler defaults to always-ask is false"
+    );
+    is(
+      handlerInfo.preferredAction,
+      Ci.nsIHandlerInfo.handleInternally,
+      "pdf handler defaults to internal"
+    );
+
+    let downloadsPanelPromise = BrowserTestUtils.waitForEvent(
+      DownloadsPanel.panel,
+      "popupshown"
+    );
+
+    // Once downloaded, the PDF will be opened as a file:// URI in a new tab
+    let previewTabPromise = BrowserTestUtils.waitForNewTab(
+      gBrowser,
+      url => {
+        let uri = NetUtil.newURI(url);
+        return uri.scheme == "file" && uri.spec.endsWith(".pdf");
+      },
+      false, // dont wait for load
+      true // any tab, not just the next one
+    );
+
+    await BrowserTestUtils.withNewTab(
+      { gBrowser, url: `data:text/html,<iframe src='${PDF_URL}'>` },
+      async function(newTabBrowser) {
+        // wait until downloadsPanel opens before continuing with test
+        info("Waiting for download panel to open");
+        await downloadsPanelPromise;
+        is(
+          DownloadsPanel.panel.state,
+          "open",
+          "Check the download panel state is 'open'"
+        );
+        let downloadList = await Downloads.getList(Downloads.PUBLIC);
+        let [download] = downloadList._downloads;
+
+        // Verify the downloaded PDF opened in a new tab,
+        // with its download file URI
+        info("Waiting for preview tab");
+        let previewTab = await previewTabPromise;
+        ok(previewTab, "PDF opened in a new tab");
+
+        is(
+          DownloadsPanel.isPanelShowing,
+          true,
+          "DownloadsPanel should be open."
+        );
+        is(
+          downloadList._downloads.length,
+          1,
+          "File should be successfully downloaded."
+        );
+
+        await BrowserTestUtils.removeTab(previewTab);
+
+        info("cleaning up downloads");
+        try {
+          if (Services.appinfo.OS === "WINNT") {
+            // We need to make the file writable to delete it on Windows.
+            await IOUtils.setPermissions(download.target.path, 0o600);
+          }
+          await IOUtils.remove(download.target.path);
+        } catch (error) {
+          info(
+            "The file " + download.target.path + " is not removed, " + error
+          );
+        }
+        await downloadList.remove(download);
+        await download.finalize();
+
+        if (DownloadsPanel.panel.state !== "closed") {
+          let hiddenPromise = BrowserTestUtils.waitForEvent(
+            DownloadsPanel.panel,
+            "popuphidden"
+          );
+          DownloadsPanel.hidePanel();
+          await hiddenPromise;
+        }
+        is(
+          DownloadsPanel.panel.state,
+          "closed",
+          "Check that the download panel is closed"
+        );
+      }
+    );
+  }
+);