You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

165 lines
3.9 KiB

14 years ago
var common = require('../common');
var assert = require('assert');
var util = require('util');
var net = require('net');
var http = require('http');
var requests_recv = 0;
var requests_sent = 0;
var request_upgradeHead = null;
function createTestServer() {
return new testServer();
}
function testServer() {
var server = this;
http.Server.call(server, function() {});
server.addListener('connection', function() {
requests_recv++;
});
server.addListener('request', function(req, res) {
res.writeHead(200, {'Content-Type': 'text/plain'});
res.write('okay');
res.end();
});
server.addListener('upgrade', function(req, socket, upgradeHead) {
socket.write('HTTP/1.1 101 Web Socket Protocol Handshake\r\n' +
'Upgrade: WebSocket\r\n' +
'Connection: Upgrade\r\n' +
'\r\n\r\n');
request_upgradeHead = upgradeHead;
socket.ondata = function(d, start, end) {
var data = d.toString('utf8', start, end);
if (data == 'kill') {
socket.end();
} else {
socket.write(data, 'utf8');
}
};
});
}
util.inherits(testServer, http.Server);
function writeReq(socket, data, encoding) {
requests_sent++;
socket.write(data);
}
/*-----------------------------------------------
connection: Upgrade with listener
-----------------------------------------------*/
function test_upgrade_with_listener(_server) {
var conn = net.createConnection(common.PORT);
conn.setEncoding('utf8');
var state = 0;
conn.addListener('connect', function() {
writeReq(conn,
'GET / HTTP/1.1\r\n' +
'Upgrade: WebSocket\r\n' +
'Connection: Upgrade\r\n' +
'\r\n' +
'WjN}|M(6');
});
conn.addListener('data', function(data) {
state++;
assert.equal('string', typeof data);
if (state == 1) {
assert.equal('HTTP/1.1 101', data.substr(0, 12));
assert.equal('WjN}|M(6', request_upgradeHead.toString('utf8'));
conn.write('test', 'utf8');
} else if (state == 2) {
assert.equal('test', data);
conn.write('kill', 'utf8');
}
});
conn.addListener('end', function() {
assert.equal(2, state);
conn.end();
_server.removeAllListeners('upgrade');
test_upgrade_no_listener();
});
}
/*-----------------------------------------------
connection: Upgrade, no listener
-----------------------------------------------*/
var test_upgrade_no_listener_ended = false;
function test_upgrade_no_listener() {
var conn = net.createConnection(common.PORT);
conn.setEncoding('utf8');
conn.addListener('connect', function() {
writeReq(conn,
'GET / HTTP/1.1\r\n' +
'Upgrade: WebSocket\r\n' +
'Connection: Upgrade\r\n' +
'\r\n');
});
conn.addListener('end', function() {
test_upgrade_no_listener_ended = true;
conn.end();
});
conn.addListener('close', function() {
test_standard_http();
});
}
/*-----------------------------------------------
connection: normal
-----------------------------------------------*/
function test_standard_http() {
var conn = net.createConnection(common.PORT);
conn.setEncoding('utf8');
conn.addListener('connect', function() {
writeReq(conn, 'GET / HTTP/1.1\r\n\r\n');
});
conn.addListener('data', function(data) {
assert.equal('string', typeof data);
assert.equal('HTTP/1.1 200', data.substr(0, 12));
conn.end();
});
conn.addListener('close', function() {
server.close();
});
}
var server = createTestServer();
server.listen(common.PORT, function() {
// All tests get chained after this:
test_upgrade_with_listener(server);
});
/*-----------------------------------------------
Fin.
-----------------------------------------------*/
process.addListener('exit', function() {
assert.equal(3, requests_recv);
assert.equal(3, requests_sent);
assert.ok(test_upgrade_no_listener_ended);
});