1 files changed, 79 insertions, 0 deletions
diff --git a/toolkit/content/tests/browser/browser_save_folder_standalone_image.js b/toolkit/content/tests/browser/browser_save_folder_standalone_image.js
new file mode 100644
index 0000000000..ce45d04fdc
--- /dev/null
+++ b/toolkit/content/tests/browser/browser_save_folder_standalone_image.js
@@ -0,0 +1,79 @@
+/* Any copyright is dedicated to the Public Domain.
+ * https://creativecommons.org/publicdomain/zero/1.0/ */
+
+/**
+ * TestCase for bug 1726801
+ * <https://bugzilla.mozilla.org/show_bug.cgi?id=1726801>
+ *
+ * Load an image in a standalone tab and verify that the per-site download
+ * folder is correctly retrieved when using "Save Page As" to save the image.
+ */
+
+/*
+ * ================
+ * Helper functions
+ * ================
+ */
+
+async function setFile(downloadLastDir, aURI, aValue) {
+  downloadLastDir.setFile(aURI, aValue);
+  await TestUtils.waitForTick();
+}
+
+function newDirectory() {
+  let dir = FileUtils.getDir("TmpD", ["testdir"]);
+  dir.createUnique(Ci.nsIFile.DIRECTORY_TYPE, FileUtils.PERMS_DIRECTORY);
+  return dir;
+}
+
+function clearHistory() {
+  Services.obs.notifyObservers(null, "browser:purge-session-history");
+}
+
+async function clearHistoryAndWait() {
+  clearHistory();
+  await TestUtils.waitForTick();
+  await TestUtils.waitForTick();
+}
+
+/*
+ * ====
+ * Test
+ * ====
+ */
+
+let MockFilePicker = SpecialPowers.MockFilePicker;
+MockFilePicker.init(window);
+
+add_task(async function () {
+  const IMAGE_URL =
+    "http://mochi.test:8888/browser/toolkit/content/tests/browser/doggy.png";
+
+  await BrowserTestUtils.withNewTab(IMAGE_URL, async function (browser) {
+    let tmpDir = FileUtils.getDir("TmpD", []);
+    let dir = newDirectory();
+    let downloadLastDir = new DownloadLastDir(null);
+    // Set the desired target directory for the IMAGE_URL
+    await setFile(downloadLastDir, IMAGE_URL, dir);
+    // Ensure that "browser.download.lastDir" points to a different directory
+    await setFile(downloadLastDir, null, tmpDir);
+    registerCleanupFunction(async function () {
+      await clearHistoryAndWait();
+      dir.remove(true);
+    });
+
+    // Prepare mock file picker.
+    let showFilePickerPromise = new Promise(resolve => {
+      MockFilePicker.showCallback = fp => resolve(fp.displayDirectory.path);
+    });
+    registerCleanupFunction(function () {
+      MockFilePicker.cleanup();
+    });
+
+    // Run "Save Page As"
+    EventUtils.synthesizeKey("s", { accelKey: true });
+
+    let dirPath = await showFilePickerPromise;
+    is(dirPath, dir.path, "Verify proposed download folder.");
+  });
+});