fix(espower): decide to be skipped first, then enter node

This fixes the case when FunctionExpression begins right after the assertion (its params will be transformed accidentally)
power-assert-js · Sep 14, 2014 · 9d0a778 · 9d0a778
1 parent 57cd8a9
commit 9d0a778
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/lib/instrumentor.js b/lib/instrumentor.js
@@ -45,12 +45,11 @@ Instrumentor.prototype.instrument = function (ast) {
                 path = controller.path(),
                 currentPath = path ? path[path.length - 1] : null;
             if (assertionVisitor) {
-                if (assertionVisitor.isCapturingArgument()) {
-                    if (toBeSkipped(currentNode, parentNode, currentPath)) {
-                        skipping = true;
-                        return controller.skip();
-                    }
-                } else {
+                if (toBeSkipped(currentNode, parentNode, currentPath)) {
+                    skipping = true;
+                    return controller.skip();
+                }
+                if (!assertionVisitor.isCapturingArgument()) {
                     return assertionVisitor.enterArgument(currentNode, parentNode, path);
                 }
             } else {