Handle ExportNamedDeclarations

gotwarlost · Dec 15, 2015 · 2a089f8 · 2a089f8
1 parent 8a678fa
commit 2a089f8
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/lib/instrumenter.js b/lib/instrumenter.js
@@ -790,6 +790,8 @@
         coverStatement: function (node, walker) {
             var sName,
                 incrStatementCount,
+                parent,
+                insertionTarget,
                 grandParent;
 
             this.maybeSkipNode(node, 'next');
@@ -817,7 +819,19 @@
                         )
                     )
                 );
-                this.splice(incrStatementCount, node, walker);
+
+                // The insertion target for `incrStatementCount` is usually the current node.
+                insertionTarget = node;
+
+                // However, to handle ExportNamedDeclarations, like `export var ...`
+                // We must insert the statement before our parent node,
+                // if it is such an export statement.
+                parent = walker.parent();
+                if (parent && parent.node.type === SYNTAX.ExportNamedDeclaration.name) {
+                  insertionTarget = parent;
+                }
+
+                this.splice(incrStatementCount, insertionTarget, walker);
             }
         },
 
@@ -1048,4 +1062,3 @@
     }
 
 }(typeof module !== 'undefined' && typeof module.exports !== 'undefined' && typeof exports !== 'undefined'));
-