devCrossNet
diff --git a/‎addon/ng2/tasks/test.js
Lines changed: 13 additions & 0 deletions b/‎addon/ng2/tasks/test.js
Lines changed: 13 additions & 0 deletions
diff --git a/‎tests/e2e/e2e_workflow.spec.js
Lines changed: 11 additions & 4 deletions b/‎tests/e2e/e2e_workflow.spec.js
Lines changed: 11 additions & 4 deletions
@@ -4,6 +4,7 @@
 var Promise = require('ember-cli/lib/ext/promise');
 var Task = require('ember-cli/lib/models/task');
 var path = require('path');
+var CoverageTask = require('./coverage');
 
 // require dependencies within the target project
 function requireDependency(root, moduleName) {
@@ -16,6 +17,12 @@ module.exports = Task.extend({
 
   run: function (options) {
     var projectRoot = this.project.root;
+    var coverageTask = new CoverageTask({
+      ui: this.ui,
+      analytics: this.analytics,
+      project: this.project
+    });
+
     return new Promise((resolve) => {
       var karma = requireDependency(projectRoot, 'karma');
       var karmaConfig = path.join(projectRoot, this.project.ngConfig.test.karma.config);
@@ -27,6 +34,12 @@ module.exports = Task.extend({
       options.configFile = karmaConfig;
       var karmaServer = new karma.Server(options, resolve);
 
+      karmaServer.on('run_complete', function () {
+        setTimeout(function () {
+          coverageTask.run();
+        }, 500);
+      });
+
       karmaServer.start();
     });
   }
 
@@ -113,10 +113,7 @@ describe('Basic end-to-end Workflow', function () {
       const exitCode = typeof result === 'object' ? result.exitCode : result;
 
       expect(exitCode).to.be.equal(0);
-
-        /**
-         * coverage tests
-         */
+      
       expect(existsSync(coverageDir)).to.be.equal(true);
       expect(existsSync(path.join(coverageDir, 'coverage-final.json'))).to.be.equal(true);
       expect(existsSync(path.join(coverageDir, 'html'))).to.be.equal(true);
@@ -203,8 +200,12 @@ describe('Basic end-to-end Workflow', function () {
     this.timeout(420000);
 
     return ng(testArgs).then(function (result) {
+      var coverageDir = path.join(process.cwd(), 'coverage');
       const exitCode = typeof result === 'object' ? result.exitCode : result;
       expect(exitCode).to.be.equal(0);
+      
+      expect(existsSync(path.join(coverageDir, 'html', 'app', 'test-component'))).to.be.equal(true);
+      expect(existsSync(path.join(coverageDir, 'html', 'app', 'test-component', 'test-component.component.ts.html'))).to.be.equal(true);
     });
   });
 
@@ -221,8 +222,11 @@ describe('Basic end-to-end Workflow', function () {
     this.timeout(420000);
 
     return ng(testArgs).then(function (result) {
+      var coverageDir = path.join(process.cwd(), 'coverage');
       const exitCode = typeof result === 'object' ? result.exitCode : result;
       expect(exitCode).to.be.equal(0);
+      
+      expect(existsSync(path.join(coverageDir, 'html', 'app', 'test-service.service.ts.html'))).to.be.equal(true);
     });
   });
 
@@ -239,8 +243,11 @@ describe('Basic end-to-end Workflow', function () {
     this.timeout(420000);
 
     return ng(testArgs).then(function (result) {
+      var coverageDir = path.join(process.cwd(), 'coverage');
       const exitC
5C8A
ode = typeof result === 'object' ? result.exitCode : result;
       expect(exitCode).to.be.equal(0);
+
+      expect(existsSync(path.join(coverageDir, 'html', 'app', 'test-pipe.pipe.ts.html'))).to.be.equal(true);
     });
   });