[babel] Fix issue with transform-react-inline-elements

oliviertassinari · Dec 23, 2015 · c52e7eb · c52e7eb
1 parent 4bd3924
commit c52e7eb
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/src/index.js b/src/index.js
@@ -3,11 +3,16 @@
 module.exports = function() {
   return {
     visitor: {
-      JSXIdentifier: function(path) {
-        if (path.node.name === 'data-test') {
-          path.parentPath.remove();
-        }
-      },
+      Program(path) {
+        // On program start, do an explicit traversal up front for your plugin.
+        path.traverse({
+          JSXIdentifier: function(path) {
+            if (path.node.name === 'data-test') {
+              path.parentPath.remove();
+            }
+          },
+       });
+      }
     },
   };
 };