Revision control
Copy as Markdown
Other Tools
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
/*
* Test that temporary files for draft are surely removed.
*/
var gMsgCompose;
var gExpectedFiles;
var progressListener = {
onStateChange(aWebProgress, aRequest, aStateFlags) {
if (aStateFlags & Ci.nsIWebProgressListener.STATE_STOP) {
do_timeout(0, checkResult);
}
},
onProgressChange() {},
onLocationChange() {},
onStatusChange() {},
onSecurityChange() {},
onContentBlockingEvent() {},
QueryInterface: ChromeUtils.generateQI([
"nsIWebProgressListener",
"nsISupportsWeakReference",
]),
};
/**
* Get the count of temporary files. Because nsIFile.createUnique creates a random
* file name, we iterate the tmp dir and count the files that match filename
* patterns.
*/
async function getTemporaryFilesCount() {
const tmpDir = Services.dirsvc.get("TmpD", Ci.nsIFile).path;
const entries = await IOUtils.getChildren(tmpDir);
const tempFiles = {
"nsmail.tmp": 0,
"nscopy.tmp": 0,
"nsemail.eml": 0,
"nsemail.tmp": 0,
"nsqmail.tmp": 0,
};
for (const path of entries) {
for (const pattern of Object.keys(tempFiles)) {
const [name, extName] = pattern.split(".");
if (PathUtils.filename(path).startsWith(name) && path.endsWith(extName)) {
tempFiles[pattern]++;
}
}
}
return tempFiles;
}
/**
* Temp files should be deleted as soon as the draft is finished saving, so the
* counts should be the same as before.
*/
async function checkResult() {
const filesCount = await getTemporaryFilesCount();
for (const [pattern, count] of Object.entries(filesCount)) {
Assert.equal(
count,
gExpectedFiles[pattern],
`${pattern} should not exists`
);
}
do_test_finished();
}
add_task(async function () {
gExpectedFiles = await getTemporaryFilesCount();
// Ensure we have at least one mail account
localAccountUtils.loadLocalMailAccount();
gMsgCompose = Cc["@mozilla.org/messengercompose/compose;1"].createInstance(
Ci.nsIMsgCompose
);
const fields = Cc[
"@mozilla.org/messengercompose/composefields;1"
].createInstance(Ci.nsIMsgCompFields);
const params = Cc[
"@mozilla.org/messengercompose/composeparams;1"
].createInstance(Ci.nsIMsgComposeParams);
fields.from = "Nobody <nobody@tinderbox.test>";
fields.body = "body text";
fields.useMultipartAlternative = true;
params.composeFields = fields;
params.format = Ci.nsIMsgCompFormat.HTML;
gMsgCompose.initialize(params, null, null);
const identity = getSmtpIdentity(null, getBasicSmtpServer());
localAccountUtils.rootFolder.createLocalSubfolder("Drafts");
const progress = Cc["@mozilla.org/messenger/progress;1"].createInstance(
Ci.nsIMsgProgress
);
progress.registerListener(progressListener);
do_test_pending();
gMsgCompose.sendMsg(
Ci.nsIMsgSend.nsMsgSaveAsDraft,
identity,
"",
null,
progress
);
});