fix: @putout/plugin-apply-arrow: no loc

Author: coderaiser

packages/plugin-apply-arrow/lib/apply-arrow.js

@@ -59,10 +59,10 @@ function isToLong(__b, maxSize) {
     if (isBlockStatement(__b.body))
         return false;
 
-    if (__b.loc)
+    if (__b.loc) {
         linesCount = __b.loc.end.line - __b.loc.start.line;
-    
-    columnsCount = __b.loc.end.column - __b.loc.start.column;
+        columnsCount = __b.loc.end.column - __b.loc.start.column;
+    }
 
     if (!__b.loc || linesCount) {
         const code = print(__b);

packages/plugin-apply-arrow/test/apply-arrow.js

@@ -4,7 +4,10 @@ const {createTest} = require('@putout/test');
 const {operator} = require('putout');
 
 const plugin = require('..');
-const {getTemplateValues} = operator;
+const {
+    compare,
+    getTemplateValues,
+} = operator;
 
 const test = createTest(__dirname, {
     plugins: [
@@ -54,10 +57,15 @@ test('putout: apply-arrow: no report: no-loc', (t) => {
         noLoc: {
             report: () => '',
             fix: (path) => {
-                const {__b} = getTemplateValues(path, FN);
+                const {__b} = getTemplateValues(path.get('body.0.declaration'), FN);
+                
                 delete __b.loc;
             },
-            include: () => [FN],
+            include: () => ['Program'],
+            filter: (path) => {
+                const fnPath = path.get('body.0.declaration');
+                return compare(fnPath, FN);
+            },
         },
     });
     t.end();

packages/plugin-apply-arrow/test/fixture/no-loc-fix.js

@@ -1,3 +1 @@
-export function equal(a, b) {
-    return a.type === b.type && a.value === b.value;
-}
+export const equal = (a, b) => a.type === b.type;

packages/plugin-apply-arrow/test/fixture/no-loc.js

@@ -1,4 +1,4 @@
 export function equal(a, b) {
-    return a.type === b.type && a.value === b.value;
+    return a.type === b.type;
 }