assert: fix CallTracker wraps the function causes the length to be lost

PR-URL: https://github.com/nodejs/node/pull/42909
Fixes: https://github.com/nodejs/node/issues/40484
Reviewed-By: Antoine du Hamel <duhamelantoine1995@gmail.com>

1 files changed, 18 insertions, 14 deletions

diff --git a/lib/internal/assert/calltracker.js b/lib/internal/assert/calltracker.js
index 0fbdf70e5d8..f00f2e33271 100644
--- a/lib/internal/assert/calltracker.js
+++ b/lib/internal/assert/calltracker.js
@@ -4,6 +4,7 @@ const {
   ArrayPrototypePush,
   Error,
   FunctionPrototype,
+  Proxy,
   ReflectApply,
   SafeSet,
 } = primordials;
@@ -46,20 +47,23 @@ class CallTracker {
     const callChecks = this.#callChecks;
     callChecks.add(context);
 
-    return function() {
-      context.actual++;
-      if (context.actual === context.exact) {
-        // Once function has reached its call count remove it from
-        // callChecks set to prevent memory leaks.
-        callChecks.delete(context);
-      }
-      // If function has been called more than expected times, add back into
-      // callchecks.
-      if (context.actual === context.exact + 1) {
-        callChecks.add(context);
-      }
-      return ReflectApply(fn, this, arguments);
-    };
+    return new Proxy(fn, {
+      __proto__: null,
+      apply(fn, thisArg, argList) {
+        context.actual++;
+        if (context.actual === context.exact) {
+          // Once function has reached its call count remove it from
+          // callChecks set to prevent memory leaks.
+          callChecks.delete(context);
+        }
+        // If function has been called more than expected times, add back into
+        // callchecks.
+        if (context.actual === context.exact + 1) {
+          callChecks.add(context);
+        }
+        return ReflectApply(fn, thisArg, argList);
+      },
+    });
   }
 
   report() {