gecko/netwerk/test/unit/test_resumable_channel.js
2007-08-30 16:00:38 -07:00

273 lines
9.0 KiB
JavaScript

/* Tests various aspects of nsIResumableChannel in combination with HTTP */
do_import_script("netwerk/test/httpserver/httpd.js");
var httpserver = null;
const NS_ERROR_ENTITY_CHANGED = 0x804b0020;
const NS_ERROR_NOT_RESUMABLE = 0x804b0019;
const rangeBody = "Body of the range request handler.\r\n";
function make_channel(url, callback, ctx) {
var ios = Cc["@mozilla.org/network/io-service;1"].
getService(Ci.nsIIOService);
return ios.newChannel(url, "", null);
}
function AuthPrompt2() {
}
AuthPrompt2.prototype = {
user: "guest",
pass: "guest",
QueryInterface: function authprompt2_qi(iid) {
if (iid.equals(Components.interfaces.nsISupports) ||
iid.equals(Components.interfaces.nsIAuthPrompt2))
return this;
throw Components.results.NS_ERROR_NO_INTERFACE;
},
promptAuth:
function ap2_promptAuth(channel, level, authInfo)
{
authInfo.username = this.user;
authInfo.password = this.pass;
return true;
},
asyncPromptAuth: function ap2_async(chan, cb, ctx, lvl, info) {
do_throw("not implemented yet")
}
};
function Requestor() {
}
Requestor.prototype = {
QueryInterface: function requestor_qi(iid) {
if (iid.equals(Components.interfaces.nsISupports) ||
iid.equals(Components.interfaces.nsIInterfaceRequestor))
return this;
throw Components.results.NS_ERROR_NO_INTERFACE;
},
getInterface: function requestor_gi(iid) {
if (iid.equals(Components.interfaces.nsIAuthPrompt2)) {
// Allow the prompt to store state by caching it here
if (!this.prompt2)
this.prompt2 = new AuthPrompt2();
return this.prompt2;
}
throw Components.results.NS_ERROR_NO_INTERFACE;
},
prompt2: null
};
function run_test() {
dump("*** run_test\n");
httpserver = new nsHttpServer();
httpserver.registerPathHandler("/auth", authHandler);
httpserver.registerPathHandler("/range", rangeHandler);
httpserver.registerPathHandler("/redir", redirHandler);
var entityID;
function get_entity_id(request, data, ctx) {
do_check_true(request instanceof Ci.nsIResumableChannel,
"must be a resumable channel");
entityID = request.entityID;
dump("*** entity id = " + entityID + "\n");
// Try a non-resumable URL (responds with 200)
var chan = make_channel("http://localhost:4444/");
chan.nsIResumableChannel.resumeAt(1, entityID);
chan.asyncOpen(new ChannelListener(try_resume, null, CL_EXPECT_FAILURE), null);
}
function try_resume(request, data, ctx) {
do_check_eq(request.status, NS_ERROR_NOT_RESUMABLE);
// Try a successful resume
var chan = make_channel("http://localhost:4444/range");
chan.nsIResumableChannel.resumeAt(1, entityID);
chan.asyncOpen(new ChannelListener(success, null), null);
}
function success(request, data, ctx) {
do_check_true(request.nsIHttpChannel.requestSucceeded);
do_check_eq(data, rangeBody.substring(1));
// Authentication (no password; working resume)
// (should not give us any data)
var chan = make_channel("http://localhost:4444/range");
chan.nsIResumableChannel.resumeAt(1, entityID);
chan.nsIHttpChannel.setRequestHeader("X-Need-Auth", "true", false);
chan.asyncOpen(new ChannelListener(test_auth_nopw, null, CL_EXPECT_FAILURE), null);
}
function test_auth_nopw(request, data, ctx) {
do_check_false(request.nsIHttpChannel.requestSucceeded);
do_check_eq(request.status, NS_ERROR_ENTITY_CHANGED);
// Authentication + not working resume
var chan = make_channel("http://localhost:4444/auth");
chan.nsIResumableChannel.resumeAt(1, entityID);
chan.notificationCallbacks = new Requestor();
chan.asyncOpen(new ChannelListener(test_auth, null, CL_EXPECT_FAILURE), null);
}
function test_auth(request, data, ctx) {
do_check_eq(request.status, NS_ERROR_NOT_RESUMABLE);
do_check_true(request.nsIHttpChannel.responseStatus < 300);
// Authentication + working resume
var chan = make_channel("http://localhost:4444/range");
chan.nsIResumableChannel.resumeAt(1, entityID);
chan.notificationCallbacks = new Requestor();
chan.nsIHttpChannel.setRequestHeader("X-Need-Auth", "true", false);
chan.asyncOpen(new ChannelListener(test_auth_resume, null), null);
}
function test_auth_resume(request, data, ctx) {
do_check_eq(data, rangeBody.substring(1));
do_check_true(request.nsIHttpChannel.requestSucceeded);
// 404 page (same content length as real content)
var chan = make_channel("http://localhost:4444/range");
chan.nsIResumableChannel.resumeAt(1, entityID);
chan.nsIHttpChannel.setRequestHeader("X-Want-404", "true", false);
chan.asyncOpen(new ChannelListener(test_404, null, CL_EXPECT_FAILURE), null);
}
function test_404(request, data, ctx) {
do_check_eq(request.status, NS_ERROR_ENTITY_CHANGED);
do_check_eq(request.nsIHttpChannel.responseStatus, 404);
// 416 Requested Range Not Satisfiable
var chan = make_channel("http://localhost:4444/range");
chan.nsIResumableChannel.resumeAt(1000, entityID);
chan.asyncOpen(new ChannelListener(test_416, null, CL_EXPECT_FAILURE), null);
}
function test_416(request, data, ctx) {
do_check_eq(request.status, NS_ERROR_ENTITY_CHANGED);
do_check_eq(request.nsIHttpChannel.responseStatus, 416);
// Redirect + successful resume
var chan = make_channel("http://localhost:4444/redir");
chan.nsIHttpChannel.setRequestHeader("X-Redir-To", "http://localhost:4444/range", false);
chan.nsIResumableChannel.resumeAt(1, entityID);
chan.asyncOpen(new ChannelListener(test_redir_resume, null), null);
}
function test_redir_resume(request, data, ctx) {
do_check_true(request.nsIHttpChannel.requestSucceeded);
do_check_eq(data, rangeBody.substring(1));
do_check_eq(request.nsIHttpChannel.responseStatus, 206);
// Redirect + failed resume
var chan = make_channel("http://localhost:4444/redir");
chan.nsIHttpChannel.setRequestHeader("X-Redir-To", "http://localhost:4444/", false);
chan.nsIResumableChannel.resumeAt(1, entityID);
chan.asyncOpen(new ChannelListener(test_redir_noresume, null, CL_EXPECT_FAILURE), null);
}
function test_redir_noresume(request, data, ctx) {
do_check_eq(request.status, NS_ERROR_NOT_RESUMABLE);
httpserver.stop();
do_test_finished();
}
httpserver.start(4444);
var chan = make_channel("http://localhost:4444/range");
chan.asyncOpen(new ChannelListener(get_entity_id, null), null);
do_test_pending();
}
// HANDLERS
function handleAuth(metadata, response) {
// btoa("guest:guest"), but that function is not available here
var expectedHeader = "Basic Z3Vlc3Q6Z3Vlc3Q=";
var body;
if (metadata.hasHeader("Authorization") &&
metadata.getHeader("Authorization") == expectedHeader)
{
response.setStatusLine(metadata.httpVersion, 200, "OK, authorized");
response.setHeader("WWW-Authenticate", 'Basic realm="secret"', false);
return true;
}
else
{
// didn't know guest:guest, failure
response.setStatusLine(metadata.httpVersion, 401, "Unauthorized");
response.setHeader("WWW-Authenticate", 'Basic realm="secret"', false);
return false;
}
}
// /auth
function authHandler(metadata, response) {
response.setHeader("Content-Type", "text/html", false);
body = handleAuth(metadata, response) ? "success" : "failure";
response.bodyOutputStream.write(body, body.length);
}
// /range
function rangeHandler(metadata, response) {
response.setHeader("Content-Type", "text/html", false);
if (metadata.hasHeader("X-Need-Auth")) {
if (!handleAuth(metadata, response)) {
body = "auth failed";
response.bodyOutputStream.write(body, body.length);
return;
}
}
if (metadata.hasHeader("X-Want-404")) {
response.setStatusLine(metadata.httpVersion, 404, "Not Found");
body = rangeBody;
response.bodyOutputStream.write(body, body.length);
return;
}
var body = rangeBody;
if (metadata.hasHeader("Range")) {
// Syntax: bytes=[from]-[to] (we don't support multiple ranges)
var matches = metadata.getHeader("Range").match(/^\s*bytes=(\d+)?-(\d+)?\s*$/);
var from = (matches[1] === undefined) ? 0 : matches[1];
var to = (matches[2] === undefined) ? rangeBody.length - 1 : matches[2];
if (from >= rangeBody.length) {
response.setStatusLine(metadata.httpVersion, 416, "Start pos too high");
response.setHeader("Content-Range", "*/" + rangeBody.length);
return;
}
body = body.substring(from, to + 1);
if (body.length != rangeBody.length) {
response.setStatusLine(metadata.httpVersion, 206, "Partial Content");
response.setHeader("Content-Range", from + "-" + to + "/" + rangeBody.length);
}
}
response.bodyOutputStream.write(body, body.length);
}
// /redir
function redirHandler(metadata, response) {
response.setStatusLine(metadata.httpVersion, 302, "Found");
response.setHeader("Content-Type", "text/html", false);
response.setHeader("Location", metadata.getHeader("X-Redir-To"));
var body = "redirect\r\n";
response.bodyOutputStream.write(body, body.length);
}