fix: address regression discussed in istanbuljs/babel-plugin-istanbul#78

(#40)
istanbuljs-archived-repos · Jan 4, 2017 · 7f458a3 · 7f458a3
1 parent a7286b0
commit 7f458a3
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/visitor.js b/src/visitor.js
@@ -145,9 +145,12 @@ class VisitState {
         } else if (path.isStatement()) {
             path.insertBefore(T.expressionStatement(increment));
         } else if ((path.isFunctionExpression() || path.isArrowFunctionExpression()) && T.isVariableDeclarator(path.parentPath)) {
-            path.parentPath.parentPath.insertBefore(T.expressionStatement(
-                increment
-            ));
+            const parent = path.findParent((path) => path.parentPath.isProgram());
+            if (parent) {
+                parent.insertBefore(T.expressionStatement(
+                    increment
+                ));
+            }
         } else /* istanbul ignore else: not expected */ if (path.isExpression()) {
             path.replaceWith(T.sequenceExpression([increment, path.node]));
         } else {