fix: use with babel-7 (#21)

ValeraS · remy · commit 893ff8f56f0d · 2018-12-01T16:51:35.000Z
diff --git a/lib/index.js b/lib/index.js
@@ -79,7 +79,8 @@ module.exports = (timeout = 100, extra = null) => {
   }
 
   return ({ types: t, transform }) => {
-    const node = extra ? transform(extra).ast.program.body[0] : null;
+    const node = extra ?
+      transform(extra, { ast: true }).ast.program.body[0] : null;
 
     let callback = null;
     if (t.isExpressionStatement(node)) {

Original file line number	Diff line number	Diff line change
`@@ -79,7 +79,8 @@ module.exports = (timeout = 100, extra = null) => {`
`79`	`79`	`}`
`80`	`80`
`81`	`81`	`return ({ types: t, transform }) => {`
`82`		`- const node = extra ? transform(extra).ast.program.body[0] : null;`
	`82`	`+ const node = extra ?`
	`83`	`+ transform(extra, { ast: true }).ast.program.body[0] : null;`
`83`	`84`
`84`	`85`	`let callback = null;`
`85`	`86`	`if (t.isExpressionStatement(node)) {`