Make these 2 tests work both on 0.4 and 0.6

egmeek · Feb 27, 2012 · 204576c · 204576c
1 parent 66ac425
commit 204576c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/transports.websocket.test.js b/test/transports.websocket.test.js
@@ -131,7 +131,7 @@ module.exports = {
       var url = '/socket.io/' + sio.protocol + '/websocket/' + sid;
       var req = cl.get(url, {headers: headers}, function (res, data) {});
       var closed = false;
-      req.socket.on('close', function() {
+      (req.socket || req).on('close', function() {
         if (closed) return;
         closed = true;
         notConnected.should.be.true;
@@ -169,7 +169,7 @@ module.exports = {
       var url = '/socket.io/' + sio.protocol + '/websocket/' + sid;
       var req = cl.get(url, {headers: headers}, function (res, data) {});
       var closed = false;
-      req.socket.on('close', function() {
+      (req.socket || req).on('close', function() {
         if (closed) return;
         closed = true;
         notConnected.should.be.true;