michael@0: Cu.import("resource://testing-common/httpd.js");
michael@0: var httpserver = new HttpServer();
michael@0: 
michael@0: var expectedOnStopRequests = 3;
michael@0: 
michael@0: function setupChannel(suffix, xRequest, flags) {
michael@0:     var ios = Components.classes["@mozilla.org/network/io-service;1"]
michael@0:             .getService(Ci.nsIIOService);
michael@0:     var chan = ios.newChannel("http://localhost:" +
michael@0:                               httpserver.identity.primaryPort +
michael@0:                               suffix, "", null);
michael@0:     if (flags)
michael@0:         chan.loadFlags |= flags;
michael@0: 
michael@0:     var httpChan = chan.QueryInterface(Components.interfaces.nsIHttpChannel);
michael@0:     httpChan.setRequestHeader("x-request", xRequest, false);
michael@0:         
michael@0:     return httpChan;
michael@0: }
michael@0: 
michael@0: function Listener(response) {
michael@0:     this._response = response;
michael@0: }
michael@0: Listener.prototype = {
michael@0:     _response: null,
michael@0:     _buffer: null,
michael@0: 
michael@0:     QueryInterface: function(iid) {
michael@0:         if (iid.equals(Components.interfaces.nsIStreamListener) ||
michael@0:             iid.equals(Components.interfaces.nsIRequestObserver) ||
michael@0:             iid.equals(Components.interfaces.nsISupports))
michael@0:           return this;
michael@0:         throw Components.results.NS_ERROR_NO_INTERFACE;
michael@0:     },
michael@0: 
michael@0:     onStartRequest: function (request, ctx) {
michael@0:         this._buffer = "";
michael@0:     },
michael@0:     onDataAvailable: function (request, ctx, stream, offset, count) {
michael@0:         this._buffer = this._buffer.concat(read_stream(stream, count));
michael@0:     },
michael@0:     onStopRequest: function (request, ctx, status) {
michael@0:         do_check_eq(this._buffer, this._response);
michael@0:         if (--expectedOnStopRequests == 0)
michael@0:             do_timeout(10, function() {
michael@0:                         httpserver.stop(do_test_finished);
michael@0:                     });
michael@0:     }
michael@0: };
michael@0: 
michael@0: function run_test() {
michael@0:     httpserver.registerPathHandler("/bug596443", handler);
michael@0:     httpserver.start(-1);
michael@0: 
michael@0:     // make sure we have a profile so we can use the disk-cache
michael@0:     do_get_profile();
michael@0: 
michael@0:     // clear cache
michael@0:     evict_cache_entries();
michael@0: 
michael@0:     var ch0 = setupChannel("/bug596443", "Response0", Ci.nsIRequest.LOAD_BYPASS_CACHE);
michael@0:     ch0.asyncOpen(new Listener("Response0"), null);
michael@0: 
michael@0:     var ch1 = setupChannel("/bug596443", "Response1", Ci.nsIRequest.LOAD_BYPASS_CACHE);
michael@0:     ch1.asyncOpen(new Listener("Response1"), null);
michael@0: 
michael@0:     var ch2 = setupChannel("/bug596443", "Should not be used");
michael@0:     ch2.asyncOpen(new Listener("Response1"), null); // Note param: we expect this to come from cache
michael@0: 
michael@0:     do_test_pending();
michael@0: }
michael@0: 
michael@0: function triggerHandlers() {
michael@0:     do_timeout(100, handlers[1]);
michael@0:     do_timeout(100, handlers[0]);
michael@0: }
michael@0: 
michael@0: var handlers = [];
michael@0: function handler(metadata, response) {
michael@0:     var func = function(body) {
michael@0:         return function() {
michael@0:             response.setStatusLine(metadata.httpVersion, 200, "Ok");
michael@0:             response.setHeader("Content-Type", "text/plain", false);
michael@0:             response.setHeader("Content-Length", "" + body.length, false);
michael@0:             response.setHeader("Cache-Control", "max-age=600", false);
michael@0:             response.bodyOutputStream.write(body, body.length);
michael@0:             response.finish();
michael@0:          }};
michael@0: 
michael@0:     response.processAsync();
michael@0:     var request = metadata.getHeader("x-request");
michael@0:     handlers.push(func(request));
michael@0: 
michael@0:     if (handlers.length > 1)
michael@0:         triggerHandlers();
michael@0: }