-var assert = chai.assert;
-var expect = chai.expect;
+const expect = chai.expect;
import Websock from '../core/websock.js';
import FakeWebSocket from './fake.websocket.js';
-import sinon from '../vendor/sinon.js';
-
-describe('Websock', function() {
+describe('Websock', function () {
"use strict";
describe('Queue methods', function () {
- var sock;
- var RQ_TEMPLATE = new Uint8Array([0, 1, 2, 3, 4, 5, 6, 7]);
+ let sock;
+ const RQ_TEMPLATE = new Uint8Array([0, 1, 2, 3, 4, 5, 6, 7]);
beforeEach(function () {
sock = new Websock();
});
describe('rQlen', function () {
it('should return the length of the receive queue', function () {
- sock.set_rQi(0);
+ sock.rQi = 0;
- expect(sock.rQlen()).to.equal(RQ_TEMPLATE.length);
+ expect(sock.rQlen).to.equal(RQ_TEMPLATE.length);
});
it("should return the proper length if we read some from the receive queue", function () {
- sock.set_rQi(1);
+ sock.rQi = 1;
- expect(sock.rQlen()).to.equal(RQ_TEMPLATE.length - 1);
+ expect(sock.rQlen).to.equal(RQ_TEMPLATE.length - 1);
});
});
describe('rQpeek8', function () {
it('should peek at the next byte without poping it off the queue', function () {
- var bef_len = sock.rQlen();
- var peek = sock.rQpeek8();
+ const bef_len = sock.rQlen;
+ const peek = sock.rQpeek8();
expect(sock.rQpeek8()).to.equal(peek);
- expect(sock.rQlen()).to.equal(bef_len);
+ expect(sock.rQlen).to.equal(bef_len);
});
});
- describe('rQshift8', function () {
+ describe('rQshift8()', function () {
it('should pop a single byte from the receive queue', function () {
- var peek = sock.rQpeek8();
- var bef_len = sock.rQlen();
+ const peek = sock.rQpeek8();
+ const bef_len = sock.rQlen;
expect(sock.rQshift8()).to.equal(peek);
- expect(sock.rQlen()).to.equal(bef_len - 1);
+ expect(sock.rQlen).to.equal(bef_len - 1);
});
});
- describe('rQshift16', function () {
+ describe('rQshift16()', function () {
it('should pop two bytes from the receive queue and return a single number', function () {
- var bef_len = sock.rQlen();
- var expected = (RQ_TEMPLATE[0] << 8) + RQ_TEMPLATE[1];
+ const bef_len = sock.rQlen;
+ const expected = (RQ_TEMPLATE[0] << 8) + RQ_TEMPLATE[1];
expect(sock.rQshift16()).to.equal(expected);
- expect(sock.rQlen()).to.equal(bef_len - 2);
+ expect(sock.rQlen).to.equal(bef_len - 2);
});
});
- describe('rQshift32', function () {
+ describe('rQshift32()', function () {
it('should pop four bytes from the receive queue and return a single number', function () {
- var bef_len = sock.rQlen();
- var expected = (RQ_TEMPLATE[0] << 24) +
+ const bef_len = sock.rQlen;
+ const expected = (RQ_TEMPLATE[0] << 24) +
(RQ_TEMPLATE[1] << 16) +
(RQ_TEMPLATE[2] << 8) +
RQ_TEMPLATE[3];
expect(sock.rQshift32()).to.equal(expected);
- expect(sock.rQlen()).to.equal(bef_len - 4);
+ expect(sock.rQlen).to.equal(bef_len - 4);
});
});
describe('rQshiftStr', function () {
it('should shift the given number of bytes off of the receive queue and return a string', function () {
- var bef_len = sock.rQlen();
- var bef_rQi = sock.get_rQi();
- var shifted = sock.rQshiftStr(3);
+ const bef_len = sock.rQlen;
+ const bef_rQi = sock.rQi;
+ const shifted = sock.rQshiftStr(3);
expect(shifted).to.be.a('string');
expect(shifted).to.equal(String.fromCharCode.apply(null, Array.prototype.slice.call(new Uint8Array(RQ_TEMPLATE.buffer, bef_rQi, 3))));
- expect(sock.rQlen()).to.equal(bef_len - 3);
+ expect(sock.rQlen).to.equal(bef_len - 3);
});
it('should shift the entire rest of the queue off if no length is given', function () {
sock.rQshiftStr();
- expect(sock.rQlen()).to.equal(0);
+ expect(sock.rQlen).to.equal(0);
+ });
+
+ it('should be able to handle very large strings', function () {
+ const BIG_LEN = 500000;
+ const RQ_BIG = new Uint8Array(BIG_LEN);
+ let expected = "";
+ let letterCode = 'a'.charCodeAt(0);
+ for (let i = 0; i < BIG_LEN; i++) {
+ RQ_BIG[i] = letterCode;
+ expected += String.fromCharCode(letterCode);
+
+ if (letterCode < 'z'.charCodeAt(0)) {
+ letterCode++;
+ } else {
+ letterCode = 'a'.charCodeAt(0);
+ }
+ }
+ sock._rQ.set(RQ_BIG);
+ sock._rQlen = RQ_BIG.length;
+
+ const shifted = sock.rQshiftStr();
+
+ expect(shifted).to.be.equal(expected);
+ expect(sock.rQlen).to.equal(0);
});
});
describe('rQshiftBytes', function () {
it('should shift the given number of bytes of the receive queue and return an array', function () {
- var bef_len = sock.rQlen();
- var bef_rQi = sock.get_rQi();
- var shifted = sock.rQshiftBytes(3);
+ const bef_len = sock.rQlen;
+ const bef_rQi = sock.rQi;
+ const shifted = sock.rQshiftBytes(3);
expect(shifted).to.be.an.instanceof(Uint8Array);
expect(shifted).to.array.equal(new Uint8Array(RQ_TEMPLATE.buffer, bef_rQi, 3));
- expect(sock.rQlen()).to.equal(bef_len - 3);
+ expect(sock.rQlen).to.equal(bef_len - 3);
});
it('should shift the entire rest of the queue off if no length is given', function () {
sock.rQshiftBytes();
- expect(sock.rQlen()).to.equal(0);
+ expect(sock.rQlen).to.equal(0);
});
});
describe('rQslice', function () {
beforeEach(function () {
- sock.set_rQi(0);
+ sock.rQi = 0;
});
it('should not modify the receive queue', function () {
- var bef_len = sock.rQlen();
+ const bef_len = sock.rQlen;
sock.rQslice(0, 2);
- expect(sock.rQlen()).to.equal(bef_len);
+ expect(sock.rQlen).to.equal(bef_len);
});
it('should return an array containing the given slice of the receive queue', function () {
- var sl = sock.rQslice(0, 2);
+ const sl = sock.rQslice(0, 2);
expect(sl).to.be.an.instanceof(Uint8Array);
expect(sl).to.array.equal(new Uint8Array(RQ_TEMPLATE.buffer, 0, 2));
});
it('should use the rest of the receive queue if no end is given', function () {
- var sl = sock.rQslice(1);
+ const sl = sock.rQslice(1);
expect(sl).to.have.length(RQ_TEMPLATE.length - 1);
expect(sl).to.array.equal(new Uint8Array(RQ_TEMPLATE.buffer, 1));
});
it('should take the current rQi in to account', function () {
- sock.set_rQi(1);
+ sock.rQi = 1;
expect(sock.rQslice(0, 2)).to.array.equal(new Uint8Array(RQ_TEMPLATE.buffer, 1, 2));
});
});
describe('rQwait', function () {
beforeEach(function () {
- sock.set_rQi(0);
+ sock.rQi = 0;
});
it('should return true if there are not enough bytes in the receive queue', function () {
});
it('should return true and reduce rQi by "goback" if there are not enough bytes', function () {
- sock.set_rQi(5);
+ sock.rQi = 5;
expect(sock.rQwait('hi', RQ_TEMPLATE.length, 4)).to.be.true;
- expect(sock.get_rQi()).to.equal(1);
+ expect(sock.rQi).to.equal(1);
});
it('should raise an error if we try to go back more than possible', function () {
- sock.set_rQi(5);
- expect(function () { sock.rQwait('hi', RQ_TEMPLATE.length, 6); }).to.throw(Error);
+ sock.rQi = 5;
+ expect(() => sock.rQwait('hi', RQ_TEMPLATE.length, 6)).to.throw(Error);
});
it('should not reduce rQi if there are enough bytes', function () {
- sock.set_rQi(5);
+ sock.rQi = 5;
sock.rQwait('hi', 1, 6);
- expect(sock.get_rQi()).to.equal(5);
+ expect(sock.rQi).to.equal(5);
});
});
it('should actually send on the websocket', function () {
sock._websocket.bufferedAmount = 8;
- sock._websocket.readyState = WebSocket.OPEN
+ sock._websocket.readyState = WebSocket.OPEN;
sock._sQ = new Uint8Array([1, 2, 3]);
sock._sQlen = 3;
- var encoded = sock._encode_message();
+ const encoded = sock._encode_message();
sock.flush();
expect(sock._websocket.send).to.have.been.calledOnce;
it('should add to the send queue', function () {
sock.send([1, 2, 3]);
- var sq = sock.get_sQ();
+ const sq = sock.sQ;
expect(new Uint8Array(sq.buffer, sock._sQlen - 3, 3)).to.array.equal(new Uint8Array([1, 2, 3]));
});
});
describe('lifecycle methods', function () {
- var old_WS;
+ let old_WS;
before(function () {
- old_WS = WebSocket;
+ old_WS = WebSocket;
});
- var sock;
+ let sock;
beforeEach(function () {
- sock = new Websock();
- WebSocket = sinon.spy();
- WebSocket.OPEN = old_WS.OPEN;
- WebSocket.CONNECTING = old_WS.CONNECTING;
- WebSocket.CLOSING = old_WS.CLOSING;
- WebSocket.CLOSED = old_WS.CLOSED;
-
- WebSocket.prototype.binaryType = 'arraybuffer';
+ sock = new Websock();
+ // eslint-disable-next-line no-global-assign
+ WebSocket = sinon.spy();
+ WebSocket.OPEN = old_WS.OPEN;
+ WebSocket.CONNECTING = old_WS.CONNECTING;
+ WebSocket.CLOSING = old_WS.CLOSING;
+ WebSocket.CLOSED = old_WS.CLOSED;
+
+ WebSocket.prototype.binaryType = 'arraybuffer';
});
describe('opening', function () {
- it('should pick the correct protocols if none are given' , function () {
+ it('should pick the correct protocols if none are given', function () {
});
});
after(function () {
+ // eslint-disable-next-line no-global-assign
WebSocket = old_WS;
});
});
describe('WebSocket Receiving', function () {
- var sock;
+ let sock;
beforeEach(function () {
- sock = new Websock();
- sock._allocate_buffers();
+ sock = new Websock();
+ sock._allocate_buffers();
});
it('should support adding binary Uint8Array data to the receive queue', function () {
- var msg = { data: new Uint8Array([1, 2, 3]) };
+ const msg = { data: new Uint8Array([1, 2, 3]) };
sock._mode = 'binary';
sock._recv_message(msg);
expect(sock.rQshiftStr(3)).to.equal('\x01\x02\x03');
it('should call the message event handler if present', function () {
sock._eventHandlers.message = sinon.spy();
- var msg = { data: new Uint8Array([1, 2, 3]).buffer };
+ const msg = { data: new Uint8Array([1, 2, 3]).buffer };
sock._mode = 'binary';
sock._recv_message(msg);
expect(sock._eventHandlers.message).to.have.been.calledOnce;
it('should not call the message event handler if there is nothing in the receive queue', function () {
sock._eventHandlers.message = sinon.spy();
- var msg = { data: new Uint8Array([]).buffer };
+ const msg = { data: new Uint8Array([]).buffer };
sock._mode = 'binary';
sock._recv_message(msg);
expect(sock._eventHandlers.message).not.to.have.been.called;
});
- it('should compact the receive queue', function () {
- // NB(sross): while this is an internal implementation detail, it's important to
- // test, otherwise the receive queue could become very large very quickly
+ it('should compact the receive queue when a message handler empties it', function () {
+ sock._eventHandlers.message = () => { sock.rQi = sock._rQlen; };
sock._rQ = new Uint8Array([0, 1, 2, 3, 4, 5, 0, 0, 0, 0]);
sock._rQlen = 6;
- sock.set_rQi(6);
- sock._rQmax = 3;
- var msg = { data: new Uint8Array([1, 2, 3]).buffer };
+ sock.rQi = 6;
+ const msg = { data: new Uint8Array([1, 2, 3]).buffer };
+ sock._mode = 'binary';
+ sock._recv_message(msg);
+ expect(sock._rQlen).to.equal(0);
+ expect(sock.rQi).to.equal(0);
+ });
+
+ it('should compact the receive queue when we reach the end of the buffer', function () {
+ sock._rQ = new Uint8Array(20);
+ sock._rQbufferSize = 20;
+ sock._rQlen = 20;
+ sock.rQi = 10;
+ const msg = { data: new Uint8Array([1, 2]).buffer };
sock._mode = 'binary';
sock._recv_message(msg);
- expect(sock._rQlen).to.equal(3);
- expect(sock.get_rQi()).to.equal(0);
+ expect(sock._rQlen).to.equal(12);
+ expect(sock.rQi).to.equal(0);
});
- it('should automatically resize the receive queue if the incoming message is too large', function () {
+ it('should automatically resize the receive queue if the incoming message is larger than the buffer', function () {
sock._rQ = new Uint8Array(20);
sock._rQlen = 0;
- sock.set_rQi(0);
+ sock.rQi = 0;
sock._rQbufferSize = 20;
- sock._rQmax = 2;
- var msg = { data: new Uint8Array(30).buffer };
+ const msg = { data: new Uint8Array(30).buffer };
sock._mode = 'binary';
sock._recv_message(msg);
expect(sock._rQlen).to.equal(30);
- expect(sock.get_rQi()).to.equal(0);
+ expect(sock.rQi).to.equal(0);
expect(sock._rQ.length).to.equal(240); // keep the invariant that rQbufferSize / 8 >= rQlen
});
+
+ it('should automatically resize the receive queue if the incoming message is larger than 1/8th of the buffer and we reach the end of the buffer', function () {
+ sock._rQ = new Uint8Array(20);
+ sock._rQlen = 16;
+ sock.rQi = 16;
+ sock._rQbufferSize = 20;
+ const msg = { data: new Uint8Array(6).buffer };
+ sock._mode = 'binary';
+ sock._recv_message(msg);
+ expect(sock._rQlen).to.equal(6);
+ expect(sock.rQi).to.equal(0);
+ expect(sock._rQ.length).to.equal(48);
+ });
});
describe('Data encoding', function () {
after(function () { FakeWebSocket.restore(); });
describe('as binary data', function () {
- var sock;
+ let sock;
beforeEach(function () {
sock = new Websock();
sock.open('ws://', 'binary');
it('should only send the send queue up to the send queue length', function () {
sock._sQ = new Uint8Array([1, 2, 3, 4, 5]);
sock._sQlen = 3;
- var res = sock._encode_message();
+ const res = sock._encode_message();
expect(res).to.array.equal(new Uint8Array([1, 2, 3]));
});