fix(mocha): fix server not closing

Author: Awk34

templates/app/mocha.global.js

@@ -1,8 +1,19 @@
 import app from './';<% if (filters.mongoose) { %>
 import mongoose from 'mongoose';<% } %>
 
-after(function(done) {
-  app.angularFullstack.on('close', () => done());<% if (filters.mongoose) { %>
-  mongoose.connection.close();<% } %>
-  app.angularFullstack.close();
+after(function() {
+  return Promise.all([
+    // Add any promises here for processes that need to be closed before the tests can finish
+    <% if (filters.mongoose) { %>
+    new Promise(resolve => {
+      mongoose.connection.close(resolve);
+    }),<% } %>
+    new Promise(resolve => {
+      app.angularFullstack.close(resolve);
+    }),
+    <%_ if(filters.ws) { -%>
+    new Promise(resolve => {
+      app.primus.end(resolve);
+    }),<% } %>
+  ]);
 });

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);

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);
     });
   });
 }