Add test for circularReferenceReplacer

PostHog · Aug 13, 2024 · 0b88932 · 0b88932
1 parent 6325059
commit 0b88932
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/src/__tests__/extensions/replay/sessionrecording-utils.test.ts b/src/__tests__/extensions/replay/sessionrecording-utils.test.ts
@@ -8,6 +8,7 @@ import {
     splitBuffer,
     SEVEN_MEGABYTES,
     estimateSize,
+    circularReferenceReplacer,
 } from '../../../extensions/replay/sessionrecording-utils'
 import { largeString, threeMBAudioURI, threeMBImageURI } from '../test_data/sessionrecording-utils-test-data'
 import { eventWithTime } from '@rrweb/types'
@@ -333,4 +334,22 @@ describe(`SessionRecording utility functions`, () => {
             expect(result).toEqual([buffer])
         })
     })
+
+    describe('circularReferenceReplacer', () => {
+        it('should handle circular references', () => {
+            const obj: any = {}
+            obj.obj = obj
+            const result = JSON.stringify(obj, circularReferenceReplacer())
+            expect(result).toEqual('{"obj":"[Circular]"}')
+        })
+
+        it('should handle nested circular references', () => {
+            const a: any = {}
+            const b: any = {}
+            a.b = b
+            b.a = a
+            const result = JSON.stringify(a, circularReferenceReplacer())
+            expect(result).toEqual('{"b":{"a":"[Circular]"}}')
+        })
+    })
 })
diff --git a/src/extensions/replay/sessionrecording-utils.ts b/src/extensions/replay/sessionrecording-utils.ts
@@ -15,7 +15,7 @@ import { isObject } from '../../utils/type-utils'
 import { SnapshotBuffer } from './sessionrecording'
 
 // taken from https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Errors/Cyclic_object_value#circular_references
-function circularReferenceReplacer() {
+export function circularReferenceReplacer() {
     const ancestors: any[] = []
     return function (this: any, _key: string, value: any) {
         if (isObject(value)) {