mozilla-central/toolkit/components/extensions/test/xpcshell/test_ext_management_uninstall_self.js (file symbol)

Source code

Revision control

Copy as Markdown

Other Tools

Test Info: Warnings

/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim: set sts=2 sw=2 et tw=80: */
"use strict";
const { AddonManager } = ChromeUtils.importESModule(
);
const { MockRegistrar } = ChromeUtils.importESModule(
);
const id = "uninstall_self_test@tests.mozilla.com";
const manifest = {
browser_specific_settings: {
gecko: {
id,
},
},
name: "test extension name",
version: "1.0",
};
const waitForUninstalled = () =>
new Promise(resolve => {
const listener = {
onUninstalled: async addon => {
equal(addon.id, id, "The expected add-on has been uninstalled");
let checkedAddon = await AddonManager.getAddonByID(addon.id);
equal(checkedAddon, null, "Add-on no longer exists");
AddonManager.removeAddonListener(listener);
resolve();
},
};
AddonManager.addAddonListener(listener);
});
let promptService = {
_response: null,
QueryInterface: ChromeUtils.generateQI(["nsIPromptService"]),
confirmEx: function (...args) {
this._confirmExArgs = args;
return this._response;
},
};
AddonTestUtils.init(this);
add_task(async function setup() {
let fakePromptService = MockRegistrar.register(
"@mozilla.org/prompter;1",
promptService
);
registerCleanupFunction(() => {
MockRegistrar.unregister(fakePromptService);
});
await ExtensionTestUtils.startAddonManager();
});
add_task(async function test_management_uninstall_no_prompt() {
function background() {
browser.test.onMessage.addListener(() => {
browser.management.uninstallSelf();
});
}
let extension = ExtensionTestUtils.loadExtension({
manifest,
background,
useAddonManager: "temporary",
});
await extension.startup();
let addon = await AddonManager.getAddonByID(id);
notEqual(addon, null, "Add-on is installed");
extension.sendMessage("uninstall");
await waitForUninstalled();
Services.obs.notifyObservers(extension.extension.file, "flush-cache-entry");
});
add_task(async function test_management_uninstall_prompt_uninstall() {
promptService._response = 0;
function background() {
browser.test.onMessage.addListener(() => {
browser.management.uninstallSelf({ showConfirmDialog: true });
});
}
let extension = ExtensionTestUtils.loadExtension({
manifest,
background,
useAddonManager: "temporary",
});
await extension.startup();
let addon = await AddonManager.getAddonByID(id);
notEqual(addon, null, "Add-on is installed");
extension.sendMessage("uninstall");
await waitForUninstalled();
// Test localization strings
equal(promptService._confirmExArgs[1], `Uninstall ${manifest.name}`);
equal(
promptService._confirmExArgs[2],
`The extension “${manifest.name}” is requesting to be uninstalled. What would you like to do?`
);
equal(promptService._confirmExArgs[4], "Uninstall");
equal(promptService._confirmExArgs[5], "Keep Installed");
Services.obs.notifyObservers(extension.extension.file, "flush-cache-entry");
});
add_task(async function test_management_uninstall_prompt_keep() {
promptService._response = 1;
function background() {
browser.test.onMessage.addListener(async () => {
await browser.test.assertRejects(
browser.management.uninstallSelf({ showConfirmDialog: true }),
"User cancelled uninstall of extension",
"Expected rejection when user declines uninstall"
);
browser.test.sendMessage("uninstall-rejected");
});
}
let extension = ExtensionTestUtils.loadExtension({
manifest,
background,
useAddonManager: "temporary",
});
await extension.startup();
let addon = await AddonManager.getAddonByID(id);
notEqual(addon, null, "Add-on is installed");
extension.sendMessage("uninstall");
await extension.awaitMessage("uninstall-rejected");
addon = await AddonManager.getAddonByID(id);
notEqual(addon, null, "Add-on remains installed");
await extension.unload();
});