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
|
/* Any copyright is dedicated to the Public Domain.
https://creativecommons.org/publicdomain/zero/1.0/ */
"use strict";
add_task(async function test_runtime_sendMessage() {
const extension = ExtensionTestUtils.loadExtension({
manifest: {
background: {
service_worker: "sw.js",
},
browser_specific_settings: {
gecko: { id: "@test-messaging" },
},
},
files: {
"sw.js": async function () {
browser.test.onMessage.addListener(msg => {
browser.test.assertEq("send", msg, "expected correct message");
browser.runtime.sendMessage("hello-from-sw");
});
browser.test.sendMessage("background-ready");
},
"page.html": '<!DOCTYPE html><script src="page.js"></script>',
"page.js"() {
browser.runtime.onMessage.addListener((msg, sender) => {
browser.test.assertEq(
"hello-from-sw",
msg,
"expected message from service worker"
);
const { contextId, ...otherProps } = sender;
browser.test.assertTrue(!!contextId, "expected a truthy contextId");
browser.test.assertDeepEq(
{
envType: "addon_child",
id: "@test-messaging",
origin: self.origin,
url: browser.runtime.getURL("sw.js"),
},
otherProps,
"expected correct sender props"
);
browser.test.sendMessage("page-done");
});
browser.test.sendMessage("page-ready");
},
},
});
await extension.startup();
await extension.awaitMessage("background-ready");
const page = await ExtensionTestUtils.loadContentPage(
`moz-extension://${extension.uuid}/page.html`,
{ extension }
);
await extension.awaitMessage("page-ready");
extension.sendMessage("send");
await extension.awaitMessage("page-done");
await page.close();
await extension.unload();
});
add_task(async function test_runtime_connect() {
const extension = ExtensionTestUtils.loadExtension({
manifest: {
background: {
service_worker: "sw.js",
},
browser_specific_settings: {
gecko: { id: "@test-messaging" },
},
},
files: {
"sw.js": async function () {
browser.test.onMessage.addListener(msg => {
browser.test.assertEq("connect", msg, "expected correct message");
browser.runtime.connect();
});
browser.test.sendMessage("background-ready");
},
"page.html": '<!DOCTYPE html><script src="page.js"></script>',
"page.js"() {
browser.runtime.onConnect.addListener(port => {
const { contextId, ...otherProps } = port.sender;
browser.test.assertTrue(!!contextId, "expected a truthy contextId");
browser.test.assertDeepEq(
{
envType: "addon_child",
id: "@test-messaging",
origin: self.origin,
url: browser.runtime.getURL("sw.js"),
},
otherProps,
"expected correct sender props"
);
browser.test.sendMessage("page-done");
});
browser.test.sendMessage("page-ready");
},
},
});
await extension.startup();
await extension.awaitMessage("background-ready");
const page = await ExtensionTestUtils.loadContentPage(
`moz-extension://${extension.uuid}/page.html`,
{ extension }
);
await extension.awaitMessage("page-ready");
extension.sendMessage("connect");
await extension.awaitMessage("page-done");
await page.close();
await extension.unload();
});
|
