mirror of
https://github.com/mozilla/gecko-dev.git
synced 2025-11-09 04:39:03 +02:00
e2caf16515
This ensures that we don't leave downloads behind in race conditions, and that that doesn't break future tests. Differential Revision: https://phabricator.services.mozilla.com/D101041
89 lines
2.3 KiB
JavaScript
89 lines
2.3 KiB
JavaScript
const { Services } = ChromeUtils.import("resource://gre/modules/Services.jsm");
|
|
|
|
ChromeUtils.defineModuleGetter(
|
|
this,
|
|
"Downloads",
|
|
"resource://gre/modules/Downloads.jsm"
|
|
);
|
|
ChromeUtils.defineModuleGetter(
|
|
this,
|
|
"DownloadsCommon",
|
|
"resource:///modules/DownloadsCommon.jsm"
|
|
);
|
|
ChromeUtils.defineModuleGetter(
|
|
this,
|
|
"FileUtils",
|
|
"resource://gre/modules/FileUtils.jsm"
|
|
);
|
|
ChromeUtils.defineModuleGetter(this, "OS", "resource://gre/modules/osfile.jsm");
|
|
ChromeUtils.defineModuleGetter(
|
|
this,
|
|
"FileTestUtils",
|
|
"resource://testing-common/FileTestUtils.jsm"
|
|
);
|
|
ChromeUtils.defineModuleGetter(
|
|
this,
|
|
"NetUtil",
|
|
"resource://gre/modules/NetUtil.jsm"
|
|
);
|
|
ChromeUtils.defineModuleGetter(
|
|
this,
|
|
"TestUtils",
|
|
"resource://testing-common/TestUtils.jsm"
|
|
);
|
|
|
|
async function createDownloadedFile(pathname, contents) {
|
|
info("createDownloadedFile: " + pathname);
|
|
let encoder = new TextEncoder();
|
|
let file = new FileUtils.File(pathname);
|
|
if (file.exists()) {
|
|
info(`File at ${pathname} already exists`);
|
|
if (!contents) {
|
|
ok(
|
|
false,
|
|
`A file already exists at ${pathname}, but createDownloadedFile was asked to create a non-existant file`
|
|
);
|
|
}
|
|
}
|
|
if (contents) {
|
|
await OS.File.writeAtomic(pathname, encoder.encode(contents));
|
|
ok(file.exists(), `Created ${pathname}`);
|
|
}
|
|
// No post-test cleanup necessary; tmp downloads directory is already removed after each test
|
|
return file;
|
|
}
|
|
|
|
let gDownloadDir;
|
|
|
|
async function setDownloadDir() {
|
|
let tmpDir = Services.dirsvc.get("TmpD", Ci.nsIFile).path;
|
|
tmpDir = PathUtils.join(
|
|
tmpDir,
|
|
"testsavedir" + Math.floor(Math.random() * 2 ** 32)
|
|
);
|
|
// Create this dir if it doesn't exist (ignores existing dirs)
|
|
await IOUtils.makeDirectory(tmpDir);
|
|
registerCleanupFunction(async function() {
|
|
try {
|
|
await IOUtils.remove(tmpDir, { recursive: true });
|
|
} catch (e) {
|
|
Cu.reportError(e);
|
|
}
|
|
});
|
|
Services.prefs.setIntPref("browser.download.folderList", 2);
|
|
Services.prefs.setCharPref("browser.download.dir", tmpDir);
|
|
return tmpDir;
|
|
}
|
|
|
|
/**
|
|
* All the tests are implemented with add_task, this starts them automatically.
|
|
*/
|
|
function run_test() {
|
|
do_get_profile();
|
|
run_next_test();
|
|
}
|
|
|
|
add_task(async function test_common_initialize() {
|
|
gDownloadDir = await setDownloadDir();
|
|
Services.prefs.setCharPref("browser.download.loglevel", "Debug");
|
|
});
|