gecko-dev/content/base/test/test_websocket_basic.html

256 lines
7.9 KiB
HTML

<!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/content/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/content/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>