@putout/plugin-tape 18.0.0 → 19.0.0

package/lib/add-args/index.js

@@ -1,7 +1,8 @@
 import {operator} from 'putout';
 
 const {addArgs} = operator;
-const {
+
+export const {
     report,
     fix,
     traverse,
@@ -15,9 +16,3 @@ const {
         'test.skip("__a", async (__args) => __body)',
     ]],
 });
-
-export {
-    report,
-    fix,
-    traverse,
-};

package/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) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@putout/plugin-tape",
-  "version": "18.0.0",
+  "version": "19.0.0",
   "type": "module",
   "author": "coderaiser <mnemonic.enemy@gmail.com> (https://github.com/coderaiser)",
   "description": "🐊Putout plugin helps with tests",
@@ -41,7 +41,7 @@
     "nodemon": "^3.0.1"
   },
   "peerDependencies": {
-    "putout": ">=39"
+    "putout": ">=40"
   },
   "license": "MIT",
   "engines": {