diff options
Diffstat (limited to 'comm/mailnews/imap/test/unit/test_saveTemplate.js')
-rw-r--r-- | comm/mailnews/imap/test/unit/test_saveTemplate.js | 96 |
1 files changed, 96 insertions, 0 deletions
diff --git a/comm/mailnews/imap/test/unit/test_saveTemplate.js b/comm/mailnews/imap/test/unit/test_saveTemplate.js new file mode 100644 index 0000000000..12560411ee --- /dev/null +++ b/comm/mailnews/imap/test/unit/test_saveTemplate.js @@ -0,0 +1,96 @@ +/* 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, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +/* + * Tests imap save of message as a template, and test initial save right after + * creation of folder. + */ + +var { MailServices } = ChromeUtils.import( + "resource:///modules/MailServices.jsm" +); +var { MailUtils } = ChromeUtils.import("resource:///modules/MailUtils.jsm"); +var { MessageGenerator } = ChromeUtils.import( + "resource://testing-common/mailnews/MessageGenerator.jsm" +); +var { PromiseTestUtils } = ChromeUtils.import( + "resource://testing-common/mailnews/PromiseTestUtils.jsm" +); + +add_setup(function () { + setupIMAPPump(); + Services.prefs.setBoolPref( + "mail.server.default.autosync_offline_stores", + false + ); +}); + +// load and update a message in the imap fake server +add_task(async function loadImapMessage() { + let gMessageGenerator = new MessageGenerator(); + // create a synthetic message with attachment + let smsg = gMessageGenerator.makeMessage(); + + let msgURI = Services.io.newURI( + "data:text/plain;base64," + btoa(smsg.toMessageString()) + ); + let imapInbox = IMAPPump.daemon.getMailbox("INBOX"); + let message = new ImapMessage(msgURI.spec, imapInbox.uidnext++, []); + IMAPPump.mailbox.addMessage(message); + let listener = new PromiseTestUtils.PromiseUrlListener(); + IMAPPump.inbox.updateFolderWithListener(null, listener); + await listener.promise; +}); + +// This is similar to the method in mailCommands.js, to test the way that +// it creates a new templates folder before saving the message as a template. +add_task(async function saveAsTemplate() { + // Prepare msgAddedListener for this test. + let msgAddedListener = new MsgAddedListener(); + MailServices.mfn.addListener(msgAddedListener, MailServices.mfn.msgAdded); + + let hdr = mailTestUtils.firstMsgHdr(IMAPPump.inbox); + let uri = IMAPPump.inbox.getUriForMsg(hdr); + let identity = MailServices.accounts.getFirstIdentityForServer( + IMAPPump.incomingServer + ); + identity.stationeryFolder = + IMAPPump.incomingServer.rootFolder.URI + "/Templates"; + let templates = MailUtils.getOrCreateFolder(identity.stationeryFolder); + // Verify that Templates folder doesn't exist, and then create it. + Assert.equal(templates.parent, null); + templates.setFlag(Ci.nsMsgFolderFlags.Templates); + let listener = new PromiseTestUtils.PromiseUrlListener({ + OnStopRunningUrl() { + let messenger = Cc["@mozilla.org/messenger;1"].createInstance( + Ci.nsIMessenger + ); + messenger.saveAs(uri, false, identity, null); + }, + }); + templates.createStorageIfMissing(listener); + await listener.promise; + + await msgAddedListener.promise; +}); + +// Cleanup +add_task(function endTest() { + teardownIMAPPump(); +}); + +// listener for saveAsTemplate adding a message to the templates folder. +function MsgAddedListener() { + this._promise = new Promise(resolve => { + this._resolve = resolve; + }); +} + +MsgAddedListener.prototype = { + msgAdded(aMsg) { + // Check this is the templates folder. + Assert.equal(aMsg.folder.prettyName, "Templates"); + this._resolve(); + }, +}; |