michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: Components.utils.import("resource://testing-common/httpd.js");
michael@0: 
michael@0: var server = new HttpServer();
michael@0: server.start(-1);
michael@0: 
michael@0: const SERVER_PORT = server.identity.primaryPort;
michael@0: const HTTP_BASE = "http://localhost:" + SERVER_PORT;
michael@0: const redirectPath = "/redirect";
michael@0: const headerCheckPath = "/headerCheck";
michael@0: const redirectURL = HTTP_BASE + redirectPath;
michael@0: const headerCheckURL = HTTP_BASE + headerCheckPath;
michael@0: 
michael@0: function redirectHandler(metadata, response) {
michael@0:   response.setStatusLine(metadata.httpVersion, 302, "Found");
michael@0:   response.setHeader("Location", headerCheckURL, false);
michael@0: }
michael@0: 
michael@0: function headerCheckHandler(metadata, response) {
michael@0:   try {
michael@0:     let headerValue = metadata.getHeader("X-Custom-Header");
michael@0:     do_check_eq(headerValue, "present");
michael@0:   } catch(e) {
michael@0:     do_throw("No header present after redirect");
michael@0:   }
michael@0:   try {
michael@0:     metadata.getHeader("X-Unwanted-Header");
michael@0:     do_throw("Unwanted header present after redirect");
michael@0:   } catch (x) {
michael@0:   }
michael@0:   response.setStatusLine(metadata.httpVersion, 200, "OK");
michael@0:   response.setHeader("Content-Type", "text/plain");
michael@0:   response.write("");
michael@0: }
michael@0: 
michael@0: function run_test() {
michael@0:   server.registerPathHandler(redirectPath, redirectHandler);
michael@0:   server.registerPathHandler(headerCheckPath, headerCheckHandler);
michael@0: 
michael@0:   do_test_pending();
michael@0:   var request = Components.classes["@mozilla.org/xmlextras/xmlhttprequest;1"]
michael@0:                 .createInstance(Components.interfaces.nsIXMLHttpRequest);
michael@0:   request.open("GET", redirectURL, true);
michael@0:   request.setRequestHeader("X-Custom-Header", "present");
michael@0:   request.addEventListener("readystatechange", function() {
michael@0:     if (request.readyState == 4) {
michael@0:       do_check_eq(request.status, 200);
michael@0:       server.stop(do_test_finished);
michael@0:     }
michael@0:   }, false);
michael@0:   request.send();
michael@0:   try {
michael@0:     request.setRequestHeader("X-Unwanted-Header", "present");
michael@0:     do_throw("Shouldn't be able to set a header after send");
michael@0:   } catch (x) {
michael@0:   }    
michael@0: }