test: fix helper to take options

iduranb · Dec 24, 2012 · 1ec7667 · 1ec7667
1 parent 67f7d57
commit 1ec7667
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/test/socket.io.js b/test/socket.io.js
@@ -6,10 +6,15 @@ var request = require('supertest');
 var expect = require('expect.js');
 
 // creates a socket.io client for the given server
-function client(srv, nsp){
+function client(srv, nsp, opts){
+  if ('object' == typeof nsp) {
+    opts = nsp;
+    nsp = null;
+  }
   var addr = srv.address();
   if (!addr) addr = srv.listen().address();
-  return ioc('ws://' + addr.address + ':' + addr.port + (nsp || ''));
+  var url = 'ws://' + addr.address + ':' + addr.port + (nsp || '');
+  return ioc(url, opts);
 }
 
 describe('socket.io', function(){