noVNC 1.0.0

[mirror_novnc.git] / tests / fake.websocket.js

diff --git a/tests/fake.websocket.js b/tests/fake.websocket.js
index 749c0eafc80a89a48f0566723aa6dd403b54423e..eb4f203dc5c8a33571ef7921389ee63a5ac53ec9 100644 (file)
--- a/tests/fake.websocket.js
+++ b/tests/fake.websocket.js
@@ -1,96 +1,87 @@
-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];
-            }
+// 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;
+    }
+    return evt;
+}
+
+export default function FakeWebSocket (uri, protocols) {
+    this.url = uri;
+    this.binaryType = "arraybuffer";
+    this.extensions = "";
+
+    if (!protocols || typeof protocols === 'string') {
+        this.protocol = protocols;
+    } else {
+        this.protocol = protocols[0];
     }
 
-    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 }));
-            }
-        },
+    this._send_queue = new Uint8Array(20000);
 
-        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;
-        },
+    this.readyState = FakeWebSocket.CONNECTING;
+    this.bufferedAmount = 0;
 
-        _get_sent_data: function () {
-            var arr = [];
-            for (var i = 0; i < this.bufferedAmount; i++) {
-                arr[i] = this._send_queue[i];
-            }
+    this.__is_fake = true;
+};
 
-            this.bufferedAmount = 0;
+FakeWebSocket.prototype = {
+    close: function (code, reason) {
+        this.readyState = FakeWebSocket.CLOSED;
+        if (this.onclose) {
+            this.onclose(make_event("close", { 'code': code, 'reason': reason, 'wasClean': true }));
+        }
+    },
 
-            return arr;
-        },
+    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;
+    },
 
-        _open: function (data) {
-            this.readyState = FakeWebSocket.OPEN;
-            if (this.onopen) {
-                this.onopen(make_event('open'));
-            }
-        },
+    _get_sent_data: function () {
+        var res = new Uint8Array(this._send_queue.buffer, 0, this.bufferedAmount);
+        this.bufferedAmount = 0;
+        return res;
+    },
 
-        _receive_data: function (data) {
-            this.onmessage(make_event("message", { 'data': data }));
+    _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.OPEN = WebSocket.OPEN;
+FakeWebSocket.CONNECTING = WebSocket.CONNECTING;
+FakeWebSocket.CLOSING = WebSocket.CLOSING;
+FakeWebSocket.CLOSED = WebSocket.CLOSED;
 
-    FakeWebSocket.__is_fake = true;
+FakeWebSocket.__is_fake = true;
 
-    FakeWebSocket.replace = function () {
-        if (!WebSocket.__is_fake) {
-            var real_version = WebSocket;
-            WebSocket = FakeWebSocket;
-            FakeWebSocket.__real_version = real_version;
-        }
-    };
+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;
-        }
-    };
-})();
+FakeWebSocket.restore = function () {
+    if (WebSocket.__is_fake) {
+        WebSocket = WebSocket.__real_version;
+    }
+};