mirror of
https://github.com/mozilla/gecko-dev.git
synced 2025-11-11 05:39:41 +02:00
4a06c925ac
# ignore-this-changeset Differential Revision: https://phabricator.services.mozilla.com/D36052 --HG-- extra : source : b5be5b4f4b47c256e28a29f665dc754f6407ee7f
120 lines
3.7 KiB
JavaScript
120 lines
3.7 KiB
JavaScript
/* -*- indent-tabs-mode: nil; js-indent-level: 2 -*- */
|
|
/* vim: set ts=2 et sw=2 tw=80: */
|
|
/* Any copyright is dedicated to the Public Domain.
|
|
* http://creativecommons.org/publicdomain/zero/1.0/ */
|
|
|
|
/**
|
|
* Tests the PDF download saver, and tests using a window as a
|
|
* source for the copy download saver.
|
|
*/
|
|
|
|
"use strict";
|
|
|
|
/**
|
|
* Helper function to make sure a window reference exists on the download source.
|
|
*/
|
|
function test_download_windowRef(aTab, aDownload) {
|
|
ok(aDownload.source.windowRef, "Download source had a window reference");
|
|
ok(
|
|
aDownload.source.windowRef instanceof Ci.xpcIJSWeakReference,
|
|
"Download window reference is a weak ref"
|
|
);
|
|
is(
|
|
aDownload.source.windowRef.get(),
|
|
aTab.linkedBrowser.contentWindow,
|
|
"Download window exists during test"
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Helper function to check the state of a completed download.
|
|
*/
|
|
function test_download_state_complete(aTab, aDownload, aPrivate, aCanceled) {
|
|
ok(aDownload.source, "Download has a source");
|
|
is(
|
|
aDownload.source.url,
|
|
aTab.linkedBrowser.contentWindow.location,
|
|
"Download source has correct url"
|
|
);
|
|
is(
|
|
aDownload.source.isPrivate,
|
|
aPrivate,
|
|
"Download source has correct private state"
|
|
);
|
|
ok(aDownload.stopped, "Download is stopped");
|
|
is(aCanceled, aDownload.canceled, "Download has correct canceled state");
|
|
is(!aCanceled, aDownload.succeeded, "Download has correct succeeded state");
|
|
is(aDownload.error, null, "Download error is not defined");
|
|
}
|
|
|
|
async function test_createDownload_common(aPrivate, aType) {
|
|
let win = await BrowserTestUtils.openNewBrowserWindow({ private: aPrivate });
|
|
|
|
let tab = await BrowserTestUtils.openNewForegroundTab(
|
|
win.gBrowser,
|
|
getRootDirectory(gTestPath) + "testFile.html"
|
|
);
|
|
let download = await Downloads.createDownload({
|
|
source: tab.linkedBrowser.contentWindow,
|
|
target: { path: getTempFile(TEST_TARGET_FILE_NAME_PDF).path },
|
|
saver: { type: aType },
|
|
});
|
|
|
|
await test_download_windowRef(tab, download);
|
|
await download.start();
|
|
|
|
await test_download_state_complete(tab, download, aPrivate, false);
|
|
if (aType == "pdf") {
|
|
let signature = await OS.File.read(download.target.path, {
|
|
bytes: 4,
|
|
encoding: "us-ascii",
|
|
});
|
|
is(signature, "%PDF", "File exists and signature matches");
|
|
} else {
|
|
ok(await OS.File.exists(download.target.path), "File exists");
|
|
}
|
|
|
|
win.gBrowser.removeTab(tab);
|
|
win.close();
|
|
}
|
|
|
|
add_task(async function test_createDownload_pdf_private() {
|
|
await test_createDownload_common(true, "pdf");
|
|
});
|
|
add_task(async function test_createDownload_pdf_not_private() {
|
|
await test_createDownload_common(false, "pdf");
|
|
});
|
|
|
|
// Even for the copy saver, using a window should produce valid results
|
|
add_task(async function test_createDownload_copy_private() {
|
|
await test_createDownload_common(true, "copy");
|
|
});
|
|
add_task(async function test_createDownload_copy_not_private() {
|
|
await test_createDownload_common(false, "copy");
|
|
});
|
|
|
|
add_task(async function test_cancel_pdf_download() {
|
|
let tab = BrowserTestUtils.addTab(
|
|
gBrowser,
|
|
getRootDirectory(gTestPath) + "testFile.html"
|
|
);
|
|
await promiseBrowserLoaded(tab.linkedBrowser);
|
|
|
|
let download = await Downloads.createDownload({
|
|
source: tab.linkedBrowser.contentWindow,
|
|
target: { path: getTempFile(TEST_TARGET_FILE_NAME_PDF).path },
|
|
saver: "pdf",
|
|
});
|
|
|
|
await test_download_windowRef(tab, download);
|
|
download.start().catch(() => {});
|
|
|
|
// Immediately cancel the download to test that it is erased correctly.
|
|
await download.cancel();
|
|
await test_download_state_complete(tab, download, false, true);
|
|
|
|
let exists = await OS.File.exists(download.target.path);
|
|
ok(!exists, "Target file does not exist");
|
|
|
|
gBrowser.removeTab(tab);
|
|
});
|