Added test case for #13

Author: sitegui

test/test.js

@@ -1,4 +1,4 @@
-/*globals describe, before, it*/
+/*globals describe, before, it, after*/
 'use strict'
 
 require('should')
@@ -17,6 +17,11 @@ describe('text frames', function () {
 		})
 	})
 
+	after(function (done) {
+		testClient.close()
+		testServer.socket.close(done)
+	})
+
 	it('should connect to a websocket server', function (done) {
 		var client = getClient()
 
@@ -115,6 +120,31 @@ describe('text frames', function () {
 	})
 })
 
+describe('handshake', function () {
+	before(function (done) {
+		testServer = ws.createServer(function (conn) {
+			testConn = conn
+			// Send frame right after handshake answer
+			conn.sendText('hello')
+		}).listen(TEST_PORT, done)
+	})
+
+	after(function (done) {
+		testServer.socket.close(done)
+	})
+
+	it('should work when the handshake response is followed by a WS frame', function (done) {
+		// Server ready, make the first connection
+		ws.connect('ws://127.0.0.1:' + TEST_PORT, function () {
+			this.on('text', function (str) {
+				str.should.be.equal('hello')
+				this.close()
+				done()
+			})
+		})
+	})
+})
+
 function getClient() {
 	testClient.removeAllListeners()
 	return testClient