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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
|
<!doctype html>
<head>
<title>Test content script runtime.getFrameId</title>
<script src="/tests/SimpleTest/SimpleTest.js"></script>
<script src="/tests/SimpleTest/ExtensionTestUtils.js"></script>
<script src="head.js"></script>
<link rel="stylesheet" href="/tests/SimpleTest/test.css"/>
</head>
<script>
"use strict";
add_task(async function test_runtime_getFrameId_invalid() {
let extension = ExtensionTestUtils.loadExtension({
async background() {
let proxy = new Proxy(window, {});
let proto = Object.create(window);
class FakeFrame extends HTMLIFrameElement {
constructor() {
super();
console.log("FakeFrame ctor"); // eslint-disable-line
}
}
customElements.define('fake-frame', FakeFrame, { extends: 'iframe' });
let custom = document.createElement("fake-frame");
let invalid = [null, 13, "blah", document.body, proxy, proto, custom];
for (let value of invalid) {
browser.test.assertThrows(
() => browser.runtime.getFrameId(value),
/Invalid argument/,
"Correct exception thrown."
);
}
let detached = document.createElement("iframe");
let id = browser.runtime.getFrameId(detached);
browser.test.assertEq(id, -1, "Detached iframe has no frameId.");
browser.test.sendMessage("done");
},
});
await extension.startup();
await extension.awaitMessage("done");
await extension.unload();
});
add_task(async function test_contentscript_runtime_getFrameId() {
let extension = ExtensionTestUtils.loadExtension({
manifest: {
permissions: ["webNavigation", "tabs"],
host_permissions: ["https://example.org/"],
},
files: {
"cs.js"() {
browser.test.log(`Content script loaded on: ${location.href}`);
let parents = {};
// Recursivelly walk descendant frames and get parent frameIds.
function visit(win) {
let frameId = browser.runtime.getFrameId(win);
let parentId = browser.runtime.getFrameId(win.parent);
parents[frameId] = (win.parent != win) ? parentId : -1;
try {
let frameEl = browser.runtime.getFrameId(win.frameElement);
browser.test.assertEq(frameId, frameEl, "frameElement id correct");
} catch (e) {
// Can't access a cross-origin .frameElement.
}
for (let i = 0; i < win.frames.length; i++) {
visit(win.frames[i]);
}
}
visit(window);
// Add the <embed> frame if it exists.
let embed = document.querySelector("embed");
if (embed) {
let id = browser.runtime.getFrameId(embed);
parents[id] = 0;
}
// Add the <object> frame if it exists.
let object = document.querySelector("object");
if (object) {
let id = browser.runtime.getFrameId(object);
parents[id] = 0;
}
browser.test.log(`Parents tree: ${JSON.stringify(parents)}`);
return parents;
},
async "closedPopup.js"() {
let popup = window.open("https://example.org/?popup");
popup.close();
for (let i = 0; i < 100; i++) {
await new Promise(r => setTimeout(r, 50));
try {
popup.blur();
} catch(e) {
if (e.message === "can't access dead object") {
browser.test.assertThrows(
() => browser.runtime.getFrameId(popup),
/An exception was thrown/,
"Passing a dead object throws."
);
browser.test.sendMessage("done");
return;
}
}
}
browser.test.fail("Timed out while waiting for popup to close.");
},
"closedPopup.html": `
<!doctype html>
<meta charset="utf-8">
<script src="closedPopup.js"><\/script>
`,
},
async background() {
let base = "https://example.org/tests/toolkit/components/extensions/test/mochitest";
let files = {
"file_contains_iframe.html": 2,
"file_WebNavigation_page1.html": 2,
"file_with_xorigin_frame.html": 2,
// Contains all of the above.
"file_with_subframes_and_embed.html": 9,
};
for (let [file, count] of Object.entries(files)) {
let tab;
let completed = new Promise(resolve => {
browser.webNavigation.onCompleted.addListener(function cb(details) {
browser.test.log(`onCompleted: ${JSON.stringify(details)}`);
if (details.tabId === tab?.id && details.frameId === 0) {
browser.webNavigation.onCompleted.removeListener(cb);
resolve();
}
});
});
browser.test.log(`Load a test page: ${file}`);
tab = await browser.tabs.create({ url: `${base}/${file}` });
await completed;
let [parents] = await browser.tabs.executeScript(tab.id, {
file: "cs.js"
});
let all = await browser.webNavigation.getAllFrames({ tabId: tab.id });
browser.test.log(`getAllFrames: ${JSON.stringify(all)}`);
browser.test.assertEq(all.length, count, "All frames accounted for.");
browser.test.assertEq(
Object.keys(parents).length,
count,
"All frames accounted for from content script."
);
for (let frame of all) {
browser.test.assertEq(
frame.parentFrameId,
parents[frame.frameId],
"Correct frame ancestor info."
);
}
await browser.tabs.remove(tab.id);
}
browser.tabs.create({ url: browser.runtime.getURL("closedPopup.html" )});
},
});
await extension.startup();
await extension.awaitMessage("done");
await extension.unload();
});
</script>
|
