@putout/plugin-tape 3.10.0 → 3.11.0

package/README.md

@@ -21,6 +21,7 @@ npm i @putout/plugin-tape -D
         "tape/apply-stub-operator": "on",
         "tape/apply-with-name": "on",
         "tape/add-t-end": "on",
+        "tape/add-stop-all": "on",
         "tape/remove-useless-t-end": "on",
         "tape/sync-with-name": "on",
         "tape/switch-expected-with-result": "on",
@@ -433,6 +434,34 @@ test('xxx', (t) => {
 });
 ```
 
+## add-stop-all
+
+### ❌ Incorrect code example
+
+```js
+test('stop-all: should be called', () => {
+    mockImport('fs/promises', {
+        readFile: stub(),
+    });
+    
+    t.end();
+});
+```
+
+### ✅ Correct code example
+
+```js
+test('stop-all: should be called', () => {
+    mockImport('fs/promises', {
+        readFile: stub(),
+    });
+    
+    stopAll();
+    
+    t.end();
+});
+```
+
 ## remove-useless-t-end
 
 ### ❌ Incorrect code example

package/lib/add-stop-all/index.js

@@ -0,0 +1,85 @@
+'use strict';
+
+const {
+    types,
+    operator,
+    template,
+} = require('putout');
+
+const {
+    traverse,
+    compare,
+} = operator;
+
+const {ExpressionStatement} = types;
+
+module.exports.report = () => `Call 'stopAll()' at the end of test when 'mockImport()' used`;
+
+module.exports.fix = (path) => {
+    const assertionPath = getAssertionsPath(path);
+    const stopAllNode = template.ast('stopAll()');
+    
+    assertionPath.insertBefore(ExpressionStatement(stopAllNode));
+};
+
+const TEST = 'test("__a", (t) => __body)';
+
+module.exports.traverse = ({push}) => ({
+    [TEST]: (path) => {
+        const {
+            hasMockImport,
+            hasStopAll,
+            hasAssertions,
+        } = check(path);
+        
+        if (!hasMockImport)
+            return;
+        
+        if (!hasAssertions)
+            return;
+        
+        if (hasStopAll)
+            return;
+        
+        push(path);
+    },
+});
+
+function check(path) {
+    let hasStopAll = false;
+    let hasMockImport = false;
+    let hasAssertions = false;
+    
+    traverse(path, {
+        'mockImport(__a, __b)': (path) => {
+            hasMockImport = true;
+        },
+        'stopAll()': () => {
+            hasStopAll = true;
+        },
+        't.__(__args)': () => {
+            hasAssertions = true;
+        },
+    });
+    
+    return {
+        hasStopAll,
+        hasMockImport,
+        hasAssertions,
+    };
+}
+
+function getAssertionsPath(path) {
+    const bodyPath = path.get('arguments.1.body.body');
+    let resultPath = null;
+    
+    for (const pathEl of bodyPath) {
+        if (compare(pathEl, 't.__(__args)')) {
+            resultPath = pathEl;
+            break;
+        }
+    }
+    
+    return resultPath;
+}
+

package/lib/index.js

@@ -27,6 +27,7 @@ module.exports.rules = {
     ...getRule('switch-expected-with-result'),
     ...getRule('apply-stub-operator'),
     ...getRule('add-t-end'),
+    ...getRule('add-stop-all'),
     ...getRule('remove-useless-t-end'),
 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@putout/plugin-tape",
-  "version": "3.10.0",
+  "version": "3.11.0",
   "author": "coderaiser <mnemonic.enemy@gmail.com> (https://github.com/coderaiser)",
   "description": "putout plugin helps with tests",
   "homepage": "http://github.com/coderaiser/putout",