test_nuke_webextension_wrappers.js

Enable keyboard shortcuts

// See https://bugzilla.mozilla.org/show_bug.cgi?id=1273251

const {NetUtil} = ChromeUtils.importESModule("resource://gre/modules/NetUtil.sys.mjs");

const {TestUtils} = ChromeUtils.importESModule("resource://testing-common/TestUtils.sys.mjs");

function getWindowlessBrowser(url) {

  let ssm = Services.scriptSecurityManager;

  let uri = NetUtil.newURI(url);

  let principal = ssm.createContentPrincipal(uri, {});

  let webnav = Services.appShell.createWindowlessBrowser(false);

  let docShell = webnav.docShell;

  docShell.createAboutBlankDocumentViewer(principal, principal);

  return webnav;

function StubPolicy(id) {

  return new WebExtensionPolicy({

id,

    mozExtensionHostname: id,

    baseURL: `file:///{id}`,

    allowedOrigins: new MatchPatternSet([]),

    localizeCallback(string) {},

});

add_task(async function() {

  let policy = StubPolicy("foo");

  policy.active = true;

  let webnavA = getWindowlessBrowser("moz-extension://foo/a.html");

  let webnavB = getWindowlessBrowser("moz-extension://foo/b.html");

  let winA = Cu.waiveXrays(webnavA.document.defaultView);

  let winB = Cu.waiveXrays(webnavB.document.defaultView);

  winB.winA = winA;

  winB.eval(`winA.thing = {foo: "bar"};`);

  let getThing = winA.eval(String(() => {

    try {

      return thing.foo;

    } catch (e) {

      return String(e);

  }));

  // Check that the object can be accessed normally before windowB is closed.

  equal(getThing(), "bar");

  webnavB.close();

  // Wrappers are nuked asynchronously, so wait for that to happen.

  await TestUtils.topicObserved("inner-window-nuked");

  // Check that it can't be accessed after he window has been closed.

  let result = getThing();

  ok(/dead object/.test(result),

     `Result should show a dead wrapper error: ${result}`);

  webnavA.close();

  policy.active = false;

});