netwerk/test/unit/test_bug1279246.js

   1 "use strict";
   2
   3 const { HttpServer } = ChromeUtils.importESModule(
   4   "resource://testing-common/httpd.sys.mjs"
   5 );
   6
   7 var httpserver = new HttpServer();
   8 var pass = 0;
   9 var responseBody = [0x0b, 0x02, 0x80, 0x74, 0x65, 0x73, 0x74, 0x0a, 0x03];
  10 var responseLen = 5;
  11 var testUrl = "/test/brotli";
  12
  13 function setupChannel() {
  14   return NetUtil.newChannel({
  15     uri: "http://localhost:" + httpserver.identity.primaryPort + testUrl,
  16     loadUsingSystemPrincipal: true,
  17   });
  18 }
  19
  20 function Listener() {}
  21
  22 Listener.prototype = {
  23   _buffer: null,
  24
  25   QueryInterface: ChromeUtils.generateQI([
  26     "nsIStreamListener",
  27     "nsIRequestObserver",
  28   ]),
  29
  30   onStartRequest(request) {
  31     Assert.equal(request.status, Cr.NS_OK);
  32     this._buffer = "";
  33   },
  34
  35   onDataAvailable(request, stream, offset, cnt) {
  36     if (pass == 0) {
  37       this._buffer = this._buffer.concat(read_stream(stream, cnt));
  38     } else {
  39       request.QueryInterface(Ci.nsICachingChannel);
  40       if (!request.isFromCache()) {
  41         do_throw("Response is not from the cache");
  42       }
  43
  44       request.cancel(Cr.NS_ERROR_ABORT);
  45     }
  46   },
  47
  48   onStopRequest() {
  49     if (pass == 0) {
  50       Assert.equal(this._buffer.length, responseLen);
  51       pass++;
  52
  53       var channel = setupChannel();
  54       channel.loadFlags = Ci.nsIRequest.VALIDATE_NEVER;
  55       channel.asyncOpen(new Listener());
  56     } else {
  57       httpserver.stop(do_test_finished);
  58       prefs.setCharPref("network.http.accept-encoding", cePref);
  59     }
  60   },
  61 };
  62
  63 var prefs;
  64 var cePref;
  65 function run_test() {
  66   do_get_profile();
  67
  68   prefs = Services.prefs;
  69   cePref = prefs.getCharPref("network.http.accept-encoding");
  70   prefs.setCharPref("network.http.accept-encoding", "gzip, deflate, br");
  71
  72   // Disable rcwn to make cache behavior deterministic.
  73   prefs.setBoolPref("network.http.rcwn.enabled", false);
  74
  75   httpserver.registerPathHandler(testUrl, handler);
  76   httpserver.start(-1);
  77
  78   var channel = setupChannel();
  79   channel.asyncOpen(new Listener());
  80
  81   do_test_pending();
  82 }
  83
  84 function handler(metadata, response) {
  85   Assert.equal(pass, 0); // the second response must be server from the cache
  86
  87   response.setStatusLine(metadata.httpVersion, 200, "OK");
  88   response.setHeader("Content-Type", "text/plain", false);
  89   response.setHeader("Content-Encoding", "br", false);
  90   response.setHeader("Content-Length", "" + responseBody.length, false);
  91
  92   var bos = Cc["@mozilla.org/binaryoutputstream;1"].createInstance(
  93     Ci.nsIBinaryOutputStream
  94   );
  95   bos.setOutputStream(response.bodyOutputStream);
  96
  97   response.processAsync();
  98   bos.writeByteArray(responseBody);
  99   response.finish();
 100 }