illiaKozyr
/
Homework


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340
							
var http = require('http');
var parse = require('url').parse;
var zlib = require('zlib');
var stream = require('stream');
var convert = require('encoding').convert;
var Multipart = require('parted').multipart;

module.exports = TestServer;

function TestServer() {
	this.server = http.createServer(this.router);
	this.port = 30001;
	this.hostname = 'localhost';
	// node 8 default keepalive timeout is 5000ms
	// make it shorter here as we want to close server quickly at the end of tests
	this.server.keepAliveTimeout = 1000;
	this.server.on('error', function(err) {
		console.log(err.stack);
	});
	this.server.on('connection', function(socket) {
		socket.setTimeout(1500);
	});
}

TestServer.prototype.start = function(cb) {
	this.server.listen(this.port, this.hostname, cb);
}

TestServer.prototype.stop = function(cb) {
	this.server.close(cb);
}

TestServer.prototype.router = function(req, res) {

	var p = parse(req.url).pathname;

	if (p === '/hello') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/plain');
		res.end('world');
	}

	if (p === '/plain') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/plain');
		res.end('text');
	}

	if (p === '/options') {
		res.statusCode = 200;
		res.setHeader('Allow', 'GET, HEAD, OPTIONS');
		res.end('hello world');
	}

	if (p === '/html') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/html');
		res.end('<html></html>');
	}

	if (p === '/json') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'application/json');
		res.end(JSON.stringify({
			name: 'value'
		}));
	}

	if (p === '/gzip') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/plain');
		res.setHeader('Content-Encoding', 'gzip');
		zlib.gzip('hello world', function(err, buffer) {
			res.end(buffer);
		});
	}

	if (p === '/deflate') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/plain');
		res.setHeader('Content-Encoding', 'deflate');
		zlib.deflate('hello world', function(err, buffer) {
			res.end(buffer);
		});
	}

	if (p === '/deflate-raw') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/plain');
		res.setHeader('Content-Encoding', 'deflate');
		zlib.deflateRaw('hello world', function(err, buffer) {
			res.end(buffer);
		});
	}

	if (p === '/sdch') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/plain');
		res.setHeader('Content-Encoding', 'sdch');
		res.end('fake sdch string');
	}

	if (p === '/invalid-content-encoding') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/plain');
		res.setHeader('Content-Encoding', 'gzip');
		res.end('fake gzip string');
	}

	if (p === '/timeout') {
		setTimeout(function() {
			res.statusCode = 200;
			res.setHeader('Content-Type', 'text/plain');
			res.end('text');
		}, 1000);
	}

	if (p === '/slow') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/plain');
		res.write('test');
		setTimeout(function() {
			res.end('test');
		}, 1000);
	}

	if (p === '/cookie') {
		res.statusCode = 200;
		res.setHeader('Set-Cookie', ['a=1', 'b=1']);
		res.end('cookie');
	}

	if (p === '/size/chunk') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/plain');
		setTimeout(function() {
			res.write('test');
		}, 50);
		setTimeout(function() {
			res.end('test');
		}, 100);
	}

	if (p === '/size/long') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/plain');
		res.end('testtest');
	}

	if (p === '/encoding/gbk') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/html');
		res.end(convert('<meta charset="gbk"><div>中文</div>', 'gbk'));
	}

	if (p === '/encoding/gb2312') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/html');
		res.end(convert('<meta http-equiv="Content-Type" content="text/html; charset=gb2312"><div>中文</div>', 'gb2312'));
	}

	if (p === '/encoding/shift-jis') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/html; charset=Shift-JIS');
		res.end(convert('<div>日本語</div>', 'Shift_JIS'));
	}

	if (p === '/encoding/euc-jp') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/xml');
		res.end(convert('<?xml version="1.0" encoding="EUC-JP"?><title>日本語</title>', 'EUC-JP'));
	}

	if (p === '/encoding/utf8') {
		res.statusCode = 200;
		res.end('中文');
	}

	if (p === '/encoding/order1') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'charset=gbk; text/plain');
		res.end(convert('中文', 'gbk'));
	}

	if (p === '/encoding/order2') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/plain; charset=gbk; qs=1');
		res.end(convert('中文', 'gbk'));
	}

	if (p === '/encoding/chunked') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/html');
		res.setHeader('Transfer-Encoding', 'chunked');
		var padding = 'a';
		for (var i = 0; i < 10; i++) {
			res.write(padding);
		}
		res.end(convert('<meta http-equiv="Content-Type" content="text/html; charset=Shift_JIS" /><div>日本語</div>', 'Shift_JIS'));
	}

	if (p === '/encoding/invalid') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'text/html');
		res.setHeader('Transfer-Encoding', 'chunked');
		// because node v0.12 doesn't have str.repeat
		var padding = new Array(120 + 1).join('a');
		for (var i = 0; i < 10; i++) {
			res.write(padding);
		}
		res.end(convert('中文', 'gbk'));
	}

	if (p === '/redirect/301') {
		res.statusCode = 301;
		res.setHeader('Location', '/inspect');
		res.end();
	}

	if (p === '/redirect/302') {
		res.statusCode = 302;
		res.setHeader('Location', '/inspect');
		res.end();
	}

	if (p === '/redirect/303') {
		res.statusCode = 303;
		res.setHeader('Location', '/inspect');
		res.end();
	}

	if (p === '/redirect/307') {
		res.statusCode = 307;
		res.setHeader('Location', '/inspect');
		res.end();
	}

	if (p === '/redirect/308') {
		res.statusCode = 308;
		res.setHeader('Location', '/inspect');
		res.end();
	}

	if (p === '/redirect/chain') {
		res.statusCode = 301;
		res.setHeader('Location', '/redirect/301');
		res.end();
	}

	if (p === '/error/redirect') {
		res.statusCode = 301;
		//res.setHeader('Location', '/inspect');
		res.end();
	}

	if (p === '/error/400') {
		res.statusCode = 400;
		res.setHeader('Content-Type', 'text/plain');
		res.end('client error');
	}

	if (p === '/error/404') {
		res.statusCode = 404;
		res.setHeader('Content-Encoding', 'gzip');
		res.end();
	}

	if (p === '/error/500') {
		res.statusCode = 500;
		res.setHeader('Content-Type', 'text/plain');
		res.end('server error');
	}

	if (p === '/error/reset') {
		res.destroy();
	}

	if (p === '/error/json') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'application/json');
		res.end('invalid json');
	}

	if (p === '/no-content') {
		res.statusCode = 204;
		res.end();
	}

	if (p === '/no-content/gzip') {
		res.statusCode = 204;
		res.setHeader('Content-Encoding', 'gzip');
		res.end();
	}

	if (p === '/not-modified') {
		res.statusCode = 304;
		res.end();
	}

	if (p === '/not-modified/gzip') {
		res.statusCode = 304;
		res.setHeader('Content-Encoding', 'gzip');
		res.end();
	}

	if (p === '/inspect') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'application/json');
		var body = '';
		req.on('data', function(c) { body += c });
		req.on('end', function() {
			res.end(JSON.stringify({
				method: req.method,
				url: req.url,
				headers: req.headers,
				body: body
			}));
		});
	}

	if (p === '/multipart') {
		res.statusCode = 200;
		res.setHeader('Content-Type', 'application/json');
		var parser = new Multipart(req.headers['content-type']);
		var body = '';
		parser.on('part', function(field, part) {
			body += field + '=' + part;
		});
		parser.on('end', function() {
			res.end(JSON.stringify({
				method: req.method,
				url: req.url,
				headers: req.headers,
				body: body
			}));
		});
		req.pipe(parser);
	}
}