The Tor Browser: browser/base/content/test/general/browser_save_private_link_perwindowpb.js@6474c204b198 (annotated)

browser/base/content/test/general/browser_save_private_link_perwindowpb.js@6474c204b198 (annotated)

browser/base/content/test/general/browser_save_private_link_perwindowpb.js

Wed, 31 Dec 2014 06:09:35 +0100

author: Michael Schloh von Bennewitz <michael@schloh.com>
date: Wed, 31 Dec 2014 06:09:35 +0100
changeset 0: 6474c204b198
permissions: -rw-r--r--

Cloned upstream origin tor-browser at tor-browser-31.3.0esr-4.5-1-build1
revision ID fc1c9ff7c1b2defdbc039f12214767608f46423f for hacking purpose.

 /* 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/. */
 let {LoadContextInfo} = Cu.import("resource://gre/modules/LoadContextInfo.jsm", null);
 function test() {
   // initialization
   waitForExplicitFinish();
   let windowsToClose = [];
   let testURI = "http://mochi.test:8888/browser/browser/base/content/test/general/bug792517.html";
   let fileName;
   let MockFilePicker = SpecialPowers.MockFilePicker;
   let cache = Cc["@mozilla.org/netwerk/cache-storage-service;1"]
               .getService(Ci.nsICacheStorageService);
   function checkDiskCacheFor(filename, goon) {
     Visitor.prototype = {
       onCacheStorageInfo: function(num, consumption)
       {
         info("disk storage contains " + num + " entries");
       },
       onCacheEntryInfo: function(entry)
       {
         info(entry.key);
         is(entry.key.contains(filename), false, "web content present in disk cache");
       },
       onCacheEntryVisitCompleted: function()
       {
         goon();
       }
     };
     function Visitor() {}
     var storage = cache.diskCacheStorage(LoadContextInfo.default, false);
     storage.asyncVisitStorage(new Visitor(), true /* Do walk entries */);
   }
   function onTransferComplete(downloadSuccess) {
     ok(downloadSuccess, "Image file should have been downloaded successfully");
     // Give the request a chance to finish and create a cache entry
     executeSoon(function() {
       checkDiskCacheFor(fileName, finish);
       mockTransferCallback = null;
     });
   }
   function createTemporarySaveDirectory() {
     var saveDir = Cc["@mozilla.org/file/directory_service;1"]
                     .getService(Ci.nsIProperties)
                     .get("TmpD", Ci.nsIFile);
     saveDir.append("testsavedir");
     if (!saveDir.exists())
       saveDir.create(Ci.nsIFile.DIRECTORY_TYPE, 0755);
     return saveDir;
   }
   function doTest(aIsPrivateMode, aWindow, aCallback) {
     function contextMenuOpened(event) {
       cache.clear();
       aWindow.document.removeEventListener("popupshown", contextMenuOpened);
       // Create the folder the image will be saved into.
       var destDir = createTemporarySaveDirectory();
       var destFile = destDir.clone();
       MockFilePicker.displayDirectory = destDir;
       MockFilePicker.showCallback = function(fp) {
         fileName = fp.defaultString;
         destFile.append (fileName);
         MockFilePicker.returnFiles = [destFile];
         MockFilePicker.filterIndex = 1; // kSaveAsType_URL
       };
       mockTransferCallback = onTransferComplete;
       mockTransferRegisterer.register();
       registerCleanupFunction(function () {
         mockTransferRegisterer.unregister();
         MockFilePicker.cleanup();
         destDir.remove(true);
       });
       // Select "Save Image As" option from context menu
       var saveVideoCommand = aWindow.document.getElementById("context-saveimage");
       saveVideoCommand.doCommand();
       event.target.hidePopup();
     }
     aWindow.gBrowser.addEventListener("pageshow", function pageShown(event) {
       // If data: -url PAC file isn't loaded soon enough, we may get about:privatebrowsing loaded
       if (event.target.location == "about:blank" ||
           event.target.location == "about:privatebrowsing") {
         aWindow.gBrowser.selectedBrowser.loadURI(testURI);
         return;
       }
       aWindow.gBrowser.removeEventListener("pageshow", pageShown);
       waitForFocus(function () {
         aWindow.document.addEventListener("popupshown", contextMenuOpened, false);
         var img = aWindow.gBrowser.selectedBrowser.contentDocument.getElementById("img");
         EventUtils.synthesizeMouseAtCenter(img,
                                            { type: "contextmenu", button: 2 },
                                            aWindow.gBrowser.contentWindow);
       }, aWindow.gBrowser.selectedBrowser.contentWindow);
     });
   }
   function testOnWindow(aOptions, aCallback) {
     whenNewWindowLoaded(aOptions, function(aWin) {
       windowsToClose.push(aWin);
       // execute should only be called when need, like when you are opening
       // web pages on the test. If calling executeSoon() is not necesary, then
       // call whenNewWindowLoaded() instead of testOnWindow() on your test.
       executeSoon(function() aCallback(aWin));
     });
   };
    // this function is called after calling finish() on the test.
   registerCleanupFunction(function() {
     windowsToClose.forEach(function(aWin) {
       aWin.close();
     });
   });
   MockFilePicker.init(window);
   // then test when on private mode
   testOnWindow({private: true}, function(aWin) {
     doTest(true, aWin, finish);
   });
 }
 Cc["@mozilla.org/moz/jssubscript-loader;1"]
   .getService(Ci.mozIJSSubScriptLoader)
   .loadSubScript("chrome://mochitests/content/browser/toolkit/content/tests/browser/common/mockTransfer.js",
                  this);