Magic-Mirror/gecko-dev
1
0
Fork 0
mirror of https://github.com/mozilla/gecko-dev.git synced 2024-12-04 11:26:09 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Bug 1671119 - tests, r=emalysz

Browse Source 
Differential Revision: https://phabricator.services.mozilla.com/D92606
This commit is contained in:
Gijs Kruitbosch 2020-11-18 17:18:08 +00:00
parent dd46ef8cfe
commit 19ce44f703
3 changed files with 98 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
browser/extensions/screenshots/test/browser/browser.ini
View File

@ -1,2 +1,4 @@
[browser_screenshots_injection.js]
support-files = injection-page.html
[browser_screenshots_ui_check.js]
skip-if = os == 'win' && debug # Bug 1394967

72
browser/extensions/screenshots/test/browser/browser_screenshots_injection.js Normal file
View File

@ -0,0 +1,72 @@
/* Any copyright is dedicated to the Public Domain.
http://creativecommons.org/publicdomain/zero/1.0/ */
"use strict";
const TEST_PATH = getRootDirectory(gTestPath).replace(
"chrome://mochitests/content",
"https://example.com"
);
/**
* Check that web content cannot break into screenshots.
*/
add_task(async function test_inject_srcdoc() {
// If Screenshots was disabled, enable it just for this test.
const addon = await AddonManager.getAddonByID("screenshots@mozilla.org");
const isEnabled = addon.enabled;
if (!isEnabled) {
await addon.enable({allowSystemAddons: true});
registerCleanupFunction(async () => {
await addon.disable({allowSystemAddons: true});
});
}
await BrowserTestUtils.withNewTab(TEST_PATH + "injection-page.html",
async (browser) => {
// Set up the content hijacking. Do this so we can see it without
// awaiting - the promise should never resolve.
let response = null;
let responsePromise = SpecialPowers.spawn(browser, [], () => {
return new Promise(resolve => {
// We can't pass `resolve` directly because of sandboxing.
// `responseHandler` gets invoked from the content page.
content.wrappedJSObject.responseHandler = Cu.exportFunction(
function(arg) {
resolve(arg)
},
content
);
});
}).then(
r => {
ok(false, "Should not have gotten HTML but got: " + r);
response = r;
},
() => {
// Do nothing - we expect this to error when the test finishes
// and the actor is destroyed, while the promise still hasn't
// been resolved. We need to catch it in order not to throw
// uncaught rejection errors and inadvertently fail the test.
}
);
let error;
let errorPromise = new Promise(resolve => {
SpecialPowers.registerConsoleListener(msg => {
if (msg.message?.match(/iframe URL does not match expected blank.html/)) {
error = msg;
resolve();
}
});
});
// Now try to start the screenshot flow:
document.querySelector("keyset[id*=screenshot] > key").doCommand();
await Promise.race([errorPromise, responsePromise]);
ok(error, "Should get the relevant error: " + error?.message);
ok(!response, "Should not get a response from the webpage.");
SpecialPowers.postConsoleSentinel();
});
});

24
browser/extensions/screenshots/test/browser/injection-page.html Normal file
View File

@ -0,0 +1,24 @@
<body>
<script>
let callback = function(mutationsList, observer) {
for (let mutation of mutationsList) {
let [added] = mutation.addedNodes;
if (added instanceof HTMLIFrameElement && added.id == "firefox-screenshots-preview-iframe") {
added.srcdoc = "<html></html>";
// Now we have to wait for the doc to be populated.
let interval = setInterval(() => {
console.log(added.contentDocument.innerHTML);
if (added.contentDocument.body.innerHTML) {
clearInterval(interval);
window.responseHandler(added.contentDocument.body.innerHTML);
}
}, 100);
observer.disconnect();
}
}
};
observer = new MutationObserver(callback);
observer.observe(document.body, {childList: true});
</script>
</body>