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
|
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
"use strict";
const { getMostRecentBrowserWindow } = require('sdk/window/utils');
const { PageMod } = require("sdk/page-mod");
const { getActiveTab, setTabURL, openTab, closeTab } = require('sdk/tabs/utils');
const xulApp = require('sdk/system/xul-app');
const windowHelpers = require('sdk/window/helpers');
const { defer } = require("sdk/core/promise");
const { isPrivate } = require('sdk/private-browsing');
const { isTabPBSupported, isWindowPBSupported } = require('sdk/private-browsing/utils');
const { cleanUI } = require('sdk/test/utils');
function openWebpage(url, enablePrivate) {
return new Promise((resolve, reject) => {
if (xulApp.is("Fennec")) {
let chromeWindow = getMostRecentBrowserWindow();
let rawTab = openTab(chromeWindow, url, {
isPrivate: enablePrivate
});
resolve(() => new Promise(resolve => {
closeTab(rawTab);
resolve();
}));
}
else {
windowHelpers.open("", {
features: {
toolbar: true,
private: enablePrivate
}
}).
then((chromeWindow) => {
if (isPrivate(chromeWindow) !== !!enablePrivate) {
reject(new Error("Window should have Private set to " + !!enablePrivate));
}
let tab = getActiveTab(chromeWindow);
setTabURL(tab, url);
resolve(() => windowHelpers.close(chromeWindow));
}).
catch(reject);
}
});
}
exports["test page-mod on private tab"] = function*(assert) {
// Only set private mode when explicitely supported.
// (fennec 19 has some intermediate PB support where isTabPBSupported
// will be false, but isPrivate(worker.tab) will be true if we open a private
// tab)
let setPrivate = isTabPBSupported || isWindowPBSupported;
let id = Date.now().toString(36);
let frameUri = "data:text/html;charset=utf-8," + id;
let uri = "data:text/html;charset=utf-8," +
encodeURIComponent(id + "<iframe src='" + frameUri + "'><iframe>");
let count = 0;
let close = yield openWebpage(uri, setPrivate);
yield new Promise(resolve => {
PageMod({
include: [uri, frameUri],
onAttach: function(worker) {
assert.ok(worker.tab.url == uri || worker.tab.url == frameUri,
"Got a worker attached to the private window tab");
if (setPrivate) {
assert.ok(isPrivate(worker), "The worker is really private");
assert.ok(isPrivate(worker.tab), "The document is really private");
}
else {
assert.ok(!isPrivate(worker),
"private browsing isn't supported, " +
"so that the worker isn't private");
assert.ok(!isPrivate(worker.tab),
"private browsing isn't supported, " +
"so that the document isn't private");
}
if (++count == 2) {
this.destroy();
resolve();
}
}
});
});
yield close();
yield cleanUI();
};
exports["test page-mod on non-private tab"] = function*(assert) {
let id = Date.now().toString(36);
let url = "data:text/html;charset=utf-8," + encodeURIComponent(id);
let close = yield openWebpage(url, false);
let mod;
let worker = yield new Promise(resolve => {
mod = PageMod({
include: url,
onAttach: resolve
});
});
assert.equal(worker.tab.url, url,
"Got a worker attached to the private window tab");
assert.ok(!isPrivate(worker), "The worker is really non-private");
assert.ok(!isPrivate(worker.tab), "The document is really non-private");
mod.destroy();
yield close();
yield cleanUI();
}
|
