feature: @putout/engine-runner: find: override traverse

Author: coderaiser

packages/engine-runner/lib/index.js

@@ -62,6 +62,7 @@ const run = ({ast, fix, shebang, pluginsFind, pluginsTraverse, template, merge,
         shebang,
         template,
         pluginsFind,
+        traverse,
     }),
     ...runWithMerge({
         ast,
@@ -100,7 +101,7 @@ function runWithMerge({ast, fix, shebang, template, pluginsTraverse, merge, trav
     return places;
 }
 
-function runWithoutMerge({ast, fix, shebang, template, pluginsFind}) {
+function runWithoutMerge({ast, fix, shebang, template, pluginsFind, traverse}) {
     const places = [];
 
     for (const {rule, plugin, msg, options} of pluginsFind) {
@@ -116,6 +117,7 @@ function runWithoutMerge({ast, fix, shebang, template, pluginsFind}) {
             fix,
             shebang,
             template,
+            traverse,
         });
 
         if (!items.length)

packages/engine-runner/lib/super-find.js

@@ -5,18 +5,19 @@ const {traverse: babelTraverse, types} = require('@putout/babel');
 const {generate} = require('@putout/engine-parser');
 const {merge} = babelTraverse.visitors;
 
-module.exports = function superFind({rule, find, ast, options, template}) {
+module.exports = function superFind({rule, find, ast, options, template, traverse = babelTraverse}) {
     const pushItems = [];
 
     const push = (a) => {
         pushItems.push(a);
     };
 
     const returnItems = find(ast, {
-        traverse: traverse({
+        traverse: createTraverse({
             rule,
             options,
             template,
+            traverse,
         }),
         generate,
         types,
@@ -30,12 +31,12 @@ module.exports = function superFind({rule, find, ast, options, template}) {
     ];
 };
 
-const traverse = ({rule, options, template}) => (ast, visitor) => {
+const createTraverse = ({rule, options, template, traverse}) => (ast, visitor) => {
     const templateVisitors = merge(template({
         rule,
         visitor,
         options,
     }));
 
-    return babelTraverse(ast, templateVisitors);
+    return traverse(ast, templateVisitors);
 };

packages/engine-runner/test/runner.js

@@ -1026,14 +1026,7 @@ test('putout: runner: runPlugins: override traverse', (t) => {
         plugins,
     });
 
-    const expected = [
-        ast, {
-            _exploded: true,
-            _verified: true,
-        },
-    ];
-    
-    t.calledWith(traverse, expected);
+    t.calledCount(traverse, 2);
     t.end();
 });