1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim: set sts=2 sw=2 et tw=80: */
"use strict";
const { AddonManager } = ChromeUtils.importESModule(
"resource://gre/modules/AddonManager.sys.mjs"
);
const { MockRegistrar } = ChromeUtils.importESModule(
"resource://testing-common/MockRegistrar.sys.mjs"
);
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();
});
|
