Better ES2015 Support

.babelrc

@@ -1,3 +1,7 @@
 {
-  "presets": ["es2015"]
+  "presets": ["es2015"],
+  "plugins": [
+    "transform-async-to-generator",
+    "transform-runtime"
+  ]
 }

package.json

@@ -89,6 +89,10 @@
   },
   "devDependencies": {
     "any-path": "^1.3.0",
+    "babel-core": "^6.9.1",
+    "babel-plugin-transform-async-to-generator": "^6.8.0",
+    "babel-plugin-transform-runtime": "^6.9.0",
+    "babel-preset-es2015": "^6.9.0",
     "chai": "^3.0.0",
     "coveralls": "^2.11.4",
     "exists-sync": "0.0.3",

test/fixtures/es2015/asyncawait.js

@@ -0,0 +1,13 @@
+const foo = async (opts, opts2) => {
+  let snuh = await bar()
+}
+
+const bar = () => {
+  return new Promise(function (resolve, reject) {
+    setTimeout(() => {
+      resolve(33)
+    }, 10)
+  })
+}
+
+foo()

test/fixtures/package.json

@@ -10,7 +10,8 @@
   "nyc": {
     "exclude": [
       "**/blarg",
-      "**/blerg"
+      "**/blerg",
+      "**/es2015"
     ]
   }
 }

test/src/es2015.js

@@ -0,0 +1,20 @@
+/* global describe, it */
+
+var NYC
+
+try {
+  NYC = require('../../index.covered.js')
+} catch (e) {
+  NYC = require('../../')
+}
+
+(new NYC()).wrap()
+
+require('babel-core/register')
+require('tap').mochaGlobals()
+
+describe('es2015 coverage', function () {
+  it('covers async/await branch', function () {
+    require('../fixtures/es2015/asyncawait')
+  })
+})

test/src/nyc-test.js

@@ -69,7 +69,7 @@ describe('nyc', function () {
         cwd: path.resolve(__dirname, '../fixtures')
       })
 
-      nyc.exclude.length.should.eql(5)
+      nyc.exclude.length.should.eql(7)
     })
 
     it("loads 'extension' patterns from package.json#nyc", function () {