fix integration tests

Author: billyvg

packages/replay-internal/test/integration/coreHandlers/handleAfterSendEvent.test.ts

@@ -53,7 +53,7 @@ describe('Integration | coreHandlers | handleAfterSendEvent', () => {
     handler(error4, { statusCode: undefined });
 
     expect(Array.from(replay.getContext().errorIds)).toEqual(['err2']);
-    expect(Array.from(replay.getContext().traceIds)).toEqual([]);
+    expect(Array.from(replay.getContext().traceIds)).toEqual([[-1, expect.any(String)]]);
   });
 
   it('records traceIds from sent transaction events', async () => {
@@ -84,20 +84,16 @@ describe('Integration | coreHandlers | handleAfterSendEvent', () => {
     handler(transaction4, { statusCode: undefined });
 
     expect(Array.from(replay.getContext().errorIds)).toEqual([]);
-    // traceIds is now a Set of [timestamp, trace_id] tuples
-    const traceIds = Array.from(replay.getContext().traceIds);
-    expect(traceIds).toHaveLength(1);
-    expect(traceIds[0][1]).toBe('tr2');
-    expect(typeof traceIds[0][0]).toBe('number');
+    const traceIds = replay.getContext().traceIds;
+    expect(traceIds).toEqual([[expect.any(Number), 'tr2']]);
 
     // Does not affect error session
     await vi.advanceTimersToNextTimerAsync();
 
     expect(Array.from(replay.getContext().errorIds)).toEqual([]);
     // Verify traceIds are still there after advancing timers
-    const traceIdsAfter = Array.from(replay.getContext().traceIds);
-    expect(traceIdsAfter).toHaveLength(1);
-    expect(traceIdsAfter[0][1]).toBe('tr2');
+    const traceIdsAfter = replay.getContext().traceIds;
+    expect(traceIdsAfter).toEqual([[expect.any(Number), 'tr2']]);
     expect(replay.isEnabled()).toBe(true);
     expect(replay.isPaused()).toBe(false);
     expect(replay.recordingMode).toBe('buffer');
@@ -126,7 +122,7 @@ describe('Integration | coreHandlers | handleAfterSendEvent', () => {
         .fill(undefined)
         .map((_, i) => `err-${i}`),
     );
-    expect(Array.from(replay.getContext().traceIds)).toEqual([]);
+    expect(replay.getContext().traceIds).toEqual([[-1, expect.any(String)]]);
   });
 
   it('limits traceIds to max. 100', async () => {

packages/replay-internal/test/integration/errorSampleRate.test.ts

@@ -722,7 +722,8 @@ describe('Integration | errorSampleRate', () => {
           replay_start_timestamp: BASE_TIMESTAMP / 1000,
           timestamp: (BASE_TIMESTAMP + DEFAULT_FLUSH_MIN_DELAY + DEFAULT_FLUSH_MIN_DELAY) / 1000,
           error_ids: [expect.any(String)],
-          trace_ids: [],
+          trace_ids: [expect.any(String)],
+          traces_by_timestamp: [[-1, expect.any(String)]],
           urls: ['http://localhost:3000/'],
           replay_id: expect.any(String),
         }),
@@ -930,7 +931,8 @@ describe('Integration | errorSampleRate', () => {
         replayEventPayload: expect.objectContaining({
           replay_start_timestamp: (BASE_TIMESTAMP + stepDuration * steps) / 1000,
           error_ids: [expect.any(String)],
-          trace_ids: [],
+          trace_ids: [expect.any(String)],
+          traces_by_timestamp: [[-1, expect.any(String)]],
           urls: ['http://localhost:3000/'],
           replay_id: expect.any(String),
         }),

packages/replay-internal/test/integration/sampling.test.ts

@@ -37,7 +37,7 @@ describe('Integration | sampling', () => {
     // This is what the `_context` member is initialized with
     expect(replay.getContext()).toEqual({
       errorIds: new Set(),
-      traceIds: new Set(),
+      traceIds: [],
       urls: [],
       initialTimestamp: expect.any(Number),
       initialUrl: '',
@@ -78,7 +78,7 @@ describe('Integration | sampling', () => {
       errorIds: new Set(),
       initialTimestamp: expect.any(Number),
       initialUrl: 'http://localhost:3000/',
-      traceIds: new Set(),
+      traceIds: [],
       urls: ['http://localhost:3000/'],
     });
     expect(replay.recordingMode).toBe('buffer');

packages/replay-internal/test/integration/sendReplayEvent.test.ts

@@ -340,7 +340,8 @@ describe('Integration | sendReplayEvent', () => {
         replay_start_timestamp: BASE_TIMESTAMP / 1000,
         // timestamp is set on first try, after 5s flush
         timestamp: (BASE_TIMESTAMP + 5000) / 1000,
-        trace_ids: [],
+        trace_ids: [expect.any(String)],
+        traces_by_timestamp: [[-1, expect.any(String)]],
         urls: ['http://localhost:3000/'],
       }),
       recordingPayloadHeader: { segment_id: 0 },

packages/replay-internal/test/integration/session.test.ts

@@ -252,7 +252,7 @@ describe('Integration | session', () => {
       initialTimestamp: newTimestamp - DEFAULT_FLUSH_MIN_DELAY,
       urls: [],
       errorIds: new Set(),
-      traceIds: new Set(),
+      traceIds: [[-1, expect.any(String)]],
     });
   });