michael@0: Cu.import("resource://testing-common/httpd.js");
michael@0: Cu.import("resource://gre/modules/Services.jsm");
michael@0: var httpserver = new HttpServer();
michael@0: 
michael@0: function setupChannel(suffix)
michael@0: {
michael@0:     var ios =
michael@0:         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:     return chan;
michael@0: }
michael@0: 
michael@0: function checkValueAndTrigger(request, data, ctx)
michael@0: {
michael@0:     do_check_eq("Ok", data);
michael@0:     httpserver.stop(do_test_finished);
michael@0: }
michael@0: 
michael@0: function run_test()
michael@0: {
michael@0:     // Allow all cookies.
michael@0:     Services.prefs.setIntPref("network.cookie.cookieBehavior", 0);
michael@0: 
michael@0:     httpserver.registerPathHandler("/redirect1", redirectHandler1);
michael@0:     httpserver.registerPathHandler("/redirect2", redirectHandler2);
michael@0:     httpserver.start(-1);
michael@0: 
michael@0:     // clear cache
michael@0:     evict_cache_entries();
michael@0: 
michael@0:     // load first time
michael@0:     var channel = setupChannel("/redirect1");
michael@0:     channel.asyncOpen(new ChannelListener(checkValueAndTrigger, null), null);
michael@0:     do_test_pending();
michael@0: }
michael@0: 
michael@0: function redirectHandler1(metadata, response)
michael@0: {
michael@0: 	if (!metadata.hasHeader("Cookie")) {
michael@0: 	    response.setStatusLine(metadata.httpVersion, 302, "Found");
michael@0: 	    response.setHeader("Cache-Control", "max-age=600", false);
michael@0: 	    response.setHeader("Location", "/redirect2?query", false);
michael@0: 	    response.setHeader("Set-Cookie", "MyCookie=1", false);
michael@0:     } else {
michael@0: 	    response.setStatusLine(metadata.httpVersion, 200, "Ok");
michael@0:         response.setHeader("Content-Type", "text/plain");
michael@0:         response.bodyOutputStream.write("Ok", "Ok".length);
michael@0:     }
michael@0: }
michael@0: 
michael@0: function redirectHandler2(metadata, response)
michael@0: {
michael@0:     response.setStatusLine(metadata.httpVersion, 302, "Found");
michael@0:     response.setHeader("Location", "/redirect1", false);
michael@0: }