Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
/* Any copyright is dedicated to the Public Domain.
"use strict";
// Need to start the server before `httpUrl` works.
startServer();
const DOWNLOAD_URL = httpUrl("interruptible.txt");
let gDownloadDir;
let gExternalHelperAppService = Cc[
"@mozilla.org/uriloader/external-helper-app-service;1"
].getService(Ci.nsIExternalHelperAppService);
gExternalHelperAppService.QueryInterface(Ci.nsIObserver);
add_setup(async function () {
await SpecialPowers.pushPrefEnv({
set: [
["browser.download.start_downloads_in_tmp_dir", true],
["browser.helperApps.deleteTempFileOnExit", true],
],
});
registerCleanupFunction(task_resetState);
gDownloadDir = new FileUtils.File(await setDownloadDir());
registerCleanupFunction(() => {
Services.prefs.clearUserPref("browser.download.dir");
Services.prefs.clearUserPref("browser.download.folderList");
});
});
add_task(async function test_download_asking_starts_in_tmp() {
await SpecialPowers.pushPrefEnv({
set: [["browser.download.always_ask_before_handling_new_types", true]],
});
let list = await Downloads.getList(Downloads.PUBLIC);
let downloadStarted = new Promise(resolve => {
let view = {
onDownloadAdded(download) {
list.removeView(view);
resolve(download);
},
};
list.addView(view);
});
// Wait for the download prompting dialog
let dialogPromise = BrowserTestUtils.domWindowOpenedAndLoaded(
null,
win => win.document.documentURI == UCT_URI
);
serveInterruptibleAsDownload();
mustInterruptResponses();
await BrowserTestUtils.withNewTab(
{
gBrowser,
url: DOWNLOAD_URL,
waitForLoad: false,
waitForStop: true,
},
async function () {
let dialogWin = await dialogPromise;
let tempFile = dialogWin.dialog.mLauncher.targetFile;
ok(
!tempFile.parent.equals(gDownloadDir),
"Should not have put temp file in the downloads dir."
);
let dialogEl = dialogWin.document.querySelector("dialog");
dialogEl.getButton("accept").disabled = false;
dialogEl.acceptDialog();
let download = await downloadStarted;
is(
PathUtils.parent(download.target.path),
gDownloadDir.path,
"Should have put final file in the downloads dir."
);
continueResponses();
await download.whenSucceeded();
await IOUtils.remove(download.target.path);
}
);
await list.removeFinished();
});
add_task(async function test_download_asking_and_opening_opens_from_tmp() {
await SpecialPowers.pushPrefEnv({
set: [["browser.download.always_ask_before_handling_new_types", true]],
});
let list = await Downloads.getList(Downloads.PUBLIC);
let downloadStarted = new Promise(resolve => {
let view = {
onDownloadAdded(download) {
list.removeView(view);
resolve(download);
},
};
list.addView(view);
});
// Wait for the download prompting dialog
let dialogPromise = BrowserTestUtils.domWindowOpenedAndLoaded(
null,
win => win.document.documentURI == UCT_URI
);
let oldLaunchFile = DownloadIntegration.launchFile;
let promiseLaunchFileCalled = new Promise(resolve => {
DownloadIntegration.launchFile = file => {
ok(true, "The file should be launched with an external application");
resolve(file);
DownloadIntegration.launchFile = oldLaunchFile;
};
});
registerCleanupFunction(() => {
DownloadIntegration.launchFile = oldLaunchFile;
});
serveInterruptibleAsDownload();
mustInterruptResponses();
await BrowserTestUtils.withNewTab(
{
gBrowser,
url: DOWNLOAD_URL,
waitForLoad: false,
waitForStop: true,
},
async function () {
let dialogWin = await dialogPromise;
let tempFile = dialogWin.dialog.mLauncher.targetFile;
ok(
!tempFile.parent.equals(gDownloadDir),
"Should not have put temp file in the downloads dir."
);
dialogWin.document.getElementById("open").click();
let dialogEl = dialogWin.document.querySelector("dialog");
dialogEl.getButton("accept").disabled = false;
dialogEl.acceptDialog();
let download = await downloadStarted;
isnot(
PathUtils.parent(download.target.path),
gDownloadDir.path,
"Should not have put final file in the downloads dir when it's supposed to be automatically opened."
);
continueResponses();
await download.whenSucceeded();
await download.refresh();
isnot(
PathUtils.parent(download.target.path),
gDownloadDir.path,
"Once finished the download should not be in the downloads dir when it's supposed to be automatically opened."
);
let file = await promiseLaunchFileCalled;
ok(
!file.parent.equals(gDownloadDir),
"Should not have put opened file in the downloads dir."
);
// Pretend that we've quit so we wipe all the files:
gExternalHelperAppService.observe(null, "profile-before-change", "");
// Now the file should go away, but that's async...
let f = new FileUtils.File(download.target.path);
await TestUtils.waitForCondition(
() => !f.exists(),
"Temp file should be removed",
500
).catch(err => ok(false, err));
ok(!f.exists(), "Temp file should be removed.");
await IOUtils.remove(download.target.path);
}
);
await list.removeFinished();
});
// Check that if we open the file automatically, it opens from the temp dir.
add_task(async function test_download_automatically_opened_from_tmp() {
await SpecialPowers.pushPrefEnv({
set: [["browser.download.always_ask_before_handling_new_types", false]],
});
serveInterruptibleAsDownload();
mustInterruptResponses();
let list = await Downloads.getList(Downloads.PUBLIC);
let downloadStarted = new Promise(resolve => {
let view = {
onDownloadAdded(download) {
list.removeView(view);
resolve(download);
},
};
list.addView(view);
});
const handlerSvc = Cc["@mozilla.org/uriloader/handler-service;1"].getService(
Ci.nsIHandlerService
);
const mimeSvc = Cc["@mozilla.org/mime;1"].getService(Ci.nsIMIMEService);
let txtHandlerInfo = mimeSvc.getFromTypeAndExtension("text/plain", "txt");
txtHandlerInfo.preferredAction = Ci.nsIHandlerInfo.useSystemDefault;
txtHandlerInfo.alwaysAskBeforeHandling = false;
handlerSvc.store(txtHandlerInfo);
registerCleanupFunction(() => handlerSvc.remove(txtHandlerInfo));
let oldLaunchFile = DownloadIntegration.launchFile;
let promiseLaunchFileCalled = new Promise(resolve => {
DownloadIntegration.launchFile = file => {
ok(true, "The file should be launched with an external application");
resolve(file);
DownloadIntegration.launchFile = oldLaunchFile;
};
});
registerCleanupFunction(() => {
DownloadIntegration.launchFile = oldLaunchFile;
});
await BrowserTestUtils.withNewTab(
{
gBrowser,
url: DOWNLOAD_URL,
waitForLoad: false,
waitForStop: true,
},
async function () {
let download = await downloadStarted;
isnot(
PathUtils.parent(download.target.partFilePath),
gDownloadDir.path,
"Should not start the download in the downloads dir."
);
continueResponses();
await download.whenSucceeded();
isnot(
PathUtils.parent(download.target.path),
gDownloadDir.path,
"Should not have put final file in the downloads dir."
);
let file = await promiseLaunchFileCalled;
ok(
!file.parent.equals(gDownloadDir),
"Should not have put opened file in the downloads dir."
);
// Pretend that we've quit so we wipe all the files:
gExternalHelperAppService.observe(null, "profile-before-change", "");
// Now the file should go away, but that's async...
let f = new FileUtils.File(download.target.path);
await TestUtils.waitForCondition(
() => !f.exists(),
"Temp file should be removed",
500
).catch(err => ok(false, err));
ok(!f.exists(), "Temp file should be removed.");
await IOUtils.remove(download.target.path);
}
);
handlerSvc.remove(txtHandlerInfo);
await list.removeFinished();
});