feature: @putout/plugin-printer: remove-legacy-test-declaration: esm

Author: coderaiser

packages/plugin-printer/lib/remove-legacy-test-declaration/fixture/esm.js

@@ -0,0 +1,16 @@
+import {extend} from 'supertape';
+import {printExtension} from '#test/printer';
+import {readFixtures} from '#test/fixture';
+
+export const createTest = (dir) => {
+    const fixture = readFixtures(dir);
+    const test = extend({
+        print: printExtension,
+    });
+    
+    return {
+        fixture,
+        test,
+    };
+};
+

packages/plugin-printer/lib/remove-legacy-test-declaration/index.js

@@ -1,7 +1,6 @@
 import {operator} from 'putout';
 
-const {traverse} = operator;
-
+const {compare} = operator;
 const EXTEND = `
     const test = extend({
         print: printExtension,
@@ -25,13 +24,15 @@ export const replace = () => ({
 });
 
 function check(vars, path) {
-    let is = true;
     const programPath = path.scope.getProgramParent().path;
 
-    traverse(programPath, {
-        'module.exports.createTest = (__args) => __body': () => {
-            is = false;
-        },
-    });
-    return is;
+    for (const current of programPath.node.body) {
+        if (compare(current, 'module.exports.createTest = (__args) => __body'))
+            return false;
+        
+        if (compare(current, 'export const createTest = (__args) => __body'))
+            return false;
+    }
+    
+    return true;
 }

packages/plugin-printer/lib/remove-legacy-test-declaration/index.spec.js

@@ -31,3 +31,8 @@ test('printer: remove-legacy-test-declaration: no report: create-test', (t) => {
     t.noReport('create-test');
     t.end();
 });
+
+test('printer: remove-legacy-test-declaration: no report: esm', (t) => {
+    t.noReport('esm');
+    t.end();
+});