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
var { MailServices } = ChromeUtils.importESModule(
);
const gAccount = createAccount("pop3");
createAccount("local");
MailServices.accounts.defaultAccount = gAccount;
const gDefaultIdentity = addIdentity(gAccount);
gDefaultIdentity.composeHtml = true;
const gNonDefaultIdentity = addIdentity(gAccount);
gNonDefaultIdentity.composeHtml = false;
const gRootFolder = gAccount.incomingServer.rootFolder;
gRootFolder.createSubfolder("test", null);
const gFolder = gRootFolder.getChildNamed("test");
createMessages(gFolder, 4);
add_task(async function testIdentity() {
const files = {
"background.js": async () => {
const accounts = await browser.accounts.list();
browser.test.assertEq(2, accounts.length, "number of accounts");
const popAccount = accounts.find(a => a.type == "pop3");
browser.test.assertEq(
2,
popAccount.identities.length,
"number of identities"
);
const [defaultIdentity, nonDefaultIdentity] = popAccount.identities;
const folder = popAccount.folders.find(f => f.name == "test");
const { messages } = await browser.messages.list(folder.id);
browser.test.assertEq(4, messages.length, "number of messages");
browser.test.log(defaultIdentity.id);
browser.test.log(nonDefaultIdentity.id);
const funcs = [
{ name: "beginNew", args: [] },
{ name: "beginReply", args: [messages[0].id] },
{ name: "beginForward", args: [messages[1].id, "forwardAsAttachment"] },
// Uses a different code path.
{ name: "beginForward", args: [messages[2].id, "forwardInline"] },
{ name: "beginNew", args: [messages[3].id] },
];
const tests = [
{ args: [], isDefault: true },
{
args: [{ identityId: defaultIdentity.id }],
isDefault: true,
},
{
args: [{ identityId: nonDefaultIdentity.id }],
isDefault: false,
},
];
for (const func of funcs) {
browser.test.log(func.name);
for (const test of tests) {
browser.test.log(JSON.stringify(test.args));
const tab = await browser.compose[func.name](
...func.args.concat(test.args)
);
browser.test.assertEq("object", typeof tab);
browser.test.assertEq("number", typeof tab.id);
await window.sendMessage("checkIdentity", test.isDefault);
}
}
browser.test.notifyPass("finished");
},
"utils.js": await getUtilsJS(),
};
const extension = ExtensionTestUtils.loadExtension({
files,
manifest: {
background: { scripts: ["utils.js", "background.js"] },
permissions: ["accountsRead", "messagesRead"],
},
});
extension.onMessage("checkIdentity", async isDefault => {
const composeWindows = [...Services.wm.getEnumerator("msgcompose")];
is(composeWindows.length, 1);
await new Promise(resolve => composeWindows[0].setTimeout(resolve));
is(
composeWindows[0].getCurrentIdentityKey(),
isDefault ? gDefaultIdentity.key : gNonDefaultIdentity.key
);
composeWindows[0].close();
extension.sendMessage();
});
await extension.startup();
await extension.awaitFinish("finished");
await extension.unload();
});