# HG changeset patch
# User Olli Pettay <Olli.Pettay@helsinki.fi>
# Date 1276848826 -10800
# Node ID bdce66d818cbf825246751487493180f50c55efd
# Parent  1be59c7b12744ce4fe898e38fd88295b71bdd718
Bug 562681, add few more tests

diff --git a/content/base/test/test_ws_basic_tests.html b/content/base/test/test_ws_basic_tests.html
--- a/content/base/test/test_ws_basic_tests.html
+++ b/content/base/test/test_ws_basic_tests.html
@@ -31,7 +31,7 @@
   }
   ws.onclose = function(e) {
     is(results.length, 0, "All the messages should have been processed!");
-    SimpleTest.finish();
+    testWebSocket2();
   }
   ws.onerror = function(e) {
     ok(false, "onerror called!");
@@ -51,7 +51,7 @@
   var testMessage = "test message";
   ws.onopen = function(e) {
     for (var i = 0; i < testCount; ++i) {
-      ws.send(testMessage);
+      ws.send(testMessage + (i + 1));
     }
     ws.send("end");
   }
@@ -64,8 +64,8 @@
     SimpleTest.finish();
   }
   ws.onmessage = function(e) {
-    is(e.data, testMessage, "Wrong message");
     ++messageCount;
+    is(e.data, testMessage + messageCount, "Wrong message");
     document.getElementById('log').textContent = messageCount;
     if (messageCount == testCount) {
       this.onmessage = null;
@@ -73,6 +73,26 @@
   }
 }
 
+function testWebSocket3() {
+  ws = new WebSocket("ws://mochi.test:8888/tests/content/base/test/file_ws_basic_tests", "test");
+  var longString = new Array(10000).join("-huge websocket message");
+  ws.onopen = function(e) {
+    ws.send(longString);
+  }
+  ws.onclose = function(e) {
+    SimpleTest.finish();
+  }
+  ws.onerror = function(e) {
+    ok(false, "onerror called!");
+    SimpleTest.finish();
+  }
+  ws.onmessage = function(e) {
+    is(e.data, longString, "Didn't get the huge message back!");
+    document.getElementById('log').textContent += "\nReceived the huge message";
+    this.close();
+  }
+}
+
 SimpleTest.waitForExplicitFinish();
 
 </script>