readme-assert 7.2.2 → 7.3.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "readme-assert",
-  "version": "7.2.2",
+  "version": "7.3.0",
   "description": "Run code blocks in your readme as tests",
   "keywords": [
     "readme",

package/src/comment-to-assert.js

@@ -13,6 +13,9 @@ import MagicString from "magic-string";
  *   expr // rejects /pat/  → assert.rejects(() => expr, /pat/)
  *   expr //=> rejects Error: msg → assert.rejects(() => expr, { message: "msg" })
  *
+ * When the expression is an AwaitExpression, throws/Error assertions are
+ * promoted to async rejects (await converts rejection → throw).
+ *
  * Uses oxc-parser for AST + comment extraction. Handles both JS and TS.
  *
  * @param {string} code - JavaScript or TypeScript source
@@ -34,6 +37,8 @@ export function commentToAssert(code, { typescript = false } = {}) {
     const comment = findTrailingComment(comments, node, code);
     if (!comment) continue;
 
+    const isAwait = node.expression.type === "AwaitExpression";
+
     const match = comment.value.match(/^\s*(=>|→|->)\s*([\s\S]*)$/);
     const throwsMatch = comment.value.match(/^\s*throws\s+([\s\S]*)$/);
     const rejectsMatch = comment.value.match(/^\s*rejects\s+([\s\S]*)$/);
@@ -75,10 +80,13 @@ export function commentToAssert(code, { typescript = false } = {}) {
         s.overwrite(
           node.start,
           comment.end,
-          `await assert.rejects(() => ${exprSource}, { ${props.join(", ")} });`,
+          isAwait
+            ? `await assert.rejects(async () => { ${exprSource}; }, { ${props.join(", ")} });`
+            : `await assert.rejects(() => ${exprSource}, { ${props.join(", ")} });`,
         );
       } else if (errorMatch) {
         // expr //=> TypeError: msg → assert.throws(() => { expr }, { name, message })
+        // await expr //=> Error: msg → assert.rejects(async () => { expr }, { name, message })
         const errorName = errorMatch[1];
         const errorMessage = errorMatch[2]?.trim();
         const exprSource = code.slice(
@@ -92,7 +100,9 @@ export function commentToAssert(code, { typescript = false } = {}) {
         s.overwrite(
           node.start,
           comment.end,
-          `assert.throws(() => { ${exprSource}; }, { ${props.join(", ")} });`,
+          isAwait
+            ? `await assert.rejects(async () => { ${exprSource}; }, { ${props.join(", ")} });`
+            : `assert.throws(() => { ${exprSource}; }, { ${props.join(", ")} });`,
         );
       } else if (isConsoleCall(node.expression)) {
         // console.log(expr) //=> value → keep log, add assertion after.
@@ -128,7 +138,9 @@ export function commentToAssert(code, { typescript = false } = {}) {
       s.overwrite(
         node.start,
         comment.end,
-        `assert.throws(() => { ${exprSource}; }, ${pattern});`,
+        isAwait
+          ? `await assert.rejects(async () => { ${exprSource}; }, ${pattern});`
+          : `assert.throws(() => { ${exprSource}; }, ${pattern});`,
       );
       changed = true;
     } else if (rejectsMatch) {
@@ -140,7 +152,9 @@ export function commentToAssert(code, { typescript = false } = {}) {
       s.overwrite(
         node.start,
         comment.end,
-        `await assert.rejects(() => ${exprSource}, ${pattern});`,
+        isAwait
+          ? `await assert.rejects(async () => { ${exprSource}; }, ${pattern});`
+          : `await assert.rejects(() => ${exprSource}, ${pattern});`,
       );
       changed = true;
     }

package/src/generate.js

@@ -66,7 +66,9 @@ function assembleUnit(blocks) {
   }
 
   const hasESM = imports.length > 0;
-  const hasCJS = /\brequire\s*\(/.test(bodyLines.join("\n"));
+  const body = bodyLines.join("\n");
+  const hasAwait = /\bawait\s/.test(body);
+  const hasCJS = !hasAwait && /\brequire\s*\(/.test(body);
 
   // Place assert import on line 0 (before markdown line 1) so line numbers
   // in the generated code match the original markdown positions exactly.