175 lines
8 KiB
JavaScript
175 lines
8 KiB
JavaScript
/*
|
|
* vim: ts=4:sw=4:expandtab
|
|
*/
|
|
|
|
;(function() {
|
|
'use strict';
|
|
|
|
describe('WebSocket-Resource', function() {
|
|
describe('requests and responses', function () {
|
|
it('receives requests and sends responses', function(done) {
|
|
// mock socket
|
|
var request_id = '1';
|
|
var socket = {
|
|
send: function(data) {
|
|
var message = textsecure.protobuf.WebSocketMessage.decode(data);
|
|
assert.strictEqual(message.type, textsecure.protobuf.WebSocketMessage.Type.RESPONSE);
|
|
assert.strictEqual(message.response.message, 'OK');
|
|
assert.strictEqual(message.response.status, 200);
|
|
assert.strictEqual(message.response.id.toString(), request_id);
|
|
done();
|
|
}
|
|
};
|
|
|
|
// actual test
|
|
var resource = new WebSocketResource(socket, {
|
|
handleRequest: function (request) {
|
|
assert.strictEqual(request.verb, 'PUT');
|
|
assert.strictEqual(request.path, '/some/path');
|
|
assertEqualArrayBuffers(request.body.toArrayBuffer(), new Uint8Array([1,2,3]).buffer);
|
|
request.respond(200, 'OK');
|
|
}
|
|
});
|
|
|
|
// mock socket request
|
|
socket.onmessage({
|
|
data: new Blob([
|
|
new textsecure.protobuf.WebSocketMessage({
|
|
type: textsecure.protobuf.WebSocketMessage.Type.REQUEST,
|
|
request: {
|
|
id: request_id,
|
|
verb: 'PUT',
|
|
path: '/some/path',
|
|
body: new Uint8Array([1,2,3]).buffer
|
|
}
|
|
}).encode().toArrayBuffer()
|
|
])
|
|
});
|
|
});
|
|
|
|
it('sends requests and receives responses', function(done) {
|
|
// mock socket and request handler
|
|
var request_id;
|
|
var socket = {
|
|
send: function(data) {
|
|
var message = textsecure.protobuf.WebSocketMessage.decode(data);
|
|
assert.strictEqual(message.type, textsecure.protobuf.WebSocketMessage.Type.REQUEST);
|
|
assert.strictEqual(message.request.verb, 'PUT');
|
|
assert.strictEqual(message.request.path, '/some/path');
|
|
assertEqualArrayBuffers(message.request.body.toArrayBuffer(), new Uint8Array([1,2,3]).buffer);
|
|
request_id = message.request.id;
|
|
}
|
|
};
|
|
|
|
// actual test
|
|
var resource = new WebSocketResource(socket);
|
|
resource.sendRequest({
|
|
verb: 'PUT',
|
|
path: '/some/path',
|
|
body: new Uint8Array([1,2,3]).buffer,
|
|
error: done,
|
|
success: function(message, status, request) {
|
|
assert.strictEqual(message, 'OK');
|
|
assert.strictEqual(status, 200);
|
|
done();
|
|
}
|
|
});
|
|
|
|
// mock socket response
|
|
socket.onmessage({
|
|
data: new Blob([
|
|
new textsecure.protobuf.WebSocketMessage({
|
|
type: textsecure.protobuf.WebSocketMessage.Type.RESPONSE,
|
|
response: { id: request_id, message: 'OK', status: 200 }
|
|
}).encode().toArrayBuffer()
|
|
])
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('close', function() {
|
|
before(function() { window.WebSocket = MockSocket; });
|
|
after (function() { window.WebSocket = WebSocket; });
|
|
it('closes the connection', function(done) {
|
|
var mockServer = new MockServer('ws://localhost:8081');
|
|
mockServer.on('connection', function(server) {
|
|
server.on('close', done);
|
|
});
|
|
var resource = new WebSocketResource(new WebSocket('ws://localhost:8081'));
|
|
resource.close();
|
|
});
|
|
});
|
|
|
|
describe('with a keepalive config', function() {
|
|
before(function() { window.WebSocket = MockSocket; });
|
|
after (function() { window.WebSocket = WebSocket; });
|
|
this.timeout(60000);
|
|
it('sends keepalives once a minute', function(done) {
|
|
var mockServer = new MockServer('ws://localhost:8081');
|
|
mockServer.on('connection', function(server) {
|
|
server.on('message', function(data) {
|
|
var message = textsecure.protobuf.WebSocketMessage.decode(data);
|
|
assert.strictEqual(message.type, textsecure.protobuf.WebSocketMessage.Type.REQUEST);
|
|
assert.strictEqual(message.request.verb, 'GET');
|
|
assert.strictEqual(message.request.path, '/v1/keepalive');
|
|
server.close();
|
|
done();
|
|
});
|
|
});
|
|
new WebSocketResource(new WebSocket('ws://localhost:8081'), {
|
|
keepalive: { path: '/v1/keepalive' }
|
|
});
|
|
});
|
|
|
|
it('uses / as a default path', function(done) {
|
|
var mockServer = new MockServer('ws://localhost:8081');
|
|
mockServer.on('connection', function(server) {
|
|
server.on('message', function(data) {
|
|
var message = textsecure.protobuf.WebSocketMessage.decode(data);
|
|
assert.strictEqual(message.type, textsecure.protobuf.WebSocketMessage.Type.REQUEST);
|
|
assert.strictEqual(message.request.verb, 'GET');
|
|
assert.strictEqual(message.request.path, '/');
|
|
server.close();
|
|
done();
|
|
});
|
|
});
|
|
new WebSocketResource(new WebSocket('ws://localhost:8081'), {
|
|
keepalive: true
|
|
});
|
|
|
|
});
|
|
|
|
it('optionally disconnects if no response', function(done) {
|
|
this.timeout(65000);
|
|
var mockServer = new MockServer('ws://localhost:8081');
|
|
var socket = new WebSocket('ws://localhost:8081');
|
|
mockServer.on('connection', function(server) {
|
|
server.on('close', done);
|
|
});
|
|
new WebSocketResource(socket, { keepalive: true });
|
|
});
|
|
|
|
it('allows resetting the keepalive timer', function(done) {
|
|
this.timeout(65000);
|
|
var mockServer = new MockServer('ws://localhost:8081');
|
|
var socket = new WebSocket('ws://localhost:8081');
|
|
var startTime = Date.now();
|
|
mockServer.on('connection', function(server) {
|
|
server.on('message', function(data) {
|
|
var message = textsecure.protobuf.WebSocketMessage.decode(data);
|
|
assert.strictEqual(message.type, textsecure.protobuf.WebSocketMessage.Type.REQUEST);
|
|
assert.strictEqual(message.request.verb, 'GET');
|
|
assert.strictEqual(message.request.path, '/');
|
|
assert(Date.now() > startTime + 60000, 'keepalive time should be longer than a minute');
|
|
server.close();
|
|
done();
|
|
});
|
|
});
|
|
var resource = new WebSocketResource(socket, { keepalive: true });
|
|
setTimeout(function() {
|
|
resource.resetKeepAliveTimer()
|
|
}, 5000);
|
|
});
|
|
});
|
|
});
|
|
}());
|