feat: Support removal of async assertion such as assert.rejects

unassert-js · twada · Jul 31, 2022 · Jul 31, 2022 · Jul 31, 2022 · Jul 31, 2022
commit a7107570acb5ed7fb7760a411fda336e61b8c119
diff --git a/src/index.mjs b/src/index.mjs
@@ -228,6 +228,18 @@ function createVisitor (options) {
           }
           break;
         }
+        case syntax.AwaitExpression: {
+          const childNode = currentNode.argument;
+          if (isExpressionStatement(parentNode) && isCallExpression(childNode)) {
+            const callee = childNode.callee;
+            if (isAssertionFunction(callee) || isAssertionMethod(callee) || isConsoleAssert(callee)) {
+              // remove parent ExpressionStatement
+              nodeToRemove.add(parentNode);
+              this.skip();
+            }
+          }
+          break;
+        }
       }
     },
     leave: function (currentNode, parentNode) {