gecko/netwerk/test/unit/test_assoc.js

const Cc = Components.classes;
const Ci = Components.interfaces;
const Cu = Components.utils;
const Cr = Components.results;

Cu.import("resource://testing-common/httpd.js");

var httpserver = new HttpServer();
var currentTestIndex = 0;
var tests = [
             // this is valid
             {url: "/assoc/assoctest?valid",
              responseheader: [ "Assoc-Req: GET http://localhost:4444/assoc/assoctest?valid",
                                "Pragma: X-Verify-Assoc-Req" ],
              flags : 0},

             // this is invalid because the method is wrong
             {url: "/assoc/assoctest?invalid",
              responseheader: [ "Assoc-Req: POST http://localhost:4444/assoc/assoctest?invalid",
                                "Pragma: X-Verify-Assoc-Req" ],
              flags : CL_EXPECT_LATE_FAILURE},
             
             // this is invalid because the url is wrong
             {url: "/assoc/assoctest?notvalid",
              responseheader: [ "Assoc-Req: GET http://localhost:4444/wrongpath/assoc/assoctest?notvalid",
                                "Pragma: X-Verify-Assoc-Req" ],
              flags : CL_EXPECT_LATE_FAILURE},

             // this is invalid because the space between method and URL is missing
             {url: "/assoc/assoctest?invalid2",
              responseheader: [ "Assoc-Req: GEThttp://localhost:4444/assoc/assoctest?invalid2",
                                "Pragma: X-Verify-Assoc-Req" ],
              flags : CL_EXPECT_LATE_FAILURE},
];

var oldPrefVal;
var domBranch;

function setupChannel(url)
{
    var ios = Components.classes["@mozilla.org/network/io-service;1"].
                         getService(Ci.nsIIOService);
    var chan = ios.newChannel("http://localhost:4444" + url, "", null);
    return chan;
}

function startIter()
{
    var channel = setupChannel(tests[currentTestIndex].url);
    channel.asyncOpen(new ChannelListener(completeIter,
                                          channel, tests[currentTestIndex].flags), null);
}

function completeIter(request, data, ctx)
{
    if (++currentTestIndex < tests.length ) {
        startIter();
    } else {
        domBranch.setBoolPref("enforce", oldPrefVal);
        httpserver.stop(do_test_finished);
    }
}

function run_test()
{
    var prefService =
        Components.classes["@mozilla.org/preferences-service;1"]
        .getService(Components.interfaces.nsIPrefService);
    domBranch = prefService.getBranch("network.http.assoc-req.");
    oldPrefVal = domBranch.getBoolPref("enforce");
    domBranch.setBoolPref("enforce", true);

    httpserver.registerPathHandler("/assoc/assoctest", handler);
    httpserver.start(4444);

    startIter();
    do_test_pending();
}

function handler(metadata, response)
{
    var body = "thequickbrownfox";
    response.setHeader("Content-Type", "text/plain", false);

    var header = tests[currentTestIndex].responseheader;
    if (header != undefined) {
        for (var i = 0; i < header.length; i++) {
            var splitHdr = header[i].split(": ");
            response.setHeader(splitHdr[0], splitHdr[1], false);
        }
    }
    
    response.setStatusLine(metadata.httpVersion, 200, "OK");
    response.bodyOutputStream.write(body, body.length);
}