From 5ed64be820dfda3ff7d39f0c1f60b35f62d0fa58 Mon Sep 17 00:00:00 2001
From: Joel Martin <github@martintribe.org>
Date: Tue, 3 Jan 2012 07:41:54 -0700
Subject: [PATCH] Switch tests/echo.html to use websock.js.

---
 tests/echo.html | 60 +++++++++++++------------------------------------
 tests/echo.py   |  2 +-
 2 files changed, 17 insertions(+), 45 deletions(-)
diff --git a/tests/echo.html b/tests/echo.html
index 795a5f6..f7269e0 100644
--- a/tests/echo.html
+++ b/tests/echo.html
@@ -2,9 +2,10 @@
 
     <head>
         <title>WebSockets Echo Test</title>
-        <script src="include/base64.js"></script>
         <script src="include/util.js"></script>
         <script src="include/webutil.js"></script> 
+        <script src="include/base64.js"></script>
+        <script src="include/websock.js"></script> 
         <!-- Uncomment to activate firebug lite -->
         <!--
         <script type='text/javascript' 
@@ -50,13 +51,8 @@
         }
 
         function send_msg() {
-            if (ws.bufferedAmount > 0) {
-                console.log("Delaying send");
-                return;
-            }
-            var str = "Message #" + send_cnt, arr = [];
-            arr.pushStr(str)
-            ws.send(Base64.encode(arr));
+            var str = "Message #" + send_cnt;
+            ws.send_string(str);
             message("Sent message: '" + str + "'");
             send_cnt++;
         }
@@ -67,11 +63,6 @@
             $D('errors').innerHTML = errors;
         }
 
-        function init_ws() {
-            console.log(">> init_ws");
-            console.log("<< init_ws");
-        }
-
         function connect() {
             var host = $D('host').value,
                 port = $D('port').value,
@@ -92,40 +83,38 @@
             }
             uri = scheme + host + ":" + port;
             message("connecting to " + uri);
-            ws = new WebSocket(uri, "base64");
 
-            ws.onmessage = function(e) {
+            ws = new Websock();
+            ws.open(uri);
+
+            ws.on('message', function(e) {
                 //console.log(">> WebSockets.onmessage");
-                var arr = Base64.decode(e.data), str = "", i;
-
-                for (i = 0; i < arr.length; i++) {
-                    str = str + String.fromCharCode(arr[i]);
-                }
+                var str = ws.rQshiftStr();
 
                 message("Received message '" + str + "'");
                 //console.log("<< WebSockets.onmessage");
-            };
-            ws.onopen = function(e) {
+            });
+            ws.on('open', function(e) {
                 console.log(">> WebSockets.onopen");
                 echo_ref = setInterval(send_msg, echoDelay);
                 console.log("<< WebSockets.onopen");
-            };
-            ws.onclose = function(e) {
+            });
+            ws.on('close', function(e) {
                 console.log(">> WebSockets.onclose");
                 if (echo_ref) {
                     clearInterval(echo_ref);
                     echo_ref = null;
                 }
                 console.log("<< WebSockets.onclose");
-            };
-            ws.onerror = function(e) {
+            });
+            ws.on('error', function(e) {
                 console.log(">> WebSockets.onerror");
                 if (echo_ref) {
                     clearInterval(echo_ref);
                     echo_ref = null;
                 }
                 console.log("<< WebSockets.onerror");
-            };
+            });
 
             $D('connectButton').value = "Stop";
             $D('connectButton').onclick = disconnect;
@@ -148,25 +137,8 @@
         }
 
 
-        /* If no builtin websockets then load web_socket.js */
-        if (window.WebSocket) {
-            VNC_native_ws = true;
-        } else {
-            VNC_native_ws = false;
-            console.log("Loading web-socket-js flash bridge");
-            var extra = "<script src='include/web-socket-js/swfobject.js'><\/script>";
-            extra += "<script src='include/web-socket-js/FABridge.js'><\/script>";
-            extra += "<script src='include/web-socket-js/web_socket.js'><\/script>";
-            document.write(extra);
-        }
-
         window.onload = function() {
             console.log("onload");
-            if (!VNC_native_ws) {
-                console.log("initializing web-socket-js flash bridge");
-                WebSocket.__swfLocation = "include/web-socket-js/WebSocketMain.swf";
-                WebSocket.__initialize();
-            }
             var url = document.location.href;
             $D('host').value = (url.match(/host=([^&#]*)/) || ['',''])[1];
             $D('port').value = (url.match(/port=([^&#]*)/) || ['',''])[1];
diff --git a/tests/echo.py b/tests/echo.py
index 878c31a..31f2fa9 100755
--- a/tests/echo.py
+++ b/tests/echo.py
@@ -45,7 +45,7 @@ class WebSocketEcho(WebSocketServer):
             if self.client in ins:
                 # Receive client data, decode it, and send it back
                 frames, closed = self.recv_frames()
-                cqueue.extend(frames)
+                cqueue.extend(["You said: " + f for f in frames])
 
                 if closed:
                     self.send_close()
