blob: d4a1db8ed2c33ab2a7d096e137b47a42232de3c4 (
plain)
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
|
"use strict";
function delay(time) {
return new Promise(resolve => {
// eslint-disable-next-line mozilla/no-arbitrary-setTimeout
setTimeout(resolve, time);
});
}
const { Extension } = ChromeUtils.importESModule(
"resource://gre/modules/Extension.sys.mjs"
);
add_task(async function test_startup_request_handler() {
const ID = "request-startup@xpcshell.mozilla.org";
let extension = ExtensionTestUtils.loadExtension({
manifest: {
browser_specific_settings: { gecko: { id: ID } },
},
files: {
"meh.txt": "Meh.",
},
});
let ready = false;
let resolvePromise;
let promise = new Promise(resolve => {
resolvePromise = resolve;
});
promise.then(() => {
ready = true;
});
let origInitLocale = Extension.prototype.initLocale;
Extension.prototype.initLocale = async function initLocale() {
await promise;
return origInitLocale.call(this);
};
let startupPromise = extension.startup();
await delay(0);
let policy = WebExtensionPolicy.getByID(ID);
let url = policy.getURL("meh.txt");
let contentPage;
let pagePromise = ExtensionTestUtils.loadContentPage(url, { extension });
let resp = pagePromise.then(page => {
contentPage = page;
ok(ready, "Shouldn't get response before extension is ready");
return page.fetch(url);
});
await delay(2000);
resolvePromise();
await startupPromise;
let body = await resp;
equal(body, "Meh.", "Got the correct response");
await contentPage.close();
await extension.unload();
Extension.prototype.initLocale = origInitLocale;
});
|