make getPluginName resilient against too low stackTraceLimit (#209)

fastify · Jan 10, 2023 · cd72d10 · cd72d10
1 parent d7c3866
commit cd72d10
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/lib/getPluginName.js b/lib/getPluginName.js
@@ -6,9 +6,12 @@ const fileNamePattern = /(\w*(\.\w*)*)\..*/
 module.exports = function getPluginName (fn) {
   if (fn.name.length > 0) return fn.name
 
+  const stackTraceLimit = Error.stackTraceLimit
+  Error.stackTraceLimit = 10
   try {
     throw new Error('anonymous function')
   } catch (e) {
+    Error.stackTraceLimit = stackTraceLimit
     return extractPluginName(e.stack)
   }
 }

diff --git a/test/test.js b/test/test.js
@@ -155,6 +155,29 @@ test('should set anonymous function name to file it was called from with a count
   t.end()
 })
 
+test('should set function name if Error.stackTraceLimit is set to 0', t => {
+  const stackTraceLimit = Error.stackTraceLimit = 0
+
+  const fp = proxyquire('../plugin.js', { stubs: {} })
+
+  const fn = fp((fastify, opts, next) => {
+    next()
+  })
+
+  t.equal(fn[Symbol.for('plugin-meta')].name, 'test-auto-0')
+  t.equal(fn[Symbol.for('fastify.display-name')], 'test-auto-0')
+
+  const fn2 = fp((fastify, opts, next) => {
+    next()
+  })
+
+  t.equal(fn2[Symbol.for('plugin-meta')].name, 'test-auto-1')
+  t.equal(fn2[Symbol.for('fastify.display-name')], 'test-auto-1')
+
+  Error.stackTraceLimit = stackTraceLimit
+  t.end()
+})
+
 test('should set display-name to meta name', t => {
   t.plan(2)