Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
/* Any copyright is dedicated to the Public Domain.
"use strict";
const RELATIVE_DIR = "toolkit/components/pdfjs/test/";
var MockFilePicker = SpecialPowers.MockFilePicker;
var tempDir;
async function promiseDownloadFinished(list) {
return new Promise(resolve => {
list.addView({
onDownloadChanged(download) {
download.launchWhenSucceeded = false;
if (download.succeeded || download.error) {
list.removeView(this);
resolve(download);
}
},
});
});
}
function createPromiseForFilePicker() {
return new Promise(resolve => {
MockFilePicker.showCallback = fp => {
let destFile = tempDir.clone();
destFile.append(fp.defaultString);
if (destFile.exists()) {
destFile.remove(false);
}
MockFilePicker.setFiles([destFile]);
MockFilePicker.filterIndex = 0; // kSaveAsType_Complete
resolve();
};
});
}
add_setup(async function () {
tempDir = createTemporarySaveDirectory();
MockFilePicker.init(window.browsingContext);
MockFilePicker.returnValue = MockFilePicker.returnOK;
MockFilePicker.displayDirectory = tempDir;
registerCleanupFunction(async function () {
MockFilePicker.cleanup();
await cleanupDownloads();
tempDir.remove(true);
});
});
/**
* Check clicking the download button saves the file and doesn't open a new viewer
*/
add_task(async function test_downloading_pdf_nonprivate_window() {
const pdfUrl = TESTROOT + "file_pdfjs_test.pdf";
await SpecialPowers.pushPrefEnv({
set: [["browser.download.always_ask_before_handling_new_types", false]],
});
await BrowserTestUtils.withNewTab(
{ gBrowser, url: "about:blank" },
async function (browser) {
await waitForPdfJS(browser, pdfUrl);
const tabCount = gBrowser.tabs.length;
info(`${tabCount} tabs are open at the start of the test`);
let downloadList = await Downloads.getList(Downloads.PUBLIC);
let filePickerShown = createPromiseForFilePicker();
let downloadFinishedPromise = promiseDownloadFinished(downloadList);
info("Clicking on the download button...");
await SpecialPowers.spawn(browser, [], () => {
content.document.querySelector("#downloadButton").click();
});
info("Waiting for a filename to be picked from the file picker");
await filePickerShown;
await downloadFinishedPromise;
ok(true, "A download was added when we clicked download");
ok(
!DownloadsPanel.isPanelShowing,
"The download panel did not open, since the file picker was shown already"
);
const allDownloads = await downloadList.getAll();
const dl = allDownloads.find(dl => dl.source.originalUrl === pdfUrl);
ok(!!dl, "The pdf download has the correct url in source.originalUrl");
SpecialPowers.clipboardCopyString("");
DownloadsCommon.copyDownloadLink(dl);
const copiedUrl = SpecialPowers.getClipboardData("text/plain");
is(copiedUrl, pdfUrl, "The copied url must be the original one");
is(
gBrowser.tabs.length,
tabCount,
"No new tab was opened to view the downloaded PDF"
);
SpecialPowers.clipboardCopyString("");
const downloadsLoaded = BrowserTestUtils.waitForEvent(
browser,
"InitialDownloadsLoaded",
true
);
await waitForPdfJSClose(browser);
BrowserTestUtils.startLoadingURIString(browser, "about:downloads");
await downloadsLoaded;
info("Wait for the clipboard to contain the url of the pdf");
await SimpleTest.promiseClipboardChange(pdfUrl, () => {
goDoCommand("cmd_copy");
});
}
);
await SpecialPowers.popPrefEnv();
});