assert: fix exception message for assert(0) on try catch block

Fixes: nodejs#30872
hidecology · Feb 22, 2023 · 76fbda3 · 76fbda3
1 parent 655b070
commit 76fbda3
Show file tree

Hide file tree

Showing 2 changed files with 67 additions and 20 deletions.
diff --git a/lib/assert.js b/lib/assert.js
@@ -78,6 +78,7 @@ let isDeepEqual;
 let isDeepStrictEqual;
 let parseExpressionAt;
 let findNodeAround;
+let tokenizer;
 let decoder;
 
 function lazyLoadComparison() {
@@ -242,39 +243,42 @@ function getCode(fd, line, column) {
 
 function parseCode(code, offset) {
   // Lazy load acorn.
-  if (parseExpressionAt === undefined) {
+  if (parseExpressionAt === undefined || tokenizer === undefined) {
     const Parser = require('internal/deps/acorn/acorn/dist/acorn').Parser;
     ({ findNodeAround } = require('internal/deps/acorn/acorn-walk/dist/walk'));
 
     parseExpressionAt = FunctionPrototypeBind(Parser.parseExpressionAt, Parser);
+    tokenizer = FunctionPrototypeBind(Parser.tokenizer, Parser);
   }
   let node;
-  let start = 0;
+  let start;
   // Parse the read code until the correct expression is found.
-  do {
+  for (const token of tokenizer(code, { ecmaVersion: 'latest' })) {
+    start = token.start;
+    if (start > offset) {
+      // No matching expression found. This could happen if the assert
+      // expression is bigger than the provided buffer.
+      break;
+    }
     try {
       node = parseExpressionAt(code, start, { ecmaVersion: 'latest' });
-      start = node.end + 1 || start;
       // Find the CallExpression in the tree.
       node = findNodeAround(node, offset, 'CallExpression');
-    } catch (err) {
-      // Unexpected token error and the like.
-      start += err.raisedAt || 1;
-      if (start > offset) {
-        // No matching expression found. This could happen if the assert
-        // expression is bigger than the provided buffer.
-        // eslint-disable-next-line no-throw-literal
-        throw null;
+      if (node && node.node.end >= offset) {
+        return [
+          node.node.start,
+          StringPrototypeReplace(StringPrototypeSlice(code,
+                                                      node.node.start, node.node.end),
+                                 escapeSequencesRegExp, escapeFn),
+        ];
       }
+    // eslint-disable-next-line no-unused-vars
+    } catch (err) {
+      continue;
     }
-  } while (node === undefined || node.node.end < offset);
-
-  return [
-    node.node.start,
-    StringPrototypeReplace(StringPrototypeSlice(code,
-                                                node.node.start, node.node.end),
-                           escapeSequencesRegExp, escapeFn),
-  ];
+  }
+  // eslint-disable-next-line no-throw-literal
+  throw null;
 }
 
 function getErrMessage(message, fn) {

diff --git a/test/parallel/test-assert.js b/test/parallel/test-assert.js
@@ -726,6 +726,49 @@ assert.throws(
              'assert.ok(null)\n'
   }
 );
+assert.throws(
+  () => {
+    try { assert.ok(0);   // eslint-disable-line no-useless-catch, brace-style
+    } catch (err) {
+      throw err;
+    }
+  },
+  {
+    code: 'ERR_ASSERTION',
+    constructor: assert.AssertionError,
+    generatedMessage: true,
+    message: 'The expression evaluated to a falsy value:\n\n  ' +
+             'assert.ok(0)\n'
+  }
+);
+assert.throws(
+  () => {
+    try {
+      throw new Error();
+    } catch (err) { assert.ok(0); }     // eslint-disable-line no-unused-vars
+  },
+  {
+    code: 'ERR_ASSERTION',
+    constructor: assert.AssertionError,
+    generatedMessage: true,
+    message: 'The expression evaluated to a falsy value:\n\n  ' +
+             'assert.ok(0)\n'
+  }
+);
+assert.throws(
+  () => {
+    function test() { assert.ok(0);     // eslint-disable-line brace-style
+    }
+    test();
+  },
+  {
+    code: 'ERR_ASSERTION',
+    constructor: assert.AssertionError,
+    generatedMessage: true,
+    message: 'The expression evaluated to a falsy value:\n\n  ' +
+             'assert.ok(0)\n'
+  }
+);
 assert.throws(
   () => assert(typeof 123n === 'string'),
   {