fix: add null check for connection.headers

webpack · Aug 17, 2019 · 96bf165 · 96bf165
1 parent 2d44ef8
commit 96bf165
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/lib/servers/SockJSServer.js b/lib/servers/SockJSServer.js
@@ -64,7 +64,7 @@ module.exports = class SockJSServer extends BaseServer {
   // f should be passed the resulting connection and the connection headers
   onConnection(f) {
     this.socket.on('connection', (connection) => {
-      f(connection, connection.headers);
+      f(connection, connection ? connection.headers : null);
     });
   }
 

diff --git a/test/server/servers/SockJSServer.test.js b/test/server/servers/SockJSServer.test.js
@@ -10,7 +10,7 @@ describe('SockJSServer', () => {
   let socketServer;
   let listeningApp;
 
-  beforeAll((done) => {
+  beforeEach((done) => {
     // eslint-disable-next-line new-cap
     const app = new express();
 
@@ -84,9 +84,20 @@ describe('SockJSServer', () => {
         done();
       }, 3000);
     });
+
+    it('should not throw an exception when connection is null', (done) => {
+      expect(() => {
+        socketServer.onConnection(() => {});
+        socketServer.socket.emit('connection', null);
+      }).not.toThrow();
+
+      setTimeout(() => {
+        done();
+      }, 1000);
+    });
   });
 
-  afterAll((done) => {
+  afterEach((done) => {
     listeningApp.close(done);
   });
 });