Current File : /home/vacivi36/vittasync.vacivitta.com.br/vittasync/node/test/parallel/test-http-upgrade-server.js
// Copyright Joyent, Inc. and other Node contributors.
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to permit
// persons to whom the Software is furnished to do so, subject to the
// following conditions:
//
// The above copyright notice and this permission notice shall be included
// in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
// USE OR OTHER DEALINGS IN THE SOFTWARE.
'use strict';
require('../common');
const assert = require('assert');
const net = require('net');
const http = require('http');
let requests_recv = 0;
let requests_sent = 0;
let request_upgradeHead = null;
function createTestServer() {
return new testServer();
}
function testServer() {
http.Server.call(this, () => {});
this.on('connection', function() {
requests_recv++;
});
this.on('request', function(req, res) {
res.writeHead(200, { 'Content-Type': 'text/plain' });
res.write('okay');
res.end();
});
this.on('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.on('data', function(d) {
const data = d.toString('utf8');
if (data === 'kill') {
socket.end();
} else {
socket.write(data, 'utf8');
}
});
});
}
Object.setPrototypeOf(testServer.prototype, http.Server.prototype);
Object.setPrototypeOf(testServer, http.Server);
function writeReq(socket, data, encoding) {
requests_sent++;
socket.write(data);
}
// connection: Upgrade with listener
function test_upgrade_with_listener() {
const conn = net.createConnection(server.address().port);
conn.setEncoding('utf8');
let state = 0;
conn.on('connect', function() {
writeReq(conn,
'GET / HTTP/1.1\r\n' +
'Host: example.com\r\n' +
'Upgrade: WebSocket\r\n' +
'Connection: Upgrade\r\n' +
'\r\n' +
'WjN}|M(6');
});
conn.on('data', function(data) {
state++;
assert.strictEqual(typeof data, 'string');
if (state === 1) {
assert.strictEqual(data.substr(0, 12), 'HTTP/1.1 101');
assert.strictEqual(request_upgradeHead.toString('utf8'), 'WjN}|M(6');
conn.write('test', 'utf8');
} else if (state === 2) {
assert.strictEqual(data, 'test');
conn.write('kill', 'utf8');
}
});
conn.on('end', function() {
assert.strictEqual(state, 2);
conn.end();
server.removeAllListeners('upgrade');
test_upgrade_no_listener();
});
}
// connection: Upgrade, no listener
function test_upgrade_no_listener() {
const conn = net.createConnection(server.address().port);
conn.setEncoding('utf8');
conn.on('connect', function() {
writeReq(conn,
'GET / HTTP/1.1\r\n' +
'Host: example.com\r\n' +
'Upgrade: WebSocket\r\n' +
'Connection: Upgrade\r\n' +
'\r\n');
});
conn.once('data', (data) => {
assert.strictEqual(typeof data, 'string');
assert.strictEqual(data.substr(0, 12), 'HTTP/1.1 200');
conn.end();
});
conn.on('close', function() {
test_standard_http();
});
}
// connection: normal
function test_standard_http() {
const conn = net.createConnection(server.address().port);
conn.setEncoding('utf8');
conn.on('connect', function() {
writeReq(conn, 'GET / HTTP/1.1\r\nHost: example.com\r\n\r\n');
});
conn.once('data', function(data) {
assert.strictEqual(typeof data, 'string');
assert.strictEqual(data.substr(0, 12), 'HTTP/1.1 200');
conn.end();
});
conn.on('close', function() {
server.close();
});
}
const server = createTestServer();
server.listen(0, function() {
// All tests get chained after this:
test_upgrade_with_listener();
});
// Fin.
process.on('exit', function() {
assert.strictEqual(requests_recv, 3);
assert.strictEqual(requests_sent, 3);
});
Mini Shell