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
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
|
"use strict";
// This is a pretty terrible hack, but it's the best we can do until we
// support |executeScript| callbacks and |lastError|.
function* testHasNoPermission(params) {
let contentSetup = params.contentSetup || (() => Promise.resolve());
async function background(contentSetup) {
browser.runtime.onMessage.addListener((msg, sender) => {
browser.test.assertEq(msg, "second script ran", "second script ran");
browser.test.notifyPass("executeScript");
});
browser.test.onMessage.addListener(msg => {
browser.test.assertEq(msg, "execute-script");
browser.tabs.query({currentWindow: true}, tabs => {
browser.tabs.executeScript({
file: "script.js",
});
// Execute a script we know we have permissions for in the
// second tab, in the hopes that it will execute after the
// first one. This has intermittent failure written all over
// it, but it's just about the best we can do until we
// support callbacks for executeScript.
browser.tabs.executeScript(tabs[1].id, {
file: "second-script.js",
});
});
});
await contentSetup();
browser.test.sendMessage("ready");
}
let extension = ExtensionTestUtils.loadExtension({
manifest: params.manifest,
background: `(${background})(${contentSetup})`,
files: {
"script.js": function() {
browser.runtime.sendMessage("first script ran");
},
"second-script.js": function() {
browser.runtime.sendMessage("second script ran");
},
},
});
yield extension.startup();
yield extension.awaitMessage("ready");
if (params.setup) {
yield params.setup(extension);
}
extension.sendMessage("execute-script");
yield extension.awaitFinish("executeScript");
yield extension.unload();
}
add_task(function* testBadPermissions() {
let tab1 = yield BrowserTestUtils.openNewForegroundTab(gBrowser, "http://example.com/");
let tab2 = yield BrowserTestUtils.openNewForegroundTab(gBrowser, "http://mochi.test:8888/");
info("Test no special permissions");
yield testHasNoPermission({
manifest: {"permissions": ["http://example.com/"]},
});
info("Test tabs permissions");
yield testHasNoPermission({
manifest: {"permissions": ["http://example.com/", "tabs"]},
});
info("Test no special permissions, commands, key press");
yield testHasNoPermission({
manifest: {
"permissions": ["http://example.com/"],
"commands": {
"test-tabs-executeScript": {
"suggested_key": {
"default": "Alt+Shift+K",
},
},
},
},
contentSetup() {
browser.commands.onCommand.addListener(function(command) {
if (command == "test-tabs-executeScript") {
browser.test.sendMessage("tabs-command-key-pressed");
}
});
return Promise.resolve();
},
setup: function* (extension) {
yield EventUtils.synthesizeKey("k", {altKey: true, shiftKey: true});
yield extension.awaitMessage("tabs-command-key-pressed");
},
});
info("Test active tab, commands, no key press");
yield testHasNoPermission({
manifest: {
"permissions": ["http://example.com/", "activeTab"],
"commands": {
"test-tabs-executeScript": {
"suggested_key": {
"default": "Alt+Shift+K",
},
},
},
},
});
info("Test active tab, browser action, no click");
yield testHasNoPermission({
manifest: {
"permissions": ["http://example.com/", "activeTab"],
"browser_action": {},
},
});
info("Test active tab, page action, no click");
yield testHasNoPermission({
manifest: {
"permissions": ["http://example.com/", "activeTab"],
"page_action": {},
},
async contentSetup() {
let [tab] = await browser.tabs.query({active: true, currentWindow: true});
await browser.pageAction.show(tab.id);
},
});
yield BrowserTestUtils.removeTab(tab2);
yield BrowserTestUtils.removeTab(tab1);
});
add_task(function* testBadURL() {
async function background() {
let promises = [
new Promise(resolve => {
browser.tabs.executeScript({
file: "http://example.com/script.js",
}, result => {
browser.test.assertEq(undefined, result, "Result value");
browser.test.assertTrue(browser.extension.lastError instanceof Error,
"runtime.lastError is Error");
browser.test.assertTrue(browser.runtime.lastError instanceof Error,
"runtime.lastError is Error");
browser.test.assertEq(
"Files to be injected must be within the extension",
browser.extension.lastError && browser.extension.lastError.message,
"extension.lastError value");
browser.test.assertEq(
"Files to be injected must be within the extension",
browser.runtime.lastError && browser.runtime.lastError.message,
"runtime.lastError value");
resolve();
});
}),
browser.tabs.executeScript({
file: "http://example.com/script.js",
}).catch(error => {
browser.test.assertTrue(error instanceof Error, "Error is Error");
browser.test.assertEq(null, browser.extension.lastError,
"extension.lastError value");
browser.test.assertEq(null, browser.runtime.lastError,
"runtime.lastError value");
browser.test.assertEq(
"Files to be injected must be within the extension",
error && error.message,
"error value");
}),
];
await Promise.all(promises);
browser.test.notifyPass("executeScript-lastError");
}
let extension = ExtensionTestUtils.loadExtension({
manifest: {
"permissions": ["<all_urls>"],
},
background,
});
yield extension.startup();
yield extension.awaitFinish("executeScript-lastError");
yield extension.unload();
});
// TODO: Test that |executeScript| fails if the tab has navigated to a
// new page, and no longer matches our expected state. This involves
// intentionally trying to trigger a race condition, and is probably not
// even worth attempting until we have proper |executeScript| callbacks.
add_task(forceGC);
|
