socks5-http-client/test/index.js

SOCKS v5 HTTP client.

/**
 * @overview
 * @author Matthew Caruana Galizia <m@m.cg>
 * @copyright Copyright (c) 2013, Matthew Caruana Galizia
 * @license MIT
 * @preserve
 */

'use strict';

/*jshint node:true*/
/*global test, suite, setup, teardown*/

var assert = require('assert');
var net = require('net');
var socks = require('node-socks/socks.js');
var http = require('../');

suite('socks5-http-client tests', function() {
	var server, timeout;

	this.timeout(5000);

	setup(function(done) {
		server = socks.createServer(function(socket, port, address, proxyReady) {
			var proxy;

			proxy = net.createConnection(port, address, proxyReady);

			proxy.on('data', function(data) {
				if (timeout) {
					setTimeout(function() {
						socket.write(data);
					}, timeout);
				} else {
					socket.write(data);
				}
			});

			socket.on('data', function(data) {
				proxy.write(data);
			});

			proxy.on('close', function() {
				socket.end();
			});

			socket.on('close', function() {
				proxy.removeAllListeners('data');
				proxy.end();
			});
		});

		server.listen(1080, 'localhost', 511, function() {
			done();
		});

		server.on('error', function(err) {
			throw err;
		});
	});

	teardown(function(done) {
		server.close(done);

		timeout = null;
	});

	test('simple request', function(done) {
		var req;

		req = http.request('http://www.example.com/', function(res, err) {
			var data = '';

			assert.ifError(err);
			assert.equal(res.statusCode, 200);

			res.setEncoding('utf8');
			res.on('readable', function() {
				data += res.read();
			});

			res.on('end', function() {
				assert(-1 !== data.indexOf('<html'));
				assert(-1 !== data.indexOf('</html>'));

				done();
			});
		});

		req.on('error', function(err) {
			assert.fail(err);
		});

		// GET request, so end without sending any data.
		req.end();
	});

	test('timeout with setTimeout()', function(done) {
		var req;

		timeout = 2;

		req = http.get('http://www.example.com/');

		req.setTimeout(timeout / 2, function() {
			done();
		});

		req.on('error', function(err) {
			assert.fail(err);
		});
	});

	test('timeout with on()', function(done) {
		var req;

		timeout = 2;

		req = http.get('http://www.example.com/');

		req.setTimeout(timeout / 2);
		req.once('timeout', function() {
			done();
		});

		req.on('error', function(err) {
			assert.fail(err);
		});
	});
});