mozilla-central/browser/components/contextualidentity/test/browser/browser_guessusercontext.js (file symbol)

Source code

Revision control

Copy as Markdown

Other Tools

Test Info:

/* Any copyright is dedicated to the Public Domain.
"use strict";
const DEFAULT = Ci.nsIScriptSecurityManager.DEFAULT_USER_CONTEXT_ID;
const PERSONAL = 1;
const WORK = 2;
const HOST_MOCHI = makeURI(
);
const HOST_EXAMPLE = makeURI(
);
const {
URILoadingHelper: { guessUserContextId },
} = ChromeUtils.importESModule("resource:///modules/URILoadingHelper.sys.mjs");
async function openTabInUserContext(uri, userContextId, win = window) {
let { gBrowser } = win;
let tab = BrowserTestUtils.addTab(gBrowser, uri, { userContextId });
gBrowser.selectedTab = tab;
tab.ownerGlobal.focus();
await BrowserTestUtils.browserLoaded(gBrowser.getBrowserForTab(tab));
return tab;
}
registerCleanupFunction(async function cleanup() {
Services.prefs.clearUserPref(
"browser.link.force_default_user_context_id_for_external_opens"
);
while (gBrowser.tabs.length > 1) {
gBrowser.removeTab(gBrowser.selectedTab, { animate: false });
}
});
add_setup(async function () {
await SpecialPowers.pushPrefEnv({
set: [["privacy.userContext.enabled", true]],
});
});
add_task(async function test() {
is(guessUserContextId(null), null, "invalid uri - null");
is(guessUserContextId(HOST_EXAMPLE), null, "no tabs - null");
await openTabInUserContext(HOST_EXAMPLE.spec, PERSONAL);
is(guessUserContextId(HOST_EXAMPLE), PERSONAL, "one tab - matches container");
is(guessUserContextId(HOST_MOCHI), null, "one tab - doesn't match container");
await openTabInUserContext(HOST_MOCHI.spec, PERSONAL);
is(guessUserContextId(HOST_MOCHI), PERSONAL, "one tab - matches container");
await openTabInUserContext(HOST_MOCHI.spec);
await openTabInUserContext(HOST_MOCHI.spec);
is(
guessUserContextId(HOST_MOCHI),
DEFAULT,
"can guess guess default container"
);
await openTabInUserContext(HOST_EXAMPLE.spec, WORK);
is(guessUserContextId(HOST_EXAMPLE), PERSONAL, "same number - use first");
await openTabInUserContext(HOST_EXAMPLE.spec, WORK);
is(guessUserContextId(HOST_EXAMPLE), WORK, "multiple per host - max");
let win = await BrowserTestUtils.openNewBrowserWindow();
await openTabInUserContext(HOST_EXAMPLE.spec, PERSONAL, win);
await openTabInUserContext(HOST_EXAMPLE.spec, PERSONAL, win);
is(guessUserContextId(HOST_EXAMPLE), PERSONAL, "count across windows");
await BrowserTestUtils.closeWindow(win);
is(guessUserContextId(HOST_EXAMPLE), WORK, "forgets closed window");
// Check the opener flow more directly
openURIFromExternal(HOST_EXAMPLE.spec + "?new");
is(
gBrowser.selectedTab.getAttribute("usercontextid"),
WORK.toString(),
"opener flow"
);
is(guessUserContextId(HOST_EXAMPLE), WORK, "still the most common");
is(
guessUserContextId(HOST_MOCHI),
DEFAULT,
"still matches default container"
);
// Force into default with the pref from https://bugzilla.mozilla.org/show_bug.cgi?id=1692124
Services.prefs.setBoolPref(
"browser.link.force_default_user_context_id_for_external_opens",
true
);
openURIFromExternal(HOST_EXAMPLE.spec + "?new");
is(
gBrowser.selectedTab.getAttribute("usercontextid"),
null,
"opener flow with default user context ID forced by pref"
);
});
async function openURIFromExternal(spec) {
let browsingContext = window.browserDOMWindow.openURI(
makeURI(spec),
null,
Ci.nsIBrowserDOMWindow.OPEN_NEWTAB,
Ci.nsIBrowserDOMWindow.OPEN_EXTERNAL,
Services.scriptSecurityManager.getSystemPrincipal()
);
await BrowserTestUtils.browserLoaded(browsingContext.embedderElement);
is(
browsingContext.embedderElement,
gBrowser.selectedBrowser,
"opener selected"
);
}