docshell/test/unit/AllowJavascriptChild.jsm

   1 "use strict";
   2 var EXPORTED_SYMBOLS = ["AllowJavascriptChild"];
   3
   4 class AllowJavascriptChild extends JSWindowActorChild {
   5   async receiveMessage(msg) {
   6     switch (msg.name) {
   7       case "CheckScriptsAllowed":
   8         return this.checkScriptsAllowed();
   9       case "CheckFiredLoadEvent":
  10         return this.contentWindow.wrappedJSObject.gFiredOnload;
  11       case "CreateIframe":
  12         return this.createIframe(msg.data.url);
  13     }
  14     return null;
  15   }
  16
  17   handleEvent(event) {
  18     if (event.type === "load") {
  19       this.sendAsyncMessage("LoadFired");
  20     }
  21   }
  22
  23   checkScriptsAllowed() {
  24     let win = this.contentWindow;
  25
  26     win.wrappedJSObject.gFiredOnclick = false;
  27     win.document.body.click();
  28     return win.wrappedJSObject.gFiredOnclick;
  29   }
  30
  31   async createIframe(url) {
  32     let doc = this.contentWindow.document;
  33
  34     let iframe = doc.createElement("iframe");
  35     iframe.src = url;
  36     doc.body.appendChild(iframe);
  37
  38     await new Promise(resolve => {
  39       iframe.addEventListener("load", resolve, { once: true });
  40     });
  41
  42     return iframe.browsingContext;
  43   }
  44 }