Bug 1273251: Part 6 - Test that extension window wrappers are nuked on close. r=billm

MozReview-Commit-ID: 3sWNuLqZBzp

--HG--
extra : rebase_source : f094357f5bf5b0de0dd1da22c9afeed233f19bb3

js/xpconnect/tests/unit/test_nuke_webextension_wrappers.js

@@ -0,0 +1,75 @@
+// See https://bugzilla.mozilla.org/show_bug.cgi?id=1273251
+
+const {classes: Cc, interfaces: Ci, utils: Cu, results: Cr} = Components;
+
+Cu.import("resource://gre/modules/Services.jsm");
+Cu.import("resource://gre/modules/NetUtil.jsm");
+Cu.import("resource://gre/modules/Timer.jsm");
+
+function promiseEvent(target, event) {
+  return new Promise(resolve => {
+    target.addEventListener(event, resolve, {capture: true, once: true});
+  });
+}
+
+let aps = Cc["@mozilla.org/addons/policy-service;1"].getService(Ci.nsIAddonPolicyService).wrappedJSObject;
+
+let oldAddonIdCallback = aps.setExtensionURIToAddonIdCallback(uri => uri.host);
+do_register_cleanup(() => {
+  aps.setExtensionURIToAddonIdCallback(oldAddonIdCallback);
+});
+
+function getWindowlessBrowser(url) {
+  let ssm = Services.scriptSecurityManager;
+
+  let uri = NetUtil.newURI(url);
+  // TODO: Remove when addonId origin attribute is removed.
+  let attrs = {addonId: uri.host};
+
+  let principal = ssm.createCodebasePrincipal(uri, attrs);
+
+  let webnav = Services.appShell.createWindowlessBrowser(false);
+
+  let docShell = webnav.QueryInterface(Ci.nsIInterfaceRequestor)
+                       .getInterface(Ci.nsIDocShell);
+
+  docShell.createAboutBlankContentViewer(principal);
+
+  return webnav;
+}
+
+add_task(function*() {
+  let ssm = Services.scriptSecurityManager;
+
+  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 a tick.
+  yield new Promise(resolve => setTimeout(resolve, 0));
+
+  // 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();
+});

js/xpconnect/tests/unit/xpcshell.ini

@@ -98,6 +98,7 @@ fail-if = os == "android"
 [test_allowedDomainsXHR.js]
 [test_nuke_sandbox.js]
 [test_nuke_sandbox_event_listeners.js]
+[test_nuke_webextension_wrappers.js]
 [test_sandbox_metadata.js]
 [test_exportFunction.js]
 [test_promise.js]