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
|
let { classes: Cc, interfaces: Ci } = Components;
const BASE_ORIGIN = "http://example.com";
const URI = BASE_ORIGIN +
"/browser/browser/components/contextualidentity/test/browser/empty_file.html";
// opens `uri' in a new tab with the provided userContextId and focuses it.
// returns the newly opened tab
function* openTabInUserContext(uri, userContextId) {
// open the tab in the correct userContextId
let tab = gBrowser.addTab(uri, {userContextId});
// select tab and make sure its browser is focused
gBrowser.selectedTab = tab;
tab.ownerGlobal.focus();
let browser = gBrowser.getBrowserForTab(tab);
yield BrowserTestUtils.browserLoaded(browser);
return {tab, browser};
}
add_task(function* setup() {
// make sure userContext is enabled.
yield new Promise(resolve => {
SpecialPowers.pushPrefEnv({"set": [
["privacy.userContext.enabled", true]
]}, resolve);
});
});
add_task(function* test() {
let receiver = yield* openTabInUserContext(URI, 2);
let channelName = "contextualidentity-broadcastchannel";
// reflect the received message on title
yield ContentTask.spawn(receiver.browser, channelName,
function (name) {
content.window.testPromise = new content.window.Promise(resolve => {
content.window.bc = new content.window.BroadcastChannel(name);
content.window.bc.onmessage = function (e) {
content.document.title += e.data;
resolve();
}
});
}
);
let sender1 = yield* openTabInUserContext(URI, 1);
let sender2 = yield* openTabInUserContext(URI, 2);
sender1.message = "Message from user context #1";
sender2.message = "Message from user context #2";
// send a message from a tab in different user context first
// then send a message from a tab in the same user context
for (let sender of [sender1, sender2]) {
yield ContentTask.spawn(
sender.browser,
{ name: channelName, message: sender.message },
function (opts) {
let bc = new content.window.BroadcastChannel(opts.name);
bc.postMessage(opts.message);
});
}
// Since sender1 sends before sender2, if the title is exactly
// sender2's message, sender1's message must've been blocked
yield ContentTask.spawn(receiver.browser, sender2.message,
function* (message) {
yield content.window.testPromise.then(function() {
is(content.document.title, message,
"should only receive messages from the same user context");
});
}
);
gBrowser.removeTab(sender1.tab);
gBrowser.removeTab(sender2.tab);
gBrowser.removeTab(receiver.tab);
});
|
