test: split test in parallel/sequential

Reviewed-By: Ben Noordhuis <info@bnoordhuis.nl>
PR-URL: https://github.com/iojs/io.js/pull/172
Fix: iojs/io.js#139

1 files changed, 185 insertions, 0 deletions

diff --git a/test/parallel/test-http-upgrade-server.js b/test/parallel/test-http-upgrade-server.js
new file mode 100644
index 0000000000..b514eff3c9
--- /dev/null
+++ b/test/parallel/test-http-upgrade-server.js
@@ -0,0 +1,185 @@
+// Copyright Joyent, Inc. and other Node contributors.
+//
+// Permission is hereby granted, free of charge, to any person obtaining a
+// copy of this software and associated documentation files (the
+// "Software"), to deal in the Software without restriction, including
+// without limitation the rights to use, copy, modify, merge, publish,
+// distribute, sublicense, and/or sell copies of the Software, and to permit
+// persons to whom the Software is furnished to do so, subject to the
+// following conditions:
+//
+// The above copyright notice and this permission notice shall be included
+// in all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
+// NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
+// DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
+// OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
+// USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+var common = require('../common');
+var assert = require('assert');
+
+var util = require('util');
+var net = require('net');
+var http = require('http');
+
+
+var requests_recv = 0;
+var requests_sent = 0;
+var request_upgradeHead = null;
+
+function createTestServer() {
+  return new testServer();
+}
+
+function testServer() {
+  var server = this;
+  http.Server.call(server, function() {});
+
+  server.on('connection', function() {
+    requests_recv++;
+  });
+
+  server.on('request', function(req, res) {
+    res.writeHead(200, {'Content-Type': 'text/plain'});
+    res.write('okay');
+    res.end();
+  });
+
+  server.on('upgrade', function(req, socket, upgradeHead) {
+    socket.write('HTTP/1.1 101 Web Socket Protocol Handshake\r\n' +
+                 'Upgrade: WebSocket\r\n' +
+                 'Connection: Upgrade\r\n' +
+                 '\r\n\r\n');
+
+    request_upgradeHead = upgradeHead;
+
+    socket.on('data', function(d) {
+      var data = d.toString('utf8');
+      if (data == 'kill') {
+        socket.end();
+      } else {
+        socket.write(data, 'utf8');
+      }
+    });
+  });
+}
+
+util.inherits(testServer, http.Server);
+
+
+function writeReq(socket, data, encoding) {
+  requests_sent++;
+  socket.write(data);
+}
+
+
+/*-----------------------------------------------
+  connection: Upgrade with listener
+-----------------------------------------------*/
+function test_upgrade_with_listener(_server) {
+  var conn = net.createConnection(common.PORT);
+  conn.setEncoding('utf8');
+  var state = 0;
+
+  conn.on('connect', function() {
+    writeReq(conn,
+             'GET / HTTP/1.1\r\n' +
+             'Upgrade: WebSocket\r\n' +
+             'Connection: Upgrade\r\n' +
+             '\r\n' +
+             'WjN}|M(6');
+  });
+
+  conn.on('data', function(data) {
+    state++;
+
+    assert.equal('string', typeof data);
+
+    if (state == 1) {
+      assert.equal('HTTP/1.1 101', data.substr(0, 12));
+      assert.equal('WjN}|M(6', request_upgradeHead.toString('utf8'));
+      conn.write('test', 'utf8');
+    } else if (state == 2) {
+      assert.equal('test', data);
+      conn.write('kill', 'utf8');
+    }
+  });
+
+  conn.on('end', function() {
+    assert.equal(2, state);
+    conn.end();
+    _server.removeAllListeners('upgrade');
+    test_upgrade_no_listener();
+  });
+}
+
+/*-----------------------------------------------
+  connection: Upgrade, no listener
+-----------------------------------------------*/
+var test_upgrade_no_listener_ended = false;
+
+function test_upgrade_no_listener() {
+  var conn = net.createConnection(common.PORT);
+  conn.setEncoding('utf8');
+
+  conn.on('connect', function() {
+    writeReq(conn,
+             'GET / HTTP/1.1\r\n' +
+             'Upgrade: WebSocket\r\n' +
+             'Connection: Upgrade\r\n' +
+             '\r\n');
+  });
+
+  conn.on('end', function() {
+    test_upgrade_no_listener_ended = true;
+    conn.end();
+  });
+
+  conn.on('close', function() {
+    test_standard_http();
+  });
+}
+
+/*-----------------------------------------------
+  connection: normal
+-----------------------------------------------*/
+function test_standard_http() {
+  var conn = net.createConnection(common.PORT);
+  conn.setEncoding('utf8');
+
+  conn.on('connect', function() {
+    writeReq(conn, 'GET / HTTP/1.1\r\n\r\n');
+  });
+
+  conn.once('data', function(data) {
+    assert.equal('string', typeof data);
+    assert.equal('HTTP/1.1 200', data.substr(0, 12));
+    conn.end();
+  });
+
+  conn.on('close', function() {
+    server.close();
+  });
+}
+
+
+var server = createTestServer();
+
+server.listen(common.PORT, function() {
+  // All tests get chained after this:
+  test_upgrade_with_listener(server);
+});
+
+
+/*-----------------------------------------------
+  Fin.
+-----------------------------------------------*/
+process.on('exit', function() {
+  assert.equal(3, requests_recv);
+  assert.equal(3, requests_sent);
+  assert.ok(test_upgrade_no_listener_ended);
+});