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 { open_compose_new_mail, save_compose_message } =
ChromeUtils.importESModule(
);
const {
open_message_from_file,
be_in_folder,
get_about_message,
get_special_folder,
select_click_row,
open_selected_message,
} = ChromeUtils.importESModule(
);
const { OpenPGPTestUtils } = ChromeUtils.importESModule(
);
const { MailServices } = ChromeUtils.importESModule(
);
function waitForComposeWindow() {
return BrowserTestUtils.domWindowOpened(null, async win => {
await BrowserTestUtils.waitForEvent(win, "load");
await BrowserTestUtils.waitForEvent(win, "focus", true);
return (
win.document.documentURI ===
);
});
}
let aliceAcct;
let aliceIdentity;
let initialKeyIdPref = "";
/**
* Delete all messages in the given folder.
*
* @param {nsIMsgFolder} folder - The folder to empty.
*/
async function cleanUpFolder(folder) {
return new Promise(resolve => {
const msgs = [...folder.msgDatabase.enumerateMessages()];
folder.deleteMessages(
msgs,
null,
true,
false,
{ onStopCopy: resolve },
false
);
});
}
/**
* Setup a mail account with a private key and an imported public key for an
* address we can send messages to.
*/
add_setup(async function () {
aliceAcct = MailServices.accounts.createAccount();
aliceAcct.incomingServer = MailServices.accounts.createIncomingServer(
"alice",
"openpgp.example",
"imap"
);
aliceIdentity = MailServices.accounts.createIdentity();
aliceIdentity.email = "alice@openpgp.example";
aliceAcct.addIdentity(aliceIdentity);
const [id] = await OpenPGPTestUtils.importPrivateKey(
window,
new FileUtils.File(
getTestFilePath(
"data/keys/alice@openpgp.example-0xf231550c4f47e38e-secret.asc"
)
)
);
Assert.ok(id, "private key id received");
initialKeyIdPref = aliceIdentity.getUnicharAttribute("openpgp_key_id");
aliceIdentity.setUnicharAttribute("openpgp_key_id", id.split("0x").join(""));
await OpenPGPTestUtils.importPublicKey(
window,
new FileUtils.File(
getTestFilePath(
"data/keys/bob@openpgp.example-0xfbfcc82a015e7330-pub.asc"
)
)
);
});
/**
* Test that drafts are encrypted.
*/
add_task(async function testDraftEncryption() {
const draftsFolder = await get_special_folder(
Ci.nsMsgFolderFlags.Drafts,
true,
aliceAcct.incomingServer.localFoldersServer
);
await be_in_folder(aliceAcct.incomingServer.rootFolder);
const composeWindow = await open_compose_new_mail();
composeWindow.document.getElementById("messageEditor").focus();
EventUtils.sendString("Hush, little baby don't say a word", composeWindow);
// Without the delay, the saving attempt that follows sometimes
// fails. The code to save the encrypted draft fails, because
// it finds no recipients. Maybe after opening the window,
// some time is needed to correctly populate the compose window.
// eslint-disable-next-line mozilla/no-arbitrary-setTimeout
await new Promise(resolve => setTimeout(resolve, 100));
await be_in_folder(draftsFolder);
await save_compose_message(composeWindow);
await TestUtils.waitForCondition(
() => draftsFolder.getTotalMessages(true) > 0,
"message saved to drafts folder"
);
composeWindow.close();
await select_click_row(0);
const aboutMessage = get_about_message();
Assert.ok(
OpenPGPTestUtils.hasEncryptedIconState(aboutMessage.document, "ok"),
"draft should be encrypted"
);
await cleanUpFolder(draftsFolder);
});
/**
* Test the "Re:" prefix remains in the compose window when opening a draft
*/
add_task(async function testDraftReplyToEncryptedMessageKeepsRePrefix() {
const draftsFolder = await get_special_folder(
Ci.nsMsgFolderFlags.Drafts,
true,
aliceAcct.incomingServer.localFoldersServer
);
await be_in_folder(draftsFolder);
// Test signed-encrypted and unsigned-encrypted messages.
const msgFiles = [
"data/eml/signed-by-0xfbfcc82a015e7330-encrypted-to-0xf231550c4f47e38e.eml",
"data/eml/unsigned-encrypted-to-0xf231550c4f47e38e-from-0xfbfcc82a015e7330.eml",
];
let wantedRow = 0;
for (const msg of msgFiles) {
const msgc = await open_message_from_file(
new FileUtils.File(getTestFilePath(msg))
);
const replyWindowPromise = waitForComposeWindow();
get_about_message(msgc).document.querySelector("#hdrReplyButton").click();
await BrowserTestUtils.closeWindow(msgc);
const replyWindow = await replyWindowPromise;
// Without the delay, the saving attempt that follows sometimes
// fails. The code to save the encrypted draft fails, because
// it finds no recipients. Maybe after opening the window,
// some time is needed to correctly populate the reply window.
// eslint-disable-next-line mozilla/no-arbitrary-setTimeout
await new Promise(resolve => setTimeout(resolve, 100));
await save_compose_message(replyWindow);
replyWindow.close();
await TestUtils.waitForCondition(
() => draftsFolder.getTotalMessages(true) > 0,
"message saved to drafts folder"
);
const draftWindowPromise = waitForComposeWindow();
await select_click_row(wantedRow);
++wantedRow;
open_selected_message();
const draftWindow = await draftWindowPromise;
Assert.ok(
draftWindow.document.querySelector("#msgSubject").value.startsWith("Re:"),
"the Re: prefix is applied"
);
draftWindow.close();
}
// Delete the messages we saved to drafts.
await cleanUpFolder(draftsFolder);
});
registerCleanupFunction(function tearDown() {
aliceIdentity.setUnicharAttribute("openpgp_key_id", initialKeyIdPref);
MailServices.accounts.removeIncomingServer(aliceAcct.incomingServer, true);
MailServices.accounts.removeAccount(aliceAcct);
});