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 { ExtensionTestUtils } = ChromeUtils.importESModule(
);
add_task(async function () {
const extension = ExtensionTestUtils.loadExtension({
background: async () => {
let id = "9b9074ff-8fa4-4c58-9c3b-bc9ea2e17db1";
const dummy = async () => {
await browser.test.assertTrue(
false,
"Should have removed this address book"
);
};
await browser.addressBooks.provider.onSearchRequest.addListener(dummy, {
addressBookName: "dummy",
isSecure: false,
id,
});
await browser.addressBooks.provider.onSearchRequest.removeListener(dummy);
id = "00e1d9af-a846-4ef5-a6ac-15e8926bf6d3";
await browser.addressBooks.provider.onSearchRequest.addListener(
async (node, searchString) => {
await browser.test.assertEq(
id,
node.id,
"Addressbook should have the id we requested"
);
return {
results: [
{
DisplayName: searchString,
PrimaryEmail: searchString + "@example.com",
},
],
isCompleteResult: true,
};
},
{
addressBookName: "xpcshell",
isSecure: false,
id,
}
);
await browser.addressBooks.provider.onSearchRequest.addListener(
async () => {
await browser.test.assertTrue(
false,
"Should not have created a duplicate address book"
);
},
{
addressBookName: "xpcshell",
isSecure: false,
id,
}
);
},
manifest: { permissions: ["addressBooks"] },
});
await extension.startup();
const dummyUID = "9b9074ff-8fa4-4c58-9c3b-bc9ea2e17db1";
let searchBook = MailServices.ab.getDirectoryFromUID(dummyUID);
Assert.equal(searchBook, null, "Dummy directory was removed by extension");
const UID = "00e1d9af-a846-4ef5-a6ac-15e8926bf6d3";
searchBook = MailServices.ab.getDirectoryFromUID(UID);
Assert.notEqual(searchBook, null, "Extension registered an async directory");
let foundCards = 0;
await new Promise(resolve => {
searchBook.search(null, "test", {
onSearchFoundCard(card) {
Assert.notEqual(card, null, "A card was found.");
equal(card.directoryUID, UID, "The card comes from the directory.");
equal(
card.primaryEmail,
"test@example.com",
"The card has the correct email address."
);
equal(
card.displayName,
"test",
"The card has the correct display name."
);
foundCards++;
},
onSearchFinished(status, isCompleteResult) {
ok(Components.isSuccessCode(status), "Search finished successfully.");
equal(foundCards, 1, "One card was found.");
ok(isCompleteResult, "A full result set was received.");
resolve();
},
});
});
const autoCompleteSearch = Cc[
"@mozilla.org/autocomplete/search;1?name=addrbook"
].createInstance(Ci.nsIAutoCompleteSearch);
await new Promise(resolve => {
autoCompleteSearch.startSearch("test", null, null, {
onSearchResult(aSearch, aResult) {
equal(aSearch, autoCompleteSearch, "This is our search.");
if (aResult.searchResult == Ci.nsIAutoCompleteResult.RESULT_SUCCESS) {
equal(aResult.matchCount, 1, "One match was found.");
equal(
aResult.getValueAt(0),
"test <test@example.com>",
"The match had the expected value."
);
resolve();
} else {
equal(
aResult.searchResult,
Ci.nsIAutoCompleteResult.RESULT_NOMATCH_ONGOING,
"We should be waiting for the extension's results."
);
}
},
});
});
await extension.unload();
searchBook = MailServices.ab.getDirectoryFromUID(UID);
Assert.equal(searchBook, null, "Extension directory removed after unload");
});
registerCleanupFunction(() => {
// Make sure any open database is given a chance to close.
Services.startup.advanceShutdownPhase(
Services.startup.SHUTDOWN_PHASE_APPSHUTDOWNCONFIRMED
);
});