feature: @putout/plugin-tape: add-t-end: speed up

Author: coderaiser

packages/plugin-tape/lib/add-t-end/fixture/contains-fix.js

@@ -0,0 +1,5 @@
+test('message', (t) => {
+    t.end();
+    something();
+    t.end();
+});

packages/plugin-tape/lib/add-t-end/index.js

@@ -4,8 +4,12 @@ import {
     operator,
 } from 'putout';
 
-const {compare, traverse} = operator;
-const {expressionStatement} = types;
+const {compare} = operator;
+const {
+    expressionStatement,
+    isCallExpression,
+    isFunction,
+} = types;
 
 export const report = () => `'t.end()' is missing at the end of the test`;
 
@@ -19,32 +23,29 @@ export const replace = () => ({
     'test(__a, async (t) => __body)': transform,
 });
 
-function check({__body}, path) {
+function check({__body}) {
     const {body} = __body;
     const {length} = body;
 
     if (!length)
         return true;
 
-    for (const element of body) {
-        if (compare(element, 't.end()'))
-            return false;
-    }
+    const last = body.at(-1);
+    
+    if (compare(last, 't.end()'))
+        return false;
 
-    let found = false;
+    if (compare(last, 'await t.__(__args)'))
+        return false;
 
-    traverse(path, {
-        'await t.__(__args)': () => {
-            found = true;
-            path.stop();
-        },
-        't.end()': (path) => {
-            found = true;
-            path.stop();
-        },
-    });
+    const {expression} = last;
+    
+    if (isCallExpression(expression)) {
+        const lastArg = expression.arguments.at(-1);
+        return !isFunction(lastArg);
+    }
 
-    return !found;
+    return true;
 }
 
 function transform({__body}, path) {

packages/plugin-tape/lib/add-t-end/index.spec.js

@@ -40,8 +40,8 @@ test('plugin-tape: add-t-end: transform: putout', (t) => {
     t.end();
 });
 
-test('plugin-tape: add-t-end: no transform: contains', (t) => {
-    t.noTransform('contains');
+test('plugin-tape: add-t-end: transform: contains', (t) => {
+    t.transform('contains');
     t.end();
 });