bug 346016 add a unit test for authentication prompts (447660b4) · Commits · The Tor Project / Applications / Mullvad Browser

netwerk/test/unit/head_http_server.js

+40 −8

Original line number	Diff line number	Diff line
		@@ -60,6 +60,29 @@ nsTestServ.prototype =
		"\r\n" +
		"Moved to a file URI";
		stream.write(response, response.length);
		},

		"/auth": function(stream, req_head) {
		// btoa("guest:guest"), but
		// that function is not available here
		var expectedHeader = "Basic Z3Vlc3Q6Z3Vlc3Q=";
		var successResponse = this.headers("200 OK, authorized") +
		'WWW-Authenticate: Basic realm="secret"\r\n' +
		"\r\n" +
		"success";
		var failedResponse = this.headers("401 Unauthorized") +
		'WWW-Authenticate: Basic realm="secret"\r\n' +
		"\r\n" +
		"failed";
		if ("authorization" in req_head) {
		// Make sure the credentials are right
		if (req_head.authorization == expectedHeader) {
		stream.write(successResponse, successResponse.length);
		return;
		}
		// fall through to failure response
		}
		stream.write(failedResponse, failedResponse.length);
		}
		},

		@@ -85,7 +108,7 @@ nsTestServ.prototype =
		var request = this.parseInput(input);

		// Strip away query parameters, then unescape the request
		var path = request.replace(/\?.*/, "");
		var path = request[0].replace(/\?.*/, "");
		try {
		path = decodeURI(path);
		} catch (ex) {
		@@ -93,9 +116,9 @@ nsTestServ.prototype =
		}

		if (path in this.handler)
		this.handler[path](output);
		this.handler[path](output, request[1]);
		else
		this.handler[404](output);
		this.handler[404](output, request[1]);

		input.close();
		output.close();
		@@ -131,20 +154,29 @@ nsTestServ.prototype =

		var request = line.value.split(/ +/);
		if (request[0] != "GET" && request[0] != "POST")
		return 400;
		return [400];

		// This server doesn't support HTTP 0.9
		if (request[2] != "HTTP/1.0" && request[2] != "HTTP/1.1")
		return 400;
		return [400];

		while (line.value != "") {
		var req_head = {};
		while (true) {
		lis.readLine(line);
		if (line.value == "")
		break;
		var matches = line.value.match(/^([a-zA-Z]+): (.)$/);
		if (matches !== null && matches.length == 3) {
		// This is slightly wrong - should probably merge headers with the
		// same name
		req_head[matches[1].toLowerCase()] = matches[2];
		}
		}

		if (request[1][0] != "/")
		return 400;
		return [400];

		return request[1];
		return [request[1], req_head];
		}
		}

netwerk/test/unit/test_authentication.js

0 → 100644

+141 −0

Original line number	Diff line number	Diff line
		// This file tests authentication prompt callbacks

		function AuthPrompt1() {
		}

		// Also ought to test:
		// - return false from authprompt
		// - return wrong username from authprompt

		AuthPrompt1.prototype = {
		user: "guest",
		pass: "guest",

		expectedRealm: "secret",

		QueryInterface: function authprompt_qi(iid) {
		if (iid.equals(Components.interfaces.nsISupports) \|\|
		iid.equals(Components.interfaces.nsIAuthPrompt))
		return this;
		throw Components.results.NS_ERROR_NO_INTERFACE;
		},

		prompt: function ap1_prompt(title, text, realm, save, defaultText, result) {
		do_throw("unexpected prompt call");
		},

		promptUsernameAndPassword:
		function ap1_promptUP(title, text, realm, savePW, user, pw)
		{
		// Note that the realm here isn't actually the realm. it's a pw mgr key.
		do_check_eq("localhost:4444 (" + this.expectedRealm + ")", realm);
		if (text.indexOf(this.expectedRealm) == -1)
		do_throw("Text must indicate the realm");
		if (text.indexOf("localhost") == -1)
		do_throw("Text must indicate the hostname");
		if (text.indexOf("4444") == -1)
		do_throw("Text must indicate the port");
		if (text.indexOf("-1") != -1)
		do_throw("Text must contain negative numbers");

		user.value = this.user;
		pw.value = this.pass;
		return true;
		},

		promptPassword: function ap1_promptPW(title, text, realm, save, pwd) {
		do_throw("unexpected promptPassword call");
		}

		};

		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.nsIAuthPrompt))
		return new AuthPrompt1();
		throw Components.results.NS_ERROR_NO_INTERFACE;
		}
		};

		var listener = {
		expectedCode: 401, // Unauthorized

		onStartRequest: function test_onStartR(request, ctx) {
		try {
		if (!Components.isSuccessCode(request.status))
		do_throw("Channel should have a success code!");

		if (!(request instanceof Components.interfaces.nsIHttpChannel))
		do_throw("Expecting an HTTP channel");

		do_check_eq(request.responseStatus, this.expectedCode);
		// The request should be succeeded iff we expect 200
		do_check_eq(request.requestSucceeded, this.expectedCode == 200);

		} catch (e) {
		do_throw("Unexpected exception: " + e);
		}

		throw Components.results.NS_ERROR_ABORT;
		},

		onDataAvailable: function test_ODA() {
		do_throw("Should not get any data!");
		},

		onStopRequest: function test_onStopR(request, ctx, status) {
		do_check_eq(status, Components.results.NS_ERROR_ABORT);

		if (current_test < (tests.length - 1)) {
		current_test++;
		tests[current_test]();
		}
		do_test_finished();
		}
		};

		function makeChan(url) {
		var ios = Components.classes["@mozilla.org/network/io-service;1"]
		.getService(Components.interfaces.nsIIOService);
		var chan = ios.newChannel(url, null, null)
		.QueryInterface(Components.interfaces.nsIHttpChannel);

		return chan;
		}

		var tests = [test_noauth, test_prompt1];
		var current_test = 0;

		function run_test() {
		start_server(4444);
		tests[0]();
		}

		function test_noauth() {
		var chan = makeChan("http://localhost:4444/auth");

		chan.asyncOpen(listener, null);

		do_test_pending();
		}

		function test_prompt1() {
		var chan = makeChan("http://localhost:4444/auth");

		chan.notificationCallbacks = new Requestor();
		listener.expectedCode = 200; // OK
		chan.asyncOpen(listener, null);

		do_test_pending();
		}