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
"use strict";
const { PromiseTestUtils } = ChromeUtils.importESModule(
);
add_setup(function () {
localAccountUtils.loadLocalMailAccount(
"@mozilla.org/msgstore/berkeleystore;1"
);
});
add_task(async function test_fileCopyErrorStatus() {
const target = localAccountUtils.rootFolder.createLocalSubfolder(
"missing-file-target"
);
const file = do_get_profile().clone();
file.append("missing-message.eml");
const listener = new PromiseTestUtils.PromiseCopyListener();
Assert.throws(
() =>
MailServices.copy.copyFileMessage(
file,
target,
null,
false,
0,
"",
listener,
null
),
error => error.result == Cr.NS_ERROR_FILE_NOT_FOUND,
"Copying a missing message file should fail with the original file error"
);
await Assert.rejects(
listener.promise,
new RegExp(Cr.NS_ERROR_FILE_NOT_FOUND),
"The copy listener should receive the original file error"
);
Assert.equal(
target.getTotalMessages(false),
0,
"The failed file copy should leave the destination empty"
);
});
/**
* Checks that a partial copy or move reports failure and preserves a valid
* destination summary after closing and reopening the database.
*
* @param {boolean} isMove - Whether to move the messages.
*/
async function checkPartialFailure(isMove) {
const root = localAccountUtils.rootFolder;
const source = root
.createLocalSubfolder(isMove ? "move-source" : "copy-source")
.QueryInterface(Ci.nsIMsgLocalMailFolder);
const target = root.createLocalSubfolder(
isMove ? "move-target" : "copy-target"
);
for (let index = 0; index < 3; index++) {
source.addMessage(
"From: sender@example.invalid\r\n" +
"To: recipient@example.invalid\r\n" +
`Subject: Partial copy ${index}\r\n` +
`Message-ID: <partial-${isMove}-${index}@example.invalid>\r\n` +
`\r\nBody ${index}.\r\n`
);
}
const messages = [...source.messages];
const firstMessageId = messages[0].messageId;
const targetDB = target.msgDatabase;
const originalToken = messages[1].storeToken;
registerCleanupFunction(() => {
messages[1].storeToken = originalToken;
});
// Fail while reading the second message, after the first copy is committed.
messages[1].storeToken = String(Number(originalToken) + 3);
const listener = new PromiseTestUtils.PromiseCopyListener();
MailServices.copy.copyMessages(
source,
messages,
target,
isMove,
listener,
null,
false
);
await Assert.rejects(
listener.promise,
new RegExp(Cr.NS_ERROR_FAILURE),
"The partial copy or move should report failure"
);
messages[1].storeToken = originalToken;
Assert.equal(
source.getTotalMessages(false),
messages.length,
"The failed batch should retain all source messages"
);
Assert.ok(
targetDB.summaryValid,
"The destination summary should remain valid after the failed batch"
);
target.ForceDBClosed();
Assert.ok(
target.msgDatabase.summaryValid,
"The destination summary should reopen without reparsing"
);
Assert.deepEqual(
[...target.messages].map(message => message.messageId),
[firstMessageId],
"The reopened destination should contain only the completed copy"
);
const [copied] = target.messages;
const content = mailTestUtils.loadMessageToString(target, copied);
Assert.ok(
content.includes("Body 0."),
"The completed copy should retain its body after reopening the database"
);
}
add_task(async function test_partialCopy() {
await checkPartialFailure(false);
});
add_task(async function test_partialMove() {
await checkPartialFailure(true);
});