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
|
/* Any copyright is dedicated to the Public Domain.
* http://creativecommons.org/publicdomain/zero/1.0/ */
"use strict";
const USER_CONTEXTS = ["default", "personal", "work"];
const COOKIE_NAMES = ["cookie0", "cookie1", "cookie2"];
const TEST_URL =
"http://example.com/browser/netwerk/cookie/test/browser/file_empty.html";
// opens `uri' in a new tab with the provided userContextId and focuses it.
// returns the newly opened tab
async function openTabInUserContext(uri, userContextId) {
// open the tab in the correct userContextId
let tab = BrowserTestUtils.addTab(gBrowser, uri, { userContextId });
// select tab and make sure its browser is focused
gBrowser.selectedTab = tab;
tab.ownerGlobal.focus();
let browser = gBrowser.getBrowserForTab(tab);
// wait for tab load
await BrowserTestUtils.browserLoaded(browser);
return { tab, browser };
}
add_setup(async function () {
// make sure userContext is enabled.
await new Promise(resolve => {
SpecialPowers.pushPrefEnv(
{ set: [["privacy.userContext.enabled", true]] },
resolve
);
});
});
add_task(async function test() {
// load the page in 3 different contexts and set a cookie
// which should only be visible in that context
for (let userContextId of Object.keys(USER_CONTEXTS)) {
// open our tab in the given user context
let { tab, browser } = await openTabInUserContext(TEST_URL, userContextId);
await SpecialPowers.spawn(
browser,
[{ names: COOKIE_NAMES, value: USER_CONTEXTS[userContextId] }],
function (opts) {
for (let name of opts.names) {
content.document.cookie = name + "=" + opts.value;
}
}
);
// remove the tab
gBrowser.removeTab(tab);
}
let expectedValues = USER_CONTEXTS.slice(0);
await checkCookies(expectedValues, "before removal");
// remove cookies that belongs to user context id #1
Services.cookies.removeCookiesWithOriginAttributes(
JSON.stringify({ userContextId: 1 })
);
expectedValues[1] = undefined;
await checkCookies(expectedValues, "after removal");
});
async function checkCookies(expectedValues, time) {
for (let userContextId of Object.keys(expectedValues)) {
let cookiesFromTitle = await getCookiesFromJS(userContextId);
let cookiesFromManager = getCookiesFromManager(userContextId);
let expectedValue = expectedValues[userContextId];
for (let name of COOKIE_NAMES) {
is(
cookiesFromTitle[name],
expectedValue,
`User context ${userContextId}: ${name} should be correct from title ${time}`
);
is(
cookiesFromManager[name],
expectedValue,
`User context ${userContextId}: ${name} should be correct from manager ${time}`
);
}
}
}
function getCookiesFromManager(userContextId) {
let cookies = {};
let allCookies = Services.cookies.getCookiesWithOriginAttributes(
JSON.stringify({ userContextId })
);
for (let cookie of allCookies) {
cookies[cookie.name] = cookie.value;
}
return cookies;
}
async function getCookiesFromJS(userContextId) {
let { tab, browser } = await openTabInUserContext(TEST_URL, userContextId);
// get the cookies
let cookieString = await SpecialPowers.spawn(browser, [], function () {
return content.document.cookie;
});
// check each item in the title and validate it meets expectatations
let cookies = {};
for (let cookie of cookieString.split(";")) {
let [name, value] = cookie.trim().split("=");
cookies[name] = value;
}
gBrowser.removeTab(tab);
return cookies;
}
|
