Add test in shimmer wrap to preserve function name (#3237)

DataDog · Jun 20, 2023 · 405d58f · 405d58f
1 parent 99b9f61
commit 405d58f
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/packages/datadog-shimmer/test/shimmer.spec.js b/packages/datadog-shimmer/test/shimmer.spec.js
@@ -132,6 +132,14 @@ describe('shimmer', () => {
       expect(obj.count).to.have.length(3)
     })
 
+    it('should preserve the original function name', () => {
+      const obj = { count (a, b, c) {} }
+
+      shimmer.wrap(obj, 'count', () => () => {})
+
+      expect(obj.count).to.have.property('name', 'count')
+    })
+
     it('should inherit from the original prototype', () => {
       const obj = { count: () => {} }
 
@@ -301,6 +309,14 @@ describe('shimmer', () => {
       expect(wrapped).to.have.length(3)
     })
 
+    it('should preserve the original function name', () => {
+      const count = function count (a, b, c) {}
+
+      const wrapped = shimmer.wrap(count, () => {})
+
+      expect(wrapped).to.have.property('name', 'count')
+    })
+
     it('should inherit from the original prototype', () => {
       const count = () => {}