tor-browser

browser_save_link_when_window_navigates.js (6152B)

---

/* Any copyright is dedicated to the Public Domain.
  http://creativecommons.org/publicdomain/zero/1.0/ */

var MockFilePicker = SpecialPowers.MockFilePicker;
MockFilePicker.init(window.browsingContext);

const SAVE_PER_SITE_PREF = "browser.download.lastDir.savePerSite";
const ALWAYS_DOWNLOAD_DIR_PREF = "browser.download.useDownloadDir";
const ALWAYS_ASK_PREF = "browser.download.always_ask_before_handling_new_types";
const UCT_URI = "chrome://mozapps/content/downloads/unknownContentType.xhtml";

Services.scriptloader.loadSubScript(
 "chrome://mochitests/content/browser/toolkit/content/tests/browser/common/mockTransfer.js",
 this
);

function createTemporarySaveDirectory() {
 var saveDir = Services.dirsvc.get("TmpD", Ci.nsIFile);
 saveDir.append("testsavedir");
 if (!saveDir.exists()) {
   info("create testsavedir!");
   saveDir.create(Ci.nsIFile.DIRECTORY_TYPE, 0o755);
 }

 Services.prefs.setIntPref("browser.download.folderList", 2);
 Services.prefs.setCharPref("browser.download.dir", saveDir);
 info("return from createTempSaveDir: " + saveDir.path);
 return saveDir;
}

function triggerSave(aWindow, aCallback) {
 info(
   "started triggerSave, persite downloads: " +
     (Services.prefs.getBoolPref(SAVE_PER_SITE_PREF) ? "on" : "off")
 );
 var fileName;
 let testBrowser = aWindow.gBrowser.selectedBrowser;
 let testURI =
   "http://mochi.test:8888/browser/browser/base/content/test/general/navigating_window_with_download.html";

 // Only observe the UTC dialog if it's enabled by pref
 if (Services.prefs.getBoolPref(ALWAYS_ASK_PREF)) {
   windowObserver.setCallback(onUCTDialog);
 }

 BrowserTestUtils.startLoadingURIString(testBrowser, testURI);

 // Create the folder the link will be saved into.
 var destDir = createTemporarySaveDirectory();
 var destFile = destDir.clone();

 MockFilePicker.displayDirectory = destDir;
 MockFilePicker.showCallback = function (fp) {
   info("showCallback");
   fileName = fp.defaultString;
   info("fileName: " + fileName);
   destFile.append(fileName);
   MockFilePicker.setFiles([destFile]);
   MockFilePicker.filterIndex = 1; // kSaveAsType_URL
   info("done showCallback");
 };

 mockTransferCallback = function (downloadSuccess) {
   info("mockTransferCallback");
   onTransferComplete(aWindow, downloadSuccess, destDir);
   destDir.remove(true);
   ok(!destDir.exists(), "Destination dir should be removed");
   ok(!destFile.exists(), "Destination file should be removed");
   mockTransferCallback = null;
   info("done mockTransferCallback");
 };

 function onUCTDialog() {
   SpecialPowers.spawn(testBrowser, [], async () => {
     content.document.querySelector("iframe").remove();
   }).then(() => executeSoon(continueDownloading));
 }

 function continueDownloading() {
   for (let win of Services.wm.getEnumerator("")) {
     if (win.location && win.location.href == UCT_URI) {
       win.document
         .getElementById("unknownContentType")
         ._fireButtonEvent("accept");
       win.close();
       return;
     }
   }
   ok(false, "No Unknown Content Type dialog yet?");
 }

 function onTransferComplete(aWindow2, downloadSuccess) {
   ok(downloadSuccess, "Link should have been downloaded successfully");
   aWindow2.close();

   executeSoon(aCallback);
 }
}

var windowObserver = {
 setCallback(aCallback) {
   if (this._callback) {
     ok(false, "Should only be dealing with one callback at a time.");
   }
   this._callback = aCallback;
 },
 observe(aSubject, aTopic) {
   if (aTopic != "domwindowopened") {
     return;
   }

   let win = aSubject;

   win.addEventListener(
     "load",
     function () {
       if (win.location == UCT_URI) {
         SimpleTest.executeSoon(function () {
           if (windowObserver._callback) {
             windowObserver._callback(win);
             delete windowObserver._callback;
           } else {
             ok(false, "Unexpected UCT dialog!");
           }
         });
       }
     },
     { once: true }
   );
 },
};

Services.ww.registerNotification(windowObserver);

function test() {
 waitForExplicitFinish();
 Services.prefs.setBoolPref(ALWAYS_ASK_PREF, false);

 function testOnWindow(options, callback) {
   info("testOnWindow(" + options + ")");
   var win = OpenBrowserWindow(options);
   info("got " + win);
   whenDelayedStartupFinished(win, () => callback(win));
 }

 function whenDelayedStartupFinished(aWindow, aCallback) {
   info("whenDelayedStartupFinished");
   Services.obs.addObserver(function observer(aSubject, aTopic) {
     info(
       "whenDelayedStartupFinished, got topic: " +
         aTopic +
         ", got subject: " +
         aSubject +
         ", waiting for " +
         aWindow
     );
     if (aWindow == aSubject) {
       Services.obs.removeObserver(observer, aTopic);
       executeSoon(aCallback);
       info("whenDelayedStartupFinished found our window");
     }
   }, "browser-delayed-startup-finished");
 }

 mockTransferRegisterer.register();

 registerCleanupFunction(function () {
   info("Running the cleanup code");
   mockTransferRegisterer.unregister();
   MockFilePicker.cleanup();
   Services.ww.unregisterNotification(windowObserver);
   Services.prefs.clearUserPref(ALWAYS_DOWNLOAD_DIR_PREF);
   Services.prefs.clearUserPref(SAVE_PER_SITE_PREF);
   Services.prefs.clearUserPref(ALWAYS_ASK_PREF);
   Services.prefs.clearUserPref("browser.download.folderList");
   Services.prefs.clearUserPref("browser.download.dir");
   info("Finished running the cleanup code");
 });

 info(
   `Running test with ${ALWAYS_ASK_PREF} set to ${Services.prefs.getBoolPref(
     ALWAYS_ASK_PREF,
     false
   )}`
 );
 testOnWindow(undefined, function (win) {
   let windowGonePromise = BrowserTestUtils.domWindowClosed(win);
   Services.prefs.setBoolPref(SAVE_PER_SITE_PREF, true);
   triggerSave(win, async function () {
     await windowGonePromise;
     Services.prefs.setBoolPref(SAVE_PER_SITE_PREF, false);
     testOnWindow(undefined, function (win2) {
       triggerSave(win2, finish);
     });
   });
 });
}