The Tor Browser: docshell/test/navigation/NavigationUtils.js@b8a032363ba2 (annotated)

docshell/test/navigation/NavigationUtils.js@b8a032363ba2 (annotated)

docshell/test/navigation/NavigationUtils.js

Thu, 22 Jan 2015 13:21:57 +0100

author: Michael Schloh von Bennewitz <michael@schloh.com>
date: Thu, 22 Jan 2015 13:21:57 +0100
branch: TOR_BUG_9701
changeset 15: b8a032363ba2
permissions: -rw-r--r--

Incorporate requested changes from Mozilla in review:
https://bugzilla.mozilla.org/show_bug.cgi?id=1123480#c6

 /* This Source Code Form is subject to the terms of the Mozilla Public
  * License, v. 2.0. If a copy of the MPL was not distributed with this
  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
 ///////////////////////////////////////////////////////////////////////////
 //
 // Utilities for navigation tests
 //
 ///////////////////////////////////////////////////////////////////////////
 var body = "This frame was navigated.";
 var target_url = "data:text/html,<html><body>" + body + "</body></html>";
 var popup_body = "This is a popup";
 var target_popup_url = "data:text/html,<html><body>" + popup_body + "</body></html>";
 ///////////////////////////////////////////////////////////////////////////
 // Functions that navigate frames
 ///////////////////////////////////////////////////////////////////////////
 function navigateByLocation(wnd) {
   try {
     wnd.location = target_url;
   } catch(ex) {
     // We need to keep our finished frames count consistent.
     // Oddly, this ends up simulating the behavior of IE7.
     window.open(target_url, "_blank", "width=10,height=10");
   }
 }
 function navigateByOpen(name) {
   window.open(target_url, name, "width=10,height=10");
 }
 function navigateByForm(name) {
   var form = document.createElement("form");
   form.action = target_url;
   form.method = "POST";
   form.target = name; document.body.appendChild(form);
   form.submit();
 }
 var hyperlink_count = 0;
 function navigateByHyperlink(name) {
   var link = document.createElement("a");
   link.href = target_url;
   link.target = name;
   link.id = "navigation_hyperlink_" + hyperlink_count++;
   document.body.appendChild(link);
   sendMouseEvent({type:"click"}, link.id);
 }
 ///////////////////////////////////////////////////////////////////////////
 // Functions that call into Mochitest framework
 ///////////////////////////////////////////////////////////////////////////
 function isNavigated(wnd, message) {
   var result = null;
   try {
     result = SpecialPowers.wrap(wnd).document.body.innerHTML;
   } catch(ex) {
     result = ex;
   }
   is(result, body, message);
 }
 function isBlank(wnd, message) {
   var result = null;
   try {
     result = wnd.document.body.innerHTML;
   } catch(ex) {
     result = ex;
   }
   is(result, "This is a blank document.", message);
 }
 function isAccessible(wnd, message) {
   try {
     wnd.document.body.innerHTML;
     ok(true, message);
   } catch(ex) {
     ok(false, message);
   }
 }
 function isInaccessible(wnd, message) {
   try {
     wnd.document.body.innerHTML;
     ok(false, message);
   } catch(ex) {
     ok(true, message);
   }
 }
 ///////////////////////////////////////////////////////////////////////////
 // Functions that require UniversalXPConnect privilege
 ///////////////////////////////////////////////////////////////////////////
 function xpcEnumerateContentWindows(callback) {
   var Ci = SpecialPowers.Ci;
   var ww = SpecialPowers.Cc["@mozilla.org/embedcomp/window-watcher;1"]
                         .getService(Ci.nsIWindowWatcher);
   var enumerator = ww.getWindowEnumerator();
   var contentWindows = [];
   while (enumerator.hasMoreElements()) {
     var win = enumerator.getNext();
     if (/ChromeWindow/.exec(win)) {
       var docshellTreeNode = win.QueryInterface(Ci.nsIInterfaceRequestor)
                                 .getInterface(Ci.nsIWebNavigation)
                                 .QueryInterface(Ci.nsIDocShellTreeItem);
       var childCount = docshellTreeNode.childCount;
       for (var i = 0; i < childCount; ++i) {
         var childTreeNode = docshellTreeNode.getChildAt(i);
         // we're only interested in content docshells
         if (SpecialPowers.unwrap(childTreeNode.itemType) != Ci.nsIDocShellTreeItem.typeContent)
           continue;
         var webNav = childTreeNode.QueryInterface(Ci.nsIWebNavigation);
         contentWindows.push(webNav.document.defaultView);
       }
     } else {
       contentWindows.push(win);
     }
   }
   while (contentWindows.length > 0)
     callback(contentWindows.pop());
 }
 // Note: This only searches for top-level frames with this name.
 function xpcGetFramesByName(name) {
   var results = [];
   xpcEnumerateContentWindows(function(win) {
     if (win.name == name)
       results.push(win);
   });
   return results;
 }
 function xpcCleanupWindows() {
   xpcEnumerateContentWindows(function(win) {
     if (win.location && win.location.protocol == "data:")
       win.close();
   });
 }
 function xpcWaitForFinishedFrames(callback, numFrames) {
   var finishedFrameCount = 0;
   function frameFinished() {
     finishedFrameCount++;
     if (finishedFrameCount == numFrames) {
       clearInterval(frameWaitInterval);
       setTimeout(callback, 0);
       return;
     }
     if (finishedFrameCount > numFrames)
       throw "Too many frames loaded.";
   }
   var finishedWindows = [];
   function contains(obj, arr) {
     for (var i = 0; i < arr.length; i++) {
       if (obj === arr[i])
         return true;
     }
     return false;
   }
   function searchForFinishedFrames(win) {
     if ((escape(unescape(win.location)) == escape(target_url) ||
          escape(unescape(win.location)) == escape(target_popup_url)) &&
         win.document &&
         win.document.body &&
         (win.document.body.textContent == body ||
          win.document.body.textContent == popup_body) &&
         win.document.readyState == "complete") {
       var util = win.QueryInterface(SpecialPowers.Ci.nsIInterfaceRequestor)
                     .getInterface(SpecialPowers.Ci.nsIDOMWindowUtils);
       var windowId = util.outerWindowID;
       if (!contains(windowId, finishedWindows)) {
         finishedWindows.push(windowId);
         frameFinished();
       }
     }
     for (var i = 0; i < win.frames.length; i++)
       searchForFinishedFrames(win.frames[i]);
   }
   function poll() {
     try {
       // This only gives us UniversalXPConnect for the current stack frame
       // We're using setInterval, so the main page's privileges are still normal
       xpcEnumerateContentWindows(searchForFinishedFrames);
     } catch(ex) {
       // We might be accessing windows before they are fully constructed,
       // which can throw.  We'll find those frames on our next poll().
     }
   }
   var frameWaitInterval = setInterval(poll, 500);
 }