Adjust tests to look at trace continuation via request handler spans

Author: andreiborza

dev-packages/e2e-tests/test-applications/react-router-7-framework-custom/tests/performance/trace-propagation.test.ts

@@ -31,7 +31,11 @@ test.describe('Trace propagation', () => {
     const clientTx = await clientTxPromise;
 
     expect(clientTx.contexts?.trace?.trace_id).toEqual(serverTx.contexts?.trace?.trace_id);
-    expect(clientTx.contexts?.trace?.parent_span_id).toBe(serverTx.contexts?.trace?.span_id);
+
+    const requestHandlerSpan = serverTx.spans?.find(span => span.op === 'request_handler.express');
+
+    expect(requestHandlerSpan).toBeDefined();
+    expect(clientTx.contexts?.trace?.parent_span_id).toBe(requestHandlerSpan?.span_id);
   });
 
   test('should not have trace connection for prerendered pages', async ({ page }) => {

dev-packages/e2e-tests/test-applications/react-router-7-framework-node-20-18/tests/performance/trace-propagation.test.ts

@@ -31,7 +31,11 @@ test.describe('Trace propagation', () => {
     const clientTx = await clientTxPromise;
 
     expect(clientTx.contexts?.trace?.trace_id).toEqual(serverTx.contexts?.trace?.trace_id);
-    expect(clientTx.contexts?.trace?.parent_span_id).toBe(serverTx.contexts?.trace?.span_id);
+
+    const requestHandlerSpan = serverTx.spans?.find(span => span.op === 'request_handler.express');
+
+    expect(requestHandlerSpan).toBeDefined();
+    expect(clientTx.contexts?.trace?.parent_span_id).toBe(requestHandlerSpan?.span_id);
   });
 
   test('should not have trace connection for prerendered pages', async ({ page }) => {

dev-packages/e2e-tests/test-applications/react-router-7-framework/tests/performance/trace-propagation.test.ts

@@ -31,7 +31,11 @@ test.describe('Trace propagation', () => {
     const clientTx = await clientTxPromise;
 
     expect(clientTx.contexts?.trace?.trace_id).toEqual(serverTx.contexts?.trace?.trace_id);
-    expect(clientTx.contexts?.trace?.parent_span_id).toBe(serverTx.contexts?.trace?.span_id);
+
+    const requestHandlerSpan = serverTx.spans?.find(span => span.op === 'request_handler.express');
+
+    expect(requestHandlerSpan).toBeDefined();
+    expect(clientTx.contexts?.trace?.parent_span_id).toBe(requestHandlerSpan?.span_id);
   });
 
   test('should not have trace connection for prerendered pages', async ({ page }) => {