Merge pull request #835 from embark-framework/fix-coverage-timing

Update coverage report after suite

lib/modules/coverage/index.js

@@ -1,6 +1,4 @@
 /*global web3*/
-const process = require('process');
-
 const fs = require('../../core/fs');
 const ContractSources = require('./contract_sources');
 
@@ -20,16 +18,16 @@ class CodeCoverage {
     embark.events.on('contracts:run:solc', this.runSolc.bind(this));
     embark.events.on('block:header', this.runSolc.bind(this));
 
+    // These events are emitted from a test-specific Embark instance, so we need to
+    // pull it in from global.
+    global.embark.events.on('tests:finished', this.updateCoverageReport.bind(this));
+
     this.seenTransactions = {};
     this.coverageReport = {};
     this.contractSources = new ContractSources([]);
 
-    var self = this;
+    this.dotEmbarkPath = fs.dappPath('.embark');
-
+    this.coverageReportPath = fs.dappPath('.embark', 'coverage.json');
-    process.on('exit', (_code) => {
-      const coverageReportPath = fs.dappPath('.embark', 'coverage.json');
-      fs.writeFileSync(coverageReportPath, JSON.stringify(self.coverageReport));
-    });
   }
 
   compileSolc(input) {
@@ -42,6 +40,12 @@ class CodeCoverage {
     this.contractSources.parseSolcOutput(output);
   }
 
+  updateCoverageReport(cb) {
+    fs.mkdirp(this.dotEmbarkPath, () => {
+      fs.writeFile(this.coverageReportPath, JSON.stringify(this.coverageReport), cb);
+    });
+  }
+
   async runSolc(receipt) {
     let block = await web3.eth.getBlock(receipt.number);
     if(block.transactions.length == 0) return;

lib/tests/run_tests.js

@@ -130,6 +130,7 @@ module.exports = {
           return next();
         }
 
+        global.embark.events.emit('tests:finished', function() {
           runCmd(`${fs.embarkPath('node_modules/.bin/istanbul')} report --root .embark --format html`,
             {silent: false, exitOnError: false}, (err) => {
               if (err) {
@@ -142,6 +143,7 @@ module.exports = {
                 .then(() => new Promise(resolve => setTimeout(resolve, 1000)))
                 .then(_next, _next);
             });
+        });
       }
     ], (err) => {
       if (err) {