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
|
/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim: set sts=2 sw=2 et tw=80: */
"use strict";
add_task(
async function test_messaging_to_self_should_not_trigger_onMessage_onConnect() {
async function background() {
browser.runtime.onMessage.addListener(msg => {
browser.test.assertEq("msg from child", msg);
browser.test.sendMessage(
"sendMessage did not call same-frame onMessage"
);
});
browser.test.onMessage.addListener(msg => {
browser.test.assertEq(
"sendMessage with a listener in another frame",
msg
);
browser.runtime.sendMessage("should only reach another frame");
});
await browser.test.assertRejects(
browser.runtime.sendMessage("should not trigger same-frame onMessage"),
"Could not establish connection. Receiving end does not exist."
);
browser.runtime.onConnect.addListener(port => {
browser.test.assertEq("from-frame", port.name);
browser.runtime.connect({ name: "from-bg-2" });
});
await new Promise(resolve => {
let port = browser.runtime.connect({ name: "from-bg-1" });
port.onDisconnect.addListener(() => {
browser.test.assertEq(
"Could not establish connection. Receiving end does not exist.",
port.error.message
);
resolve();
});
});
let anotherFrame = document.createElement("iframe");
anotherFrame.src = browser.runtime.getURL("extensionpage.html");
document.body.appendChild(anotherFrame);
}
function lastScript() {
browser.runtime.onMessage.addListener(msg => {
browser.test.assertEq("should only reach another frame", msg);
browser.runtime.sendMessage("msg from child");
});
browser.test.sendMessage("sendMessage callback called");
browser.runtime.onConnect.addListener(port => {
browser.test.assertEq("from-bg-2", port.name);
browser.test.sendMessage("connect did not call same-frame onConnect");
});
browser.runtime.connect({ name: "from-frame" });
}
let extensionData = {
background,
files: {
"lastScript.js": lastScript,
"extensionpage.html": `<!DOCTYPE html><meta charset="utf-8"><script src="lastScript.js"></script>`,
},
};
let extension = ExtensionTestUtils.loadExtension(extensionData);
await extension.startup();
await extension.awaitMessage("sendMessage callback called");
extension.sendMessage("sendMessage with a listener in another frame");
await Promise.all([
extension.awaitMessage("connect did not call same-frame onConnect"),
extension.awaitMessage("sendMessage did not call same-frame onMessage"),
]);
await extension.unload();
}
);
|
