fix(mocha): fix server not closing

Author: Awk34

Diff for: templates/app/mocha.global.js

@@ -1,8 +1,22 @@
 import app from './';<% if (filters.mongoose) { %>
 import mongoose from 'mongoose';<% } %>
 
+<%_ if(!filters.ws) { -%>
 after(function(done) {
   app.angularFullstack.on('close', () => done());<% if (filters.mongoose) { %>
   mongoose.connection.close();<% } %>
   app.angularFullstack.close();
-});
+  app.primus.end();
+});<% } %>
+<%_ if(filters.ws) { -%>
+let count = 0;
+after(function(done) {
+  app.angularFullstack.on('close', () => {
+    if(++count === 2) done();
+  });
+  mongoose.connection.close();
+  app.angularFullstack.close();
+  app.primus.end(() => {
+    if(++count === 2) done();
+  });
+});<% } %>

Diff for: templates/app/server/app.js

@@ -45,14 +45,20 @@ function startServer() {
 <% if(filters.sequelize) { %>
 sqldb.sequelize.sync()
   <%_ if(filters.ws) { -%>
-  .then(wsInitPromise)<% } %>
+  .then(wsInitPromise)
+  .then(primus => {
+    app.primus = primus;
+  })<% } %>
   .then(startServer)
   .catch(err => {
     console.log('Server failed to start due to error: %s', err);
   });
 <% } else { %>
 <%_ if(filters.ws) { -%>
 wsInitPromise
+  .then(primus => {
+    app.primus = primus;
+  })
   .then(startServer)
   .catch(err => {
     console.log('Server failed to start due to error: %s', err);

Diff for: templates/app/server/config/websockets(ws).js

@@ -53,7 +53,7 @@ export default function initWebSocketServer(server) {
     primus.save(path.join(__dirname, '../../client/components/socket/primus.js'), err => {
       if(err) return reject(err);
 
-      resolve();
+      resolve(primus);
     });
   });
 }