diff --git a/test/server.js b/test/server.js index c74f831..15f8fe9 100644 --- a/test/server.js +++ b/test/server.js @@ -14,12 +14,17 @@ describe('PeerServer', function() { describe('#_checkKey', function() { var p; - before(function() { + before(function(done) { PeerServer.prototype._initializeHTTP = sinon.stub(); - p = PeerServer({ port: 8000 }); + p = PeerServer({ port: 8000 }, done); p._checkKey('peerjs', 'myip', function() {}); }); + after(function(done) { + p._server.close(done); + }); + + it('should reject keys that are not the default', function(done) { p._checkKey('bad key', null, function(response) { expect(response).to.be('Invalid key provided'); @@ -70,9 +75,9 @@ describe('PeerServer', function() { describe('#_removePeer', function() { var p; - before(function() { + before(function(done) { PeerServer.prototype._initializeHTTP = sinon.stub(); - p = PeerServer({ port: 8000 }); + p = new PeerServer({ port: 8000 }, done); var fake = {ip: '0.0.0.0'}; p._ips[fake.ip] = 1; @@ -80,6 +85,10 @@ describe('PeerServer', function() { p._clients['peerjs']['test'] = fake; }); + after(function(done) { + p._server.close(done); + }); + it('should decrement the number of ips being used and remove the connection', function() { expect(p._ips['0.0.0.0']).to.be(1); p._removePeer('peerjs', 'test'); @@ -92,12 +101,16 @@ describe('PeerServer', function() { var p; var KEY = 'peerjs'; var ID = 'test'; - before(function() { + before(function(done) { PeerServer.prototype._initializeHTTP = sinon.stub(); - p = PeerServer({ port: 8000 }); + p = PeerServer({ port: 8000 }, done); p._clients[KEY] = {}; }); + after(function(done) { + p._server.close(done); + }); + it('should send to the socket when appropriate', function() { var send = sinon.spy(); var write = sinon.spy(); @@ -177,9 +190,12 @@ describe('PeerServer', function() { describe('#_generateClientId', function() { var p; - before(function() { - PeerServer.prototype._initializeHTTP = sinon.stub(); - p = PeerServer({ port: 8000 }); + before(function(done) { + p = new PeerServer({ port: 8000 }, done); + }); + + after(function(done) { + p._server.close(done); }); it('should generate a 16-character ID', function() {