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
|
/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim: set sts=2 sw=2 et tw=80: */
"use strict";
add_task(async function () {
let tab = await BrowserTestUtils.openNewForegroundTab(
gBrowser,
"http://mochi.test:8888/"
);
let extension = ExtensionTestUtils.loadExtension({
manifest: {
permissions: ["http://mochi.test/"],
},
background: function () {
let ports_received = 0;
let port_messages_received = 0;
browser.runtime.onConnect.addListener(port => {
browser.test.assertTrue(!!port, "port1 received");
ports_received++;
browser.test.assertEq(1, ports_received, "1 port received");
port.onMessage.addListener((msg, msgPort) => {
browser.test.assertEq(
"port message",
msg,
"listener1 port message received"
);
browser.test.assertEq(
port,
msgPort,
"onMessage should receive port as second argument"
);
port_messages_received++;
browser.test.assertEq(
1,
port_messages_received,
"1 port message received"
);
});
});
browser.runtime.onConnect.addListener(port => {
browser.test.assertTrue(!!port, "port2 received");
ports_received++;
browser.test.assertEq(2, ports_received, "2 ports received");
port.onMessage.addListener((msg, msgPort) => {
browser.test.assertEq(
"port message",
msg,
"listener2 port message received"
);
browser.test.assertEq(
port,
msgPort,
"onMessage should receive port as second argument"
);
port_messages_received++;
browser.test.assertEq(
2,
port_messages_received,
"2 port messages received"
);
browser.test.notifyPass("contentscript_connect.pass");
});
});
browser.tabs.executeScript({ file: "script.js" }).catch(e => {
browser.test.fail(`Error: ${e} :: ${e.stack}`);
browser.test.notifyFail("contentscript_connect.pass");
});
},
files: {
"script.js": function () {
let port = browser.runtime.connect();
port.postMessage("port message");
},
},
});
await extension.startup();
await extension.awaitFinish("contentscript_connect.pass");
await extension.unload();
BrowserTestUtils.removeTab(tab);
});
|