diff options
Diffstat (limited to 'browser/base/content/test/general/browser_save_link_when_window_navigates.js')
| -rw-r--r-- | browser/base/content/test/general/browser_save_link_when_window_navigates.js | 197 |
1 files changed, 197 insertions, 0 deletions
diff --git a/browser/base/content/test/general/browser_save_link_when_window_navigates.js b/browser/base/content/test/general/browser_save_link_when_window_navigates.js new file mode 100644 index 0000000000..49901e8bfa --- /dev/null +++ b/browser/base/content/test/general/browser_save_link_when_window_navigates.js @@ -0,0 +1,197 @@ +/* Any copyright is dedicated to the Public Domain. + http://creativecommons.org/publicdomain/zero/1.0/ */ + +var MockFilePicker = SpecialPowers.MockFilePicker; +MockFilePicker.init(window); + +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.loadURIString(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(dialog) { + 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, aData) { + if (aTopic != "domwindowopened") { + return; + } + + let win = aSubject; + + win.addEventListener( + "load", + function (event) { + 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); + }); + }); + }); +} |