1 files changed, 98 insertions, 0 deletions
diff --git a/toolkit/components/downloads/test/unit/test_DownloadLegacy.js b/toolkit/components/downloads/test/unit/test_DownloadLegacy.js
new file mode 100644
index 0000000000..55d5884d6e
--- /dev/null
+++ b/toolkit/components/downloads/test/unit/test_DownloadLegacy.js
@@ -0,0 +1,98 @@
+/* -*- indent-tabs-mode: nil; js-indent-level: 2 -*- */
+/* vim: set ts=2 et sw=2 tw=80: */
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+/**
+ * Tests the integration with legacy interfaces for downloads.
+ */
+
+"use strict";
+
+// Execution of common tests
+
+var gUseLegacySaver = true;
+
+var scriptFile = do_get_file("common_test_Download.js");
+Services.scriptloader.loadSubScript(NetUtil.newURI(scriptFile).spec);
+
+/**
+ * Checks the referrer for restart downloads.
+ * If the legacy download is stopped and restarted, the saving method
+ * is changed from DownloadLegacySaver to the DownloadCopySaver.
+ * The referrer header should be passed correctly.
+ */
+add_task(async function test_referrer_restart() {
+  let sourcePath = "/test_referrer_restart.txt";
+  let sourceUrl = httpUrl("test_referrer_restart.txt");
+
+  function cleanup() {
+    gHttpServer.registerPathHandler(sourcePath, null);
+  }
+  registerCleanupFunction(cleanup);
+
+  registerInterruptibleHandler(
+    sourcePath,
+    function firstPart(aRequest, aResponse) {
+      aResponse.setHeader("Content-Type", "text/plain", false);
+      Assert.ok(aRequest.hasHeader("Referer"));
+      Assert.equal(aRequest.getHeader("Referer"), TEST_REFERRER_URL);
+
+      aResponse.setHeader(
+        "Content-Length",
+        "" + TEST_DATA_SHORT.length * 2,
+        false
+      );
+      aResponse.write(TEST_DATA_SHORT);
+    },
+    function secondPart(aRequest, aResponse) {
+      Assert.ok(aRequest.hasHeader("Referer"));
+      Assert.equal(aRequest.getHeader("Referer"), TEST_REFERRER_URL);
+
+      aResponse.write(TEST_DATA_SHORT);
+    }
+  );
+
+  let referrerInfo = new ReferrerInfo(
+    Ci.nsIReferrerInfo.UNSAFE_URL,
+    true,
+    NetUtil.newURI(TEST_REFERRER_URL)
+  );
+
+  async function restart_and_check_referrer(download) {
+    let promiseSucceeded = download.whenSucceeded();
+
+    // Cancel the first download attempt.
+    await promiseDownloadMidway(download);
+    await download.cancel();
+
+    // The second request is allowed to complete.
+    continueResponses();
+    download.start().catch(() => {});
+
+    // Wait for the download to finish by waiting on the whenSucceeded promise.
+    await promiseSucceeded;
+
+    Assert.ok(download.stopped);
+    Assert.ok(download.succeeded);
+    Assert.ok(!download.canceled);
+
+    checkEqualReferrerInfos(download.source.referrerInfo, referrerInfo);
+  }
+
+  mustInterruptResponses();
+
+  let download = await promiseStartLegacyDownload(sourceUrl, {
+    referrerInfo,
+  });
+  await restart_and_check_referrer(download);
+
+  mustInterruptResponses();
+  download = await promiseStartLegacyDownload(sourceUrl, {
+    referrerInfo,
+    isPrivate: true,
+  });
+  await restart_and_check_referrer(download);
+
+  cleanup();
+});