blob: 1bd5b54f2b10177c6a58ca977c098d72beb1378b (
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
|
"use strict";
const HOSTS = new Set(["example.com"]);
const server = createHttpServer({ hosts: HOSTS });
server.registerPathHandler("/dummy", (request, response) => {
response.setStatusLine(request.httpVersion, 200, "OK");
response.write("ok");
});
add_task(async function test_webSocket() {
let extension = ExtensionTestUtils.loadExtension({
manifest: {
permissions: ["webRequest", "webRequestBlocking", "<all_urls>"],
},
background() {
browser.webRequest.onBeforeRequest.addListener(
details => {
browser.test.assertEq(
"ws:",
new URL(details.url).protocol,
"ws protocol worked"
);
browser.test.notifyPass("websocket");
},
{ urls: ["ws://example.com/*"] },
["blocking"]
);
browser.test.onMessage.addListener(() => {
let ws = new WebSocket("ws://example.com/dummy");
ws.onopen = () => {
ws.send("data");
};
ws.onclose = () => {};
ws.onerror = () => {};
ws.onmessage = () => {
ws.close();
};
});
browser.test.sendMessage("ready");
},
});
await extension.startup();
await extension.awaitMessage("ready");
extension.sendMessage("go");
await extension.awaitFinish("websocket");
// Wait until the next tick so that listener responses are processed
// before we unload.
await new Promise(executeSoon);
await extension.unload();
});
|
