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
/*
* Tests detach to file.
*/
var { PromiseTestUtils } = ChromeUtils.importESModule(
);
var { AttachmentInfo } = ChromeUtils.importESModule(
);
var { MailServices } = ChromeUtils.importESModule(
);
var { MsgHdrToMimeMessage } = ChromeUtils.importESModule(
);
function SaveAttachmentCallback() {
this.attachments = null;
this._promise = new Promise((resolve, reject) => {
this._resolve = resolve;
this._reject = reject;
});
}
SaveAttachmentCallback.prototype = {
callback: function saveAttachmentCallback_callback(aMsgHdr, aMimeMessage) {
this.attachments = aMimeMessage.allAttachments;
this._resolve();
},
get promise() {
return this._promise;
},
};
var gCallbackObject = new SaveAttachmentCallback();
add_setup(async function () {
if (!localAccountUtils.inboxFolder) {
localAccountUtils.loadLocalMailAccount();
}
});
add_task(async function startCopy() {
// Get a message into the local filestore.
const mailFile = do_get_file("../../../data/external-attach-test");
const listener = new PromiseTestUtils.PromiseCopyListener();
MailServices.copy.copyFileMessage(
mailFile,
localAccountUtils.inboxFolder,
null,
false,
0,
"",
listener,
null
);
await listener.promise;
});
// process the message through mime
add_task(async function startMime() {
const msgHdr = mailTestUtils.firstMsgHdr(localAccountUtils.inboxFolder);
MsgHdrToMimeMessage(
msgHdr,
gCallbackObject,
gCallbackObject.callback,
true // allowDownload
);
await gCallbackObject.promise;
});
// detach any found attachments
add_task(async function detachAttachments() {
let msgHdr = mailTestUtils.firstMsgHdr(localAccountUtils.inboxFolder);
const attachment = new AttachmentInfo(gCallbackObject.attachments[0]);
const profileDir = do_get_profile();
await AttachmentInfo.detachAttachments(msgHdr, [attachment], profileDir.path);
// test that the detachment was successful
// The message contained a file "check.pdf" which should
// now exist in the profile directory.
const checkFile = profileDir.clone();
checkFile.append(attachment.name);
Assert.ok(checkFile.exists(), `${checkFile.path} should exist`);
const fileInfo = await IOUtils.stat(checkFile.path);
Assert.equal(fileInfo.type, "regular", `The file type should be correct`);
Assert.greater(
fileInfo.size,
0,
`The file ${checkFile.path} should have size`
);
// The message should now have a detached attachment. Read the message,
// and search for "AttachmentDetached" which is added on detachment.
// Get the message header
msgHdr = mailTestUtils.firstMsgHdr(localAccountUtils.inboxFolder);
const messageContent = await getContentFromMessage(msgHdr);
Assert.ok(messageContent.includes("AttachmentDetached"));
});
/**
* Get the full message content.
*
* @param {nsIMsgDBHdr} aMsgHdr - Message whose text body will be read.
* @returns {Promise<string>} full message contents.
*/
function getContentFromMessage(aMsgHdr) {
const msgFolder = aMsgHdr.folder;
const msgUri = msgFolder.getUriForMsg(aMsgHdr);
return new Promise((resolve, reject) => {
const streamListener = {
QueryInterface: ChromeUtils.generateQI(["nsIStreamListener"]),
sis: Cc["@mozilla.org/scriptableinputstream;1"].createInstance(
Ci.nsIScriptableInputStream
),
content: "",
onDataAvailable(request, inputStream, offset, count) {
this.sis.init(inputStream);
this.content += this.sis.read(count);
},
onStartRequest() {},
onStopRequest(request, statusCode) {
this.sis.close();
if (Components.isSuccessCode(statusCode)) {
resolve(this.content);
} else {
reject(new Error(statusCode));
}
},
};
MailServices.messageServiceFromURI(msgUri).streamMessage(
msgUri,
streamListener,
null,
null,
false,
"",
false
);
});
}