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
|
"use strict";
const REFERRER1 = "http://example.org/?" + Date.now();
const REFERRER2 = "http://example.org/?" + Math.random();
const REFERRER3 = "http://example.org/?" + Math.random();
add_task(async function () {
function getExpectedReferrer(referrer) {
let defaultPolicy = Services.prefs.getIntPref(
"network.http.referer.defaultPolicy"
);
Assert.greater(
[2, 3].indexOf(defaultPolicy),
-1,
"default referrer policy should be either strict-origin-when-cross-origin(2) or no-referrer-when-downgrade(3)"
);
if (defaultPolicy == 2) {
return referrer.match(/https?:\/\/[^\/]+\/?/i)[0];
}
return referrer;
}
async function checkDocumentReferrer(referrer, msg) {
await SpecialPowers.spawn(
gBrowser.selectedBrowser,
[{ referrer, msg }],
async function (args) {
Assert.equal(content.document.referrer, args.referrer, args.msg);
}
);
}
let ReferrerInfo = Components.Constructor(
"@mozilla.org/referrer-info;1",
"nsIReferrerInfo",
"init"
);
// Add a new tab.
let tab = (gBrowser.selectedTab = BrowserTestUtils.addTab(
gBrowser,
"about:blank"
));
let browser = tab.linkedBrowser;
await promiseBrowserLoaded(browser);
// Load a new URI with a specific referrer.
let referrerInfo1 = new ReferrerInfo(
Ci.nsIReferrerInfo.EMPTY,
true,
Services.io.newURI(REFERRER1)
);
browser.loadURI(Services.io.newURI("http://example.org"), {
referrerInfo: referrerInfo1,
triggeringPrincipal: Services.scriptSecurityManager.createNullPrincipal({}),
});
await promiseBrowserLoaded(browser);
await TabStateFlusher.flush(browser);
let tabState = JSON.parse(ss.getTabState(tab));
let actualReferrerInfo = E10SUtils.deserializeReferrerInfo(
tabState.entries[0].referrerInfo
);
is(
actualReferrerInfo.originalReferrer.spec,
REFERRER1,
"Referrer retrieved via getTabState matches referrer set via loadURI."
);
let referrerInfo2 = new ReferrerInfo(
Ci.nsIReferrerInfo.EMPTY,
true,
Services.io.newURI(REFERRER2)
);
tabState.entries[0].referrerInfo =
E10SUtils.serializeReferrerInfo(referrerInfo2);
await promiseTabState(tab, tabState);
await checkDocumentReferrer(
getExpectedReferrer(REFERRER2),
"document.referrer matches referrer set via setTabState using referrerInfo."
);
gBrowser.removeCurrentTab();
// Restore the closed tab.
tab = ss.undoCloseTab(window, 0);
await promiseTabRestored(tab);
await checkDocumentReferrer(
getExpectedReferrer(REFERRER2),
"document.referrer is still correct after closing and reopening the tab."
);
tabState.entries[0].referrerInfo = null;
tabState.entries[0].referrer = REFERRER3;
await promiseTabState(tab, tabState);
await checkDocumentReferrer(
getExpectedReferrer(REFERRER3),
"document.referrer matches referrer set via setTabState using referrer."
);
gBrowser.removeCurrentTab();
// Restore the closed tab.
tab = ss.undoCloseTab(window, 0);
await promiseTabRestored(tab);
await checkDocumentReferrer(
getExpectedReferrer(REFERRER3),
"document.referrer is still correct after closing and reopening the tab."
);
gBrowser.removeCurrentTab();
});
|