1 files changed, 255 insertions, 0 deletions
diff --git a/dom/base/test/test_websocket_basic.html b/dom/base/test/test_websocket_basic.html
new file mode 100644
index 000000000..e3269382e
--- /dev/null
+++ b/dom/base/test/test_websocket_basic.html
@@ -0,0 +1,255 @@
+<!DOCTYPE HTML>
+<html>
+<head>
+  <title>Basic WebSocket test</title>
+  <script type="text/javascript" src="/tests/SimpleTest/SimpleTest.js"></script>
+  <link rel="stylesheet" type="text/css" href="/tests/SimpleTest/test.css" />
+</head>
+
+<body onload="testWebSocket()">
+<a target="_blank" href="https://bugzilla.mozilla.org/show_bug.cgi?id=472529">Mozilla Bug 472529</a>
+<p id="display"></p>
+<div id="content" style="display: none">
+</div>
+<pre id="test">
+<script class="testbody" type="text/javascript">
+
+const kUrl = "ws://mochi.test:8888/tests/dom/base/test/file_websocket_basic";
+
+var gTestElement;
+var ws;
+
+function forcegc() {
+  SpecialPowers.forceGC();
+  SpecialPowers.gc();
+}
+
+function testWebSocket() {
+  gTestElement = document.getElementById("test");
+
+  SimpleTest.executeSoon(testWebSocket1);
+}
+
+/**
+ * Sends message keywords, then receives their values.
+ */
+function testWebSocket1() {
+  gTestElement.textContent = "Running testWebSocket1()";
+
+  var results = ["test",
+                 "/tests/dom/base/test/file_websocket_basic",
+                 "http://mochi.test:8888",
+                 "end"];
+
+  ws = new WebSocket(kUrl, "test");
+  is(ws.url, kUrl, "[1] WebSocket.url");
+  ws.onopen = function(e) {
+    const params = ["protocol", "resource", "origin", "end"];
+
+    gTestElement.textContent += "\nSending :";
+    for (var i = 0; i < params.length; ++i) {
+      gTestElement.textContent += " " + params[i];
+      ws.send(params[i]);
+    }
+
+    // Set this before onmessage() is called, so it is displayed once only.
+    gTestElement.textContent += "\nReceived:";
+  };
+  ws.onclose = function(e) {
+    is(results.length, 0, "[1] Number of unreceived messages");
+    ok(e.wasClean, "[1] Connection closed cleanly");
+
+    SimpleTest.executeSoon(testWebSocket2);
+  };
+  ws.onerror = function(e) {
+    ok(false, "[1] onerror() should not have been called!");
+    gTestElement.textContent += "\nonerror() should not have been called!";
+    SimpleTest.executeSoon(SimpleTest.finish);
+  };
+  ws.onmessage = function(e) {
+    is(e.data, results[0], "[1] Received message");
+    gTestElement.textContent += " " + e.data;
+    results.shift();
+  };
+}
+
+/**
+ * Sends 1000+1 test messages, then receives them.
+ */
+function testWebSocket2() {
+  gTestElement.textContent = "Running testWebSocket2()";
+
+  const displayInterval = 100;
+  const testCount = 1000;
+  const testMessage = "test message 2.";
+
+  var messageCount = 0;
+
+  ws = new WebSocket(kUrl, "test");
+  ws.onopen = function(e) {
+    gTestElement.textContent += "\nSending :";
+    for (var i = 1; i <= testCount; ++i) {
+      if (i % displayInterval == 1) {
+        gTestElement.textContent += " " + i;
+      }
+      ws.send(testMessage + i);
+    }
+    gTestElement.textContent += " end";
+    ws.send("end");
+
+    // Set this before onmessage() is called, so it is displayed once only.
+    gTestElement.textContent += "\nReceived:";
+  };
+  ws.onclose = function(e) {
+    is(messageCount, testCount + 1, "[2] Number of received messages");
+    ok(e.wasClean, "[2] Connection closed cleanly");
+
+    SimpleTest.executeSoon(testWebSocket3);
+  };
+  ws.onerror = function(e) {
+    ok(false, "[2] onerror() should not have been called!");
+    gTestElement.textContent += "\nonerror() should not have been called!";
+    SimpleTest.executeSoon(SimpleTest.finish);
+  };
+  ws.onmessage = function(e) {
+    ++messageCount;
+    if (messageCount > testCount)
+      is(e.data, "end", "[2] Received message");
+    else
+      is(e.data, testMessage + messageCount, "[2] Received message");
+    if (messageCount % displayInterval == 1) {
+      gTestElement.textContent += " " + messageCount;
+    }
+  };
+}
+
+/**
+ * Sends testcount+1 test messages, then receives them, calling forcegc() at each step.
+ */
+function testWebSocket3() {
+  gTestElement.textContent = "Running testWebSocket3() [can take a little while]";
+
+  const displayInterval = 10;
+  const testCount = 10;
+  const testMessage = "test message 3.";
+
+  var messageCount = 0;
+
+  ws = new WebSocket(kUrl, "test");
+  // Set this before onopen() is called,
+  // otherwise its display would be delayed by forcegc() calls...
+  gTestElement.textContent += "\nSending :";
+  ws.onopen = function(e) {
+    for (var i = 1; i <= testCount; ++i) {
+      forcegc();
+      if (i % displayInterval == 1) {
+        // Actual display is delayed by forcegc() calls...
+        gTestElement.textContent += " " + i;
+      }
+      ws.send(testMessage + i);
+    }
+    forcegc();
+    gTestElement.textContent += " end";
+    ws.send("end");
+
+    // Set this before onmessage() is called, so it is displayed once only.
+    gTestElement.textContent += "\nReceived:";
+  };
+  ws.onclose = function(e) {
+    is(messageCount, testCount + 1, "[3] Number of received messages");
+    ok(e.wasClean, "[3] Connection closed cleanly");
+
+    SimpleTest.executeSoon(testWebSocket4);
+  };
+  ws.onerror = function(e) {
+    ok(false, "[3] onerror() should not have been called!");
+    gTestElement.textContent += "\nonerror() should not have been called!";
+    SimpleTest.executeSoon(SimpleTest.finish);
+  };
+  ws.onmessage = function(e) {
+    forcegc();
+    ++messageCount;
+    if (messageCount > testCount)
+      is(e.data, "end", "[3] Received message");
+    else
+      is(e.data, testMessage + messageCount, "[3] Received message");
+    if (messageCount % displayInterval == 1) {
+      // Actual display is delayed by forcegc() call(s)...
+      gTestElement.textContent += " " + messageCount;
+    }
+  };
+}
+
+/**
+ * Sends a huge test message, then receives it, then closes the WebSocket from client-side.
+ */
+function testWebSocket4() {
+  gTestElement.textContent = "Running testWebSocket4()";
+
+  // String length = 13 + ((10,000 - 1) * 26) + 11 = 259,998 = almost 254 KiB.
+  const longString = "messageStart " + new Array(10000).join(" -huge WebSocket message- ") + " messageEnd";
+
+  ws = new WebSocket(kUrl, "test");
+  ws.onopen = function(e) {
+    is(this, ws, "[4, onopen()] 'this' should point to the WebSocket.");
+    gTestElement.textContent += "\nSending the huge message";
+    ws.send(longString);
+  };
+  ws.onclose = function(e) {
+    is(this, ws, "[4, onclose()] 'this' should point to the WebSocket.");
+    ok(e.wasClean, "[4] Connection closed cleanly");
+
+    SimpleTest.executeSoon(testWebSocket5);
+  };
+  ws.onerror = function(e) {
+    is(this, ws, "[4, onerror()] 'this' should point to the WebSocket.");
+    ok(false, "[4, onerror()] should not have been called!");
+    gTestElement.textContent += "\nonerror() should not have been called!";
+    SimpleTest.executeSoon(SimpleTest.finish);
+  };
+  ws.onmessage = function(e) {
+    is(this, ws, "[4, onmessage()] 'this' should point to the WebSocket.");
+    // Do not use |is(e.data, longString, "...");| that results in a _very_ long line.
+    is(e.data.length, longString.length, "[4] Length of received message");
+    ok(e.data == longString, "[4] Content of received message");
+    gTestElement.textContent += "\nReceived the huge message";
+    this.close();
+  };
+}
+
+/**
+ * Closes the WebSocket from client-side, then sends a test message that should be buffered.
+ */
+function testWebSocket5() {
+  gTestElement.textContent = "Running testWebSocket5()";
+
+  ws = new WebSocket(kUrl, "test");
+  ws.onopen = function(e) {
+    is(this.bufferedAmount, 0, "[5] Length of empty buffer before closing");
+    this.close();
+  };
+  ws.onclose = function(e) {
+    ok(e.wasClean, "[5] Connection closed cleanly");
+    is(this.bufferedAmount, 0, "[5] Length of empty buffer after closing");
+
+    var msg = "test message to be buffered";
+    this.send(msg);
+    is(this.bufferedAmount, msg.length, "[5] Length of buffered message sent after closing");
+
+    gTestElement.textContent += "\ntestWebSocket5() completed";
+
+    SimpleTest.executeSoon(SimpleTest.finish);
+  };
+  ws.onerror = function(e) {
+    ok(false, "[5] onerror() should not have been called!");
+    gTestElement.textContent += "\nonerror() should not have been called!";
+    SimpleTest.executeSoon(SimpleTest.finish);
+  };
+}
+
+SimpleTest.waitForExplicitFinish();
+
+</script>
+</pre>
+</body>
+</html>