Source code

Revision control

Copy as Markdown

Other Tools

Test Info:

/* Any copyright is dedicated to the Public Domain.
"use strict";
const { PromptTestUtils } = ChromeUtils.importESModule(
);
let { HttpServer } = ChromeUtils.importESModule(
);
const TEST_ROOT = getRootDirectory(gTestPath).replace(
);
async function checkAlert(
pageToLoad,
expectedTitle,
) {
function openFn(browser) {
return SpecialPowers.spawn(browser, [], () => {
if (content.document.nodePrincipal.isSystemPrincipal) {
// Can't eval in privileged contexts due to CSP, just call directly:
content.alert("Test");
} else {
// Eval everywhere else so it gets the principal of the loaded page.
content.eval("alert('Test')");
}
});
}
return checkDialog(pageToLoad, openFn, expectedTitle, expectedIcon);
}
async function checkBeforeunload(
pageToLoad,
expectedTitle,
) {
async function openFn(browser) {
let tab = gBrowser.getTabForBrowser(browser);
await BrowserTestUtils.synthesizeMouseAtCenter(
"body",
{},
browser.browsingContext
);
return gBrowser.removeTab(tab); // trigger beforeunload.
}
return checkDialog(pageToLoad, openFn, expectedTitle, expectedIcon);
}
async function checkDialog(
pageToLoad,
openFn,
expectedTitle,
expectedIcon,
modalType = Ci.nsIPrompt.MODAL_TYPE_CONTENT
) {
return BrowserTestUtils.withNewTab(pageToLoad, async browser => {
let promptPromise = PromptTestUtils.waitForPrompt(browser, {
modalType,
});
let spawnPromise = openFn(browser);
let dialog = await promptPromise;
let doc = dialog.ui.prompt.document;
let titleEl = doc.getElementById("titleText");
if (expectedTitle.value) {
is(titleEl.textContent, expectedTitle.value, "Title should match.");
} else {
is(
titleEl.dataset.l10nId,
expectedTitle.l10nId,
"Title l10n id should match."
);
}
ok(
!titleEl.parentNode.hasAttribute("overflown"),
"Title should fit without overflowing."
);
ok(BrowserTestUtils.isVisible(titleEl), "New title should be shown.");
ok(
BrowserTestUtils.isHidden(doc.getElementById("infoTitle")),
"Old title should be hidden."
);
let iconCS = doc.ownerGlobal.getComputedStyle(
doc.querySelector(".titleIcon")
);
is(
iconCS.backgroundImage,
`url("${expectedIcon}")`,
"Icon is as expected."
);
// This is not particularly neat, but we want to also test overflow
// Our test systems don't have hosts that long, so just fake it:
if (browser.currentURI.asciiHost == "example.com") {
let longerDomain = "extravagantly.long.".repeat(10) + "example.com";
doc.documentElement.setAttribute(
"headertitle",
JSON.stringify({ raw: longerDomain, shouldUseMaskFade: true })
);
info("Wait for the prompt title to update.");
await BrowserTestUtils.waitForMutationCondition(
titleEl,
{ characterData: true, attributes: true },
() =>
titleEl.textContent == longerDomain &&
titleEl.parentNode.hasAttribute("overflown")
);
is(titleEl.textContent, longerDomain, "The longer domain is reflected.");
ok(
titleEl.parentNode.hasAttribute("overflown"),
"The domain should overflow."
);
}
// Close the prompt again.
await PromptTestUtils.handlePrompt(dialog);
// The alert in the content process was sync, we need to make sure it gets
// cleaned up, but couldn't await it above because that'd hang the test!
await spawnPromise;
});
}
add_task(async function test_check_prompt_origin_display() {
await checkAlert("https://example.com/", { value: "example.com" });
// eslint-disable-next-line @microsoft/sdl/no-insecure-url
await checkAlert("http://example.com/", { value: "example.com" });
await checkAlert("data:text/html,<body>", {
l10nId: "common-dialog-title-null",
});
let homeDir = Services.dirsvc.get("Home", Ci.nsIFile);
let fileURI = Services.io.newFileURI(homeDir).spec;
await checkAlert(fileURI, { value: "file://" });
await checkAlert(
"about:config",
{ l10nId: "common-dialog-title-system" },
);
await checkBeforeunload(TEST_ROOT + "file_beforeunload_stop.html", {
value: "example.com",
});
});
add_task(async function test_check_auth() {
let server = new HttpServer();
registerCleanupFunction(() => {
return new Promise(resolve => {
server.stop(() => {
server = null;
resolve();
});
});
});
function forbiddenHandler(meta, res) {
res.setStatusLine(meta.httpVersion, 401, "Unauthorized");
res.setHeader("WWW-Authenticate", 'Basic realm="Realm"');
}
function pageHandler(meta, res) {
res.setStatusLine(meta.httpVersion, 200, "OK");
res.setHeader("Content-Type", "text/html");
let body = "<html><body></body></html>";
res.bodyOutputStream.write(body, body.length);
}
server.registerPathHandler("/forbidden", forbiddenHandler);
server.registerPathHandler("/page", pageHandler);
server.start(-1);
const HOST = `localhost:${server.identity.primaryPort}`;
// eslint-disable-next-line @microsoft/sdl/no-insecure-url
const AUTH_URI = `http://${HOST}/forbidden`;
// Try a simple load:
await checkDialog(
browser => BrowserTestUtils.startLoadingURIString(browser, AUTH_URI),
HOST,
Ci.nsIPrompt.MODAL_TYPE_TAB
);
let subframeLoad = function (browser) {
return SpecialPowers.spawn(browser, [AUTH_URI], uri => {
let f = content.document.createElement("iframe");
f.src = uri;
content.document.body.appendChild(f);
});
};
// Try x-origin subframe:
await checkDialog(
// eslint-disable-next-line @microsoft/sdl/no-insecure-url
subframeLoad,
HOST,
/* Because this is x-origin, we expect a different icon: */
Ci.nsIPrompt.MODAL_TYPE_TAB
);
});