fake.websocket.js

var FakeWebSocket;

(function () {
    // PhantomJS can't create Event objects directly, so we need to use this
    function make_event(name, props) {
        var evt = document.createEvent('Event');
        evt.initEvent(name, true, true);
        if (props) {
            for (var prop in props) {
                evt[prop] = props[prop];
            }
        }
        return evt;
    }

    FakeWebSocket = function (uri, protocols) {
        this.url = uri;
        this.binaryType = "arraybuffer";
        this.extensions = "";

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

        this._send_queue = new Uint8Array(20000);

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

        this.__is_fake = true;
    };

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

        send: function (data) {
            if (this.protocol == 'base64') {
                data = Base64.decode(data);
            } else {
                data = new Uint8Array(data);
            }
            this._send_queue.set(data, this.bufferedAmount);
            this.bufferedAmount += data.length;
        },

        _get_sent_data: function () {
            var res = new Uint8Array(this._send_queue.buffer, 0, this.bufferedAmount);
            this.bufferedAmount = 0;
            return res;
        },

        _open: function (data) {
            this.readyState = FakeWebSocket.OPEN;
            if (this.onopen) {
                this.onopen(make_event('open'));
            }
        },

        _receive_data: function (data) {
            this.onmessage(make_event("message", { 'data': data }));
        }
    };

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

    FakeWebSocket.__is_fake = true;

    FakeWebSocket.replace = function () {
        if (!WebSocket.__is_fake) {
            var real_version = WebSocket;
            WebSocket = FakeWebSocket;
            FakeWebSocket.__real_version = real_version;
        }
    };

    FakeWebSocket.restore = function () {
        if (WebSocket.__is_fake) {
            WebSocket = WebSocket.__real_version;
        }
    };
})();