"use strict"; const { HttpServer } = ChromeUtils.importESModule( "resource://testing-common/httpd.sys.mjs" ); var httpserver = null; ChromeUtils.defineLazyGetter(this, "uri", function () { return "http://localhost:" + httpserver.identity.primaryPort + "/multipart"; }); function make_channel(url) { return NetUtil.newChannel({ uri: url, loadUsingSystemPrincipal: true }); } var multipartBody = "--boundary\r\nSet-Cookie: foo=bar\r\n\r\nSome text\r\n--boundary\r\nContent-Type: text/x-test\r\n\r\n\r\n\r\n--boundary\r\n\r\n\r\n--boundary--"; function contentHandler(metadata, response) { response.setHeader("Content-Type", 'multipart/mixed; boundary="boundary"'); response.bodyOutputStream.write(multipartBody, multipartBody.length); } var numTests = 2; var testNum = 0; var testData = [ { data: "Some text", type: "text/plain" }, { data: "\r\n", type: "text/x-test" }, { data: "", type: "text/xml" }, ]; function responseHandler(request, buffer) { Assert.equal(buffer, testData[testNum].data); Assert.equal( request.QueryInterface(Ci.nsIChannel).contentType, testData[testNum].type ); if (++testNum == numTests) { httpserver.stop(do_test_finished); } } var multipartListener = { _buffer: "", _index: 0, QueryInterface: ChromeUtils.generateQI([ "nsIStreamListener", "nsIRequestObserver", ]), onStartRequest(request) { this._buffer = ""; }, onDataAvailable(request, stream, offset, count) { try { this._buffer = this._buffer.concat(read_stream(stream, count)); dump("BUFFEEE: " + this._buffer + "\n\n"); } catch (ex) { do_throw("Error in onDataAvailable: " + ex); } }, onStopRequest(request, status) { this._index++; // Second part should be last part Assert.equal( request.QueryInterface(Ci.nsIMultiPartChannel).isLastPart, this._index == testData.length ); try { responseHandler(request, this._buffer); } catch (ex) { do_throw("Error in closure function: " + ex); } }, }; function run_test() { httpserver = new HttpServer(); httpserver.registerPathHandler("/multipart", contentHandler); httpserver.start(-1); var streamConv = Cc["@mozilla.org/streamConverters;1"].getService( Ci.nsIStreamConverterService ); var conv = streamConv.asyncConvertData( "multipart/mixed", "*/*", multipartListener, null ); var chan = make_channel(uri); chan.asyncOpen(conv); do_test_pending(); }