84 lines
2.1 KiB
JavaScript
84 lines
2.1 KiB
JavaScript
const { EventEmitter } = require('events');
|
|
const assert = require('assert');
|
|
const { UWSServer } = require('../../lib/io/uws');
|
|
const WebSocket = require('uws');
|
|
|
|
describe('UWSServer', () => {
|
|
const endpoint = 'ws://127.0.0.1:3000';
|
|
|
|
let server;
|
|
let socket;
|
|
beforeEach(done => {
|
|
server = new UWSServer();
|
|
server.on('error', e => { throw e; });
|
|
server.once('listening', done);
|
|
});
|
|
|
|
function connect() {
|
|
let socket = new WebSocket(endpoint);
|
|
socket.test = new EventEmitter();
|
|
|
|
socket.onmessage = message => {
|
|
const args = JSON.parse(message.data);
|
|
const frame = args.shift();
|
|
socket.test.emit(frame, ...args);
|
|
};
|
|
socket.onerror = e => { throw e; };
|
|
|
|
return socket;
|
|
}
|
|
|
|
afterEach(() => {
|
|
if (socket) socket.terminate();
|
|
socket = null;
|
|
if (server) server.shutdown();
|
|
server = null;
|
|
});
|
|
|
|
it('accepts a connection immediately if there is no middleware', done => {
|
|
socket = connect();
|
|
socket.test.on('connect', done);
|
|
});
|
|
|
|
it('accepts a connection with middleware', done => {
|
|
let m1 = false, m2 = false;
|
|
server.use((socket, next) => {
|
|
m1 = true;
|
|
next();
|
|
});
|
|
server.use((socket, next) => {
|
|
m2 = true;
|
|
next();
|
|
});
|
|
|
|
socket = connect();
|
|
socket.test.on('connect', () => {
|
|
assert(m1);
|
|
assert(m2);
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('rejects a connection with middleware', done => {
|
|
let m1 = false, m2 = false;
|
|
server.use((socket, next) => {
|
|
m1 = true;
|
|
next(new Error('broken'));
|
|
});
|
|
server.use((socket, next) => {
|
|
m2 = true;
|
|
next();
|
|
});
|
|
|
|
socket = connect();
|
|
socket.test.on('connect', () => {
|
|
throw new Error('Unexpected connect callback');
|
|
});
|
|
socket.test.on('error', e => {
|
|
assert.strictEqual(e, 'broken');
|
|
assert(!m2);
|
|
done();
|
|
});
|
|
});
|
|
});
|