devtools/client/netmonitor/test/browser_net_save_response_as.js

   1 /* Any copyright is dedicated to the Public Domain.
   2    http://creativecommons.org/publicdomain/zero/1.0/ */
   3
   4 "use strict";
   5
   6 var MockFilePicker = SpecialPowers.MockFilePicker;
   7 MockFilePicker.init(window.browsingContext);
   8
   9 /**
  10  * Tests if saving a response to a file works..
  11  */
  12
  13 add_task(async function () {
  14   const { tab, monitor } = await initNetMonitor(
  15     CONTENT_TYPE_WITHOUT_CACHE_URL,
  16     { requestCount: 1 }
  17   );
  18   info("Starting test... ");
  19
  20   const { document } = monitor.panelWin;
  21
  22   // Execute requests.
  23   await performRequests(monitor, tab, CONTENT_TYPE_WITHOUT_CACHE_REQUESTS);
  24
  25   // Create the folder the gzip file will be saved into
  26   const destDir = createTemporarySaveDirectory();
  27   let destFile;
  28
  29   MockFilePicker.displayDirectory = destDir;
  30   const saveDialogClosedPromise = new Promise(resolve => {
  31     MockFilePicker.showCallback = function (fp) {
  32       info("MockFilePicker showCallback");
  33       const fileName = fp.defaultString;
  34       destFile = destDir.clone();
  35       destFile.append(fileName);
  36       MockFilePicker.setFiles([destFile]);
  37
  38       resolve(destFile.path);
  39     };
  40   });
  41
  42   registerCleanupFunction(function () {
  43     MockFilePicker.cleanup();
  44     destDir.remove(true);
  45   });
  46
  47   // Select gzip request.
  48
  49   info("Open the context menu");
  50
  51   EventUtils.sendMouseEvent(
  52     { type: "mousedown" },
  53     document.querySelectorAll(".request-list-item")[6]
  54   );
  55
  56   EventUtils.sendMouseEvent(
  57     { type: "contextmenu" },
  58     document.querySelectorAll(".request-list-item")[6]
  59   );
  60
  61   info("Open the save dialog");
  62   await selectContextMenuItem(monitor, "request-list-context-save-response-as");
  63
  64   info("Wait for the save dialog to close");
  65   const savedPath = await saveDialogClosedPromise;
  66
  67   const expectedFile = destDir.clone();
  68   expectedFile.append("sjs_content-type-test-server.sjs");
  69
  70   is(savedPath, expectedFile.path, "Response was saved to correct path");
  71
  72   info("Wait for the downloaded file to be fully saved to disk: " + savedPath);
  73   await TestUtils.waitForCondition(async () => {
  74     if (!(await IOUtils.exists(savedPath))) {
  75       return false;
  76     }
  77     const { size } = await IOUtils.stat(savedPath);
  78     return size > 0;
  79   });
  80
  81   const buffer = await IOUtils.read(savedPath);
  82   const savedFileContent = new TextDecoder().decode(buffer);
  83
  84   // The content is set by https://searchfox.org/mozilla-central/source/devtools/client/netmonitor/test/sjs_content-type-test-server.sjs#360
  85   // (the "gzip" case)
  86   is(
  87     savedFileContent,
  88     new Array(1000).join("Hello gzip!"),
  89     "Saved response has the correct text"
  90   );
  91
  92   await teardown(monitor);
  93 });
  94
  95 function createTemporarySaveDirectory() {
  96   const saveDir = Services.dirsvc.get("TmpD", Ci.nsIFile);
  97   saveDir.append("testsavedir");
  98   if (!saveDir.exists()) {
  99     saveDir.create(Ci.nsIFile.DIRECTORY_TYPE, 0o755);
 100   }
 101   return saveDir;
 102 }