summaryrefslogtreecommitdiffstats
path: root/netwerk/test/unit/test_bug596443.js
blob: 07e4bd760be91a8013f79f8b0402f37d84697661 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
"use strict";

const { HttpServer } = ChromeUtils.import("resource://testing-common/httpd.js");

var httpProtocolHandler = Cc[
  "@mozilla.org/network/protocol;1?name=http"
].getService(Ci.nsIHttpProtocolHandler);

var httpserver = new HttpServer();

var expectedOnStopRequests = 3;

function setupChannel(suffix, xRequest, flags) {
  var chan = NetUtil.newChannel({
    uri: "http://localhost:" + httpserver.identity.primaryPort + suffix,
    loadUsingSystemPrincipal: true,
  });
  if (flags) {
    chan.loadFlags |= flags;
  }

  var httpChan = chan.QueryInterface(Ci.nsIHttpChannel);
  httpChan.setRequestHeader("x-request", xRequest, false);

  return httpChan;
}

function Listener(response) {
  this._response = response;
}
Listener.prototype = {
  _response: null,
  _buffer: null,

  QueryInterface: ChromeUtils.generateQI([
    "nsIStreamListener",
    "nsIRequestObserver",
  ]),

  onStartRequest(request) {
    this._buffer = "";
  },
  onDataAvailable(request, stream, offset, count) {
    this._buffer = this._buffer.concat(read_stream(stream, count));
  },
  onStopRequest(request, status) {
    Assert.equal(this._buffer, this._response);
    if (--expectedOnStopRequests == 0) {
      do_timeout(10, function() {
        httpserver.stop(do_test_finished);
      });
    }
  },
};

function run_test() {
  httpserver.registerPathHandler("/bug596443", handler);
  httpserver.start(-1);

  Services.prefs.setBoolPref("network.http.rcwn.enabled", false);

  // make sure we have a profile so we can use the disk-cache
  do_get_profile();

  // clear cache
  evict_cache_entries();

  httpProtocolHandler.EnsureHSTSDataReady().then(function() {
    var ch0 = setupChannel(
      "/bug596443",
      "Response0",
      Ci.nsIRequest.LOAD_BYPASS_CACHE
    );
    ch0.asyncOpen(new Listener("Response0"));

    var ch1 = setupChannel(
      "/bug596443",
      "Response1",
      Ci.nsIRequest.LOAD_BYPASS_CACHE
    );
    ch1.asyncOpen(new Listener("Response1"));

    var ch2 = setupChannel("/bug596443", "Should not be used");
    ch2.asyncOpen(new Listener("Response1")); // Note param: we expect this to come from cache
  });

  do_test_pending();
}

function triggerHandlers() {
  do_timeout(100, handlers[1]);
  do_timeout(100, handlers[0]);
}

var handlers = [];
function handler(metadata, response) {
  var func = function(body) {
    return function() {
      response.setStatusLine(metadata.httpVersion, 200, "Ok");
      response.setHeader("Content-Type", "text/plain", false);
      response.setHeader("Content-Length", "" + body.length, false);
      response.setHeader("Cache-Control", "max-age=600", false);
      response.bodyOutputStream.write(body, body.length);
      response.finish();
    };
  };

  response.processAsync();
  var request = metadata.getHeader("x-request");
  handlers.push(func(request));

  if (handlers.length > 1) {
    triggerHandlers();
  }
}