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 file,
const { PromiseTestUtils } = ChromeUtils.importESModule(
);
/**
* nsIMsgPluggableStore interface tests
*/
function test_discoverSubFolders() {
const mailbox = setup_mailbox("none", create_temporary_directory());
mailbox.msgStore.discoverSubFolders(mailbox, true);
}
function test_sliceStream() {
const mailbox = setup_mailbox("none", create_temporary_directory());
const str = "Just a test string.";
const strStream = Cc["@mozilla.org/io/string-input-stream;1"].createInstance(
Ci.nsIStringInputStream
);
strStream.setData(str, str.length);
const sliced = mailbox.msgStore.sliceStream(strStream, 7, 4);
const s = Cc["@mozilla.org/scriptableinputstream;1"].createInstance(
Ci.nsIScriptableInputStream
);
s.init(sliced);
const chunk = s.read(1024);
Assert.equal(chunk, "test", "Check we got the expected subset.");
Assert.equal(s.available(), 0, "Check no more bytes available.");
Assert.equal(s.read(1024), "", "Check read() returns EOF.");
}
// Load messages into a msgStore and make sure we can read
// them back correctly using asyncScan().
async function test_AsyncScan() {
// NOTE: we should be able to create stand-alone msgStore to run tests on,
// but currently they are tightly coupled with folders, msgDB et al...
localAccountUtils.loadLocalMailAccount();
const inbox = localAccountUtils.inboxFolder;
// Populate the folder with the test messages.
const msg1 =
"To: bob@invalid\r\n" +
"From: alice@invalid\r\n" +
"Subject: Hello\r\n" +
"\r\n" +
"Hello, Bob! Haven't heard\r\n" +
"From you in a while...\r\n"; // escaping will be required on this line.
const msg2 =
"To: alice@invalid\r\n" +
"From: bob@invalid\r\n" +
"Subject: Re: Hello\r\n" +
"\r\n" +
"Hi there Alice! All good here.\r\n";
const messages = [msg1, msg2];
inbox.addMessageBatch(messages);
// Perform an async scan on the folder, and make sure we get back all
// the messages we put in.
const listener = new PromiseTestUtils.PromiseStoreScanListener();
inbox.msgStore.asyncScan(inbox, listener);
await listener.promise;
// Note: can't rely on message ordering (especially on maildir).
Assert.deepEqual(listener.messages.toSorted(), messages.toSorted());
// Clear up so we can run again on different store type.
localAccountUtils.clearAll();
}
// Return a wrapper which sets the store type before running fn().
function withStore(store, fn) {
return async () => {
Services.prefs.setCharPref("mail.serverDefaultStoreContractID", store);
await fn();
};
}
for (const store of localAccountUtils.pluggableStores) {
add_task(withStore(store, test_discoverSubFolders));
add_task(withStore(store, test_sliceStream));
add_task(withStore(store, test_AsyncScan));
}