crabapples
/
noVNC
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add tests for base64 websocket protocol

This commit is contained in:
Cyrille Bourgois 2019-09-19 16:14:00 +02:00
parent 082223fdca
commit ad8615ba35
No known key found for this signature in database
GPG Key ID: 6CE60F56870D5FD7
1 changed files with 33 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
tests/test.websock.js
View File

@ -358,20 +358,30 @@ describe('Websock', function () {
let sock;
beforeEach(function () {
sock = new Websock();
sock._websocket = {
protocol: 'binary'
};
sock._allocate_buffers();
});
it('should support adding binary Uint8Array data to the receive queue', function () {
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 support adding base64 data to the receive queue', function () {
const msg = { data: 'MTIz' };
sock._websocket = {
protocol: 'base64'
};
sock._recv_message(msg);
expect(sock.rQshiftStr(3)).to.equal('123');
});
it('should call the message event handler if present', function () {
sock._eventHandlers.message = sinon.spy();
const msg = { data: new Uint8Array([1, 2, 3]).buffer };
sock._mode = 'binary';
sock._recv_message(msg);
expect(sock._eventHandlers.message).to.have.been.calledOnce;
});
@ -379,7 +389,6 @@ describe('Websock', function () {
it('should not call the message event handler if there is nothing in the receive queue', function () {
sock._eventHandlers.message = sinon.spy();
const msg = { data: new Uint8Array([]).buffer };
sock._mode = 'binary';
sock._recv_message(msg);
expect(sock._eventHandlers.message).not.to.have.been.called;
});
@ -392,7 +401,6 @@ describe('Websock', function () {
sock.rQi = 6;
sock._rQmax = 3;
const msg = { data: new Uint8Array([1, 2, 3]).buffer };
sock._mode = 'binary';
sock._recv_message(msg);
expect(sock._rQlen).to.equal(3);
expect(sock.rQi).to.equal(0);
@ -405,7 +413,6 @@ describe('Websock', function () {
sock._rQbufferSize = 20;
sock._rQmax = 2;
const msg = { data: new Uint8Array(30).buffer };
sock._mode = 'binary';
sock._recv_message(msg);
expect(sock._rQlen).to.equal(30);
expect(sock.rQi).to.equal(0);
@ -437,5 +444,26 @@ describe('Websock', function () {
expect(sock._websocket._get_sent_data()).to.array.equal(new Uint8Array([1, 2, 3]));
});
});
describe('as base64 data', function () {
let sock;
beforeEach(function () {
sock = new Websock();
sock.open('ws://', 'base64');
sock._websocket._open();
});
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;
const res = sock._encode_message();
expect(res).to.array.equal('AQID');
});
it('should properly pass the encoded data off to the actual WebSocket', function () {
sock.send([1, 2, 3]);
expect(sock._websocket._get_sent_data()).to.array.equal(new Uint8Array([1, 2, 3]));
});
});
});
});