Fix test.

gregberge · Sep 3, 2014 · eab3077 · eab3077
1 parent 7f63bc2
commit eab3077
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test.js b/test.js
@@ -98,10 +98,13 @@ describe('WebSocket', function () {
     });
 
     describe('#close', function () {
-      it('should send and receive message', function () {
+      it('should close the socket', function (done) {
         ws.connect({url: 'ws://echo.websocket.org'});
         ws.close();
-        expect(ws.getReadyState()).to.equal(3);
+        ws.on('close', function () {
+          expect(ws.getReadyState()).to.equal(3);
+          done();
+        });
       });
     });
   });