noVNC/tests/fake.websocket.js

import Base64 from '../core/base64.js';

export default class FakeWebSocket {
    constructor(uri, protocols) {
        this.url = uri;
        this.binaryType = "arraybuffer";
        this.extensions = "";

        this.onerror = null;
        this.onmessage = null;
        this.onopen = null;

        if (!protocols || typeof protocols === 'string') {
            this.protocol = protocols;
        } else {
            this.protocol = protocols[0];
        }

        this._sendQueue = new Uint8Array(20000);

        this.readyState = FakeWebSocket.CONNECTING;
        this.bufferedAmount = 0;

        this._isFake = true;
    }

    close(code, reason) {
        this.readyState = FakeWebSocket.CLOSED;
        if (this.onclose) {
            this.onclose(new CloseEvent("close", { 'code': code, 'reason': reason, 'wasClean': true }));
        }
    }

    send(data) {
        if (this.protocol == 'base64') {
            data = Base64.decode(data);
        } else {
            data = new Uint8Array(data);
        }
        if (this.bufferedAmount + data.length > this._sendQueue.length) {
            let newlen = this._sendQueue.length;
            while (this.bufferedAmount + data.length > newlen) {
                newlen *= 2;
            }
            let newbuf = new Uint8Array(newlen);
            newbuf.set(this._sendQueue);
            this._sendQueue = newbuf;
        }
        this._sendQueue.set(data, this.bufferedAmount);
        this.bufferedAmount += data.length;
    }

    _getSentData() {
        const res = this._sendQueue.slice(0, this.bufferedAmount);
        this.bufferedAmount = 0;
        return res;
    }

    _open() {
        this.readyState = FakeWebSocket.OPEN;
        if (this.onopen) {
            this.onopen(new Event('open'));
        }
    }

    _receiveData(data) {
        if (data.length < 4096) {
            // Break apart the data to expose bugs where we assume data is
            // neatly packaged
            for (let i = 0;i < data.length;i++) {
                let buf = data.slice(i, i+1);
                this.onmessage(new MessageEvent("message", { 'data': buf.buffer }));
            }
        } else {
            this.onmessage(new MessageEvent("message", { 'data': data.buffer }));
        }
    }
}

FakeWebSocket.OPEN = WebSocket.OPEN;
FakeWebSocket.CONNECTING = WebSocket.CONNECTING;
FakeWebSocket.CLOSING = WebSocket.CLOSING;
FakeWebSocket.CLOSED = WebSocket.CLOSED;

FakeWebSocket._isFake = true;

FakeWebSocket.replace = () => {
    if (!WebSocket._isFake) {
        const realVersion = WebSocket;
        // eslint-disable-next-line no-global-assign
        WebSocket = FakeWebSocket;
        FakeWebSocket._realVersion = realVersion;
    }
};

FakeWebSocket.restore = () => {
    if (WebSocket._isFake) {
        // eslint-disable-next-line no-global-assign
        WebSocket = WebSocket._realVersion;
    }
};