michael@0: Cu.import("resource://testing-common/httpd.js");
michael@0: 
michael@0: XPCOMUtils.defineLazyGetter(this, "URL", function() {
michael@0:   return "http://localhost:" + httpserver.identity.primaryPort;
michael@0: });
michael@0: 
michael@0: var httpserver = new HttpServer();
michael@0: var testpath = "/simple";
michael@0: var httpbody = "0123456789";
michael@0: 
michael@0: var last = 0, max = 0;
michael@0: 
michael@0: const STATUS_RECEIVING_FROM = 0x804b0006;
michael@0: const LOOPS = 50000;
michael@0: 
michael@0: const TYPE_ONSTATUS = 1;
michael@0: const TYPE_ONPROGRESS = 2;
michael@0: const TYPE_ONSTARTREQUEST = 3;
michael@0: const TYPE_ONDATAAVAILABLE = 4;
michael@0: const TYPE_ONSTOPREQUEST = 5;
michael@0: 
michael@0: var progressCallback = {
michael@0:   _listener: null,
michael@0:   _got_onstartrequest: false,
michael@0:   _got_onstatus_after_onstartrequest: false,
michael@0:   _last_callback_handled: null,
michael@0: 
michael@0:   QueryInterface: function (iid) {
michael@0:     if (iid.equals(Ci.nsISupports) ||
michael@0: 	iid.equals(Ci.nsIProgressEventSink) ||
michael@0:         iid.equals(Ci.nsIStreamListener) ||
michael@0:         iid.equals(Ci.nsIRequestObserver))
michael@0:       return this;
michael@0:     throw Cr.NS_ERROR_NO_INTERFACE;
michael@0:   },
michael@0: 
michael@0:   getInterface: function (iid) {
michael@0:     if (iid.equals(Ci.nsIProgressEventSink) ||
michael@0:         iid.equals(Ci.nsIStreamListener) ||
michael@0:         iid.equals(Ci.nsIRequestObserver))
michael@0:       return this;
michael@0:     throw Cr.NS_ERROR_NO_INTERFACE;
michael@0:   },
michael@0: 
michael@0:   onStartRequest: function(request, context) {
michael@0:     do_check_eq(this._last_callback_handled, TYPE_ONSTATUS);
michael@0:     this._got_onstartrequest = true;
michael@0:     this._last_callback_handled = TYPE_ONSTARTREQUEST;
michael@0: 
michael@0:     this._listener = new ChannelListener(checkRequest, request);
michael@0:     this._listener.onStartRequest(request, context);
michael@0:   },
michael@0: 
michael@0:   onDataAvailable: function(request, context, data, offset, count) {
michael@0:     do_check_eq(this._last_callback_handled, TYPE_ONPROGRESS);
michael@0:     this._last_callback_handled = TYPE_ONDATAAVAILABLE;
michael@0: 
michael@0:     this._listener.onDataAvailable(request, context, data, offset, count);
michael@0:   },
michael@0: 
michael@0:   onStopRequest: function(request, context, status) {
michael@0:     do_check_eq(this._last_callback_handled, TYPE_ONDATAAVAILABLE);
michael@0:     do_check_true(this._got_onstatus_after_onstartrequest);
michael@0:     this._last_callback_handled = TYPE_ONSTOPREQUEST;
michael@0: 
michael@0:     this._listener.onStopRequest(request, context, status);
michael@0:     delete this._listener;
michael@0:   },
michael@0: 
michael@0:   onProgress: function (request, context, progress, progressMax) {
michael@0:     do_check_eq(this._last_callback_handled, TYPE_ONSTATUS);
michael@0:     this._last_callback_handled = TYPE_ONPROGRESS;
michael@0: 
michael@0:     do_check_eq(mStatus, STATUS_RECEIVING_FROM);
michael@0:     last = progress;
michael@0:     max = progressMax;
michael@0:   },
michael@0: 
michael@0:   onStatus: function (request, context, status, statusArg) {
michael@0:     if (!this._got_onstartrequest) {
michael@0:       // Ensure that all messages before onStartRequest are onStatus
michael@0:       if (this._last_callback_handled)
michael@0:         do_check_eq(this._last_callback_handled, TYPE_ONSTATUS);
michael@0:     } else if (this._last_callback_handled == TYPE_ONSTARTREQUEST) {
michael@0:       this._got_onstatus_after_onstartrequest = true;
michael@0:     } else {
michael@0:       do_check_eq(this._last_callback_handled, TYPE_ONDATAAVAILABLE);
michael@0:     }
michael@0:     this._last_callback_handled = TYPE_ONSTATUS;
michael@0: 
michael@0:     do_check_eq(statusArg, "localhost");
michael@0:     mStatus = status;
michael@0:   },
michael@0: 
michael@0:   mStatus: 0,
michael@0: };
michael@0: 
michael@0: function run_test() {
michael@0:   httpserver.registerPathHandler(testpath, serverHandler);
michael@0:   httpserver.start(-1);
michael@0:   var channel = setupChannel(testpath);
michael@0:   channel.asyncOpen(progressCallback, null);
michael@0:   do_test_pending();
michael@0: }
michael@0: 
michael@0: function setupChannel(path) {
michael@0:   var ios = Cc["@mozilla.org/network/io-service;1"].
michael@0:                        getService(Ci.nsIIOService);
michael@0:   var chan = ios.newChannel(URL + path, "", null);
michael@0:   chan.QueryInterface(Ci.nsIHttpChannel);
michael@0:   chan.requestMethod = "GET";
michael@0:   chan.notificationCallbacks = progressCallback;
michael@0:   return chan;
michael@0: }
michael@0: 
michael@0: function serverHandler(metadata, response) {
michael@0:   response.setHeader("Content-Type", "text/plain", false);
michael@0:   for (let i = 0; i < LOOPS; i++)
michael@0:     response.bodyOutputStream.write(httpbody, httpbody.length);
michael@0: }
michael@0: 
michael@0: function checkRequest(request, data, context) {
michael@0:   do_check_eq(last, httpbody.length*LOOPS);
michael@0:   do_check_eq(max, httpbody.length*LOOPS);
michael@0:   httpserver.stop(do_test_finished);
michael@0: }