fix(remix): Update defer injection logic.

getsentry · Oct 13, 2023 · 2cafe14 · 2cafe14
1 parent 6062d0c
commit 2cafe14
Show file tree

Hide file tree

Showing 8 changed files with 20 additions and 10 deletions.
diff --git a/packages/remix/src/utils/instrumentServer.ts b/packages/remix/src/utils/instrumentServer.ts
@@ -259,11 +259,11 @@ function makeWrappedRootLoader(remixVersion: number) {
       const traceAndBaggage = getTraceAndBaggage();
 
       if (isDeferredData(res)) {
-        return {
-          ...res.data,
-          ...traceAndBaggage,
-          remixVersion,
-        };
+        res.data['sentryTrace'] = traceAndBaggage.sentryTrace;
+        res.data['sentryBaggage'] = traceAndBaggage.sentryBaggage;
+        res.data['remixVersion'] = remixVersion;
+
+        return res;
       }
 
       if (isResponse(res)) {

diff --git a/packages/remix/test/integration/app_v1/entry.server.tsx b/packages/remix/test/integration/app_v1/entry.server.tsx
@@ -9,6 +9,7 @@ Sentry.init({
   tracePropagationTargets: ['example.org'],
   // Disabling to test series of envelopes deterministically.
   autoSessionTracking: false,
+  debug: true,
 });
 
 export default function handleRequest(

diff --git a/packages/remix/test/integration/app_v1/routes/loader-defer-response/$id.tsx b/packages/remix/test/integration/app_v1/routes/loader-defer-response/$id.tsx
@@ -0,0 +1,2 @@
+export * from '../../../common/routes/loader-defer-response.$id';
+export { default } from '../../../common/routes/loader-defer-response.$id';
diff --git a/packages/remix/test/integration/app_v1/routes/loader-defer-response/index.tsx b/packages/remix/test/integration/app_v1/routes/loader-defer-response/index.tsx
diff --git a/packages/remix/test/integration/app_v2/routes/loader-defer-response.$id.tsx b/packages/remix/test/integration/app_v2/routes/loader-defer-response.$id.tsx
@@ -0,0 +1,2 @@
+export * from '../../common/routes/loader-defer-response.$id';
+export { default } from '../../common/routes/loader-defer-response.$id';
diff --git a/packages/remix/test/integration/app_v2/routes/loader-defer-response.tsx b/packages/remix/test/integration/app_v2/routes/loader-defer-response.tsx
diff --git a/...n/common/routes/loader-defer-response.tsx → ...mmon/routes/loader-defer-response.$id.tsx b/...n/common/routes/loader-defer-response.tsx → ...mmon/routes/loader-defer-response.$id.tsx
@@ -14,7 +14,7 @@ export default function LoaderJSONResponse() {
 
   return (
     <div>
-      <h1>{data && data.id ? data.id : 'Not Found'}</h1>
+      <h1 id="data-render">{data && data.id ? data.id : 'Not Found'}</h1>
     </div>
   );
 }
diff --git a/packages/remix/test/integration/test/client/deferred-response.test.ts b/packages/remix/test/integration/test/client/deferred-response.test.ts
@@ -0,0 +1,9 @@
+import { test, expect } from '@playwright/test';
+
+test('should receive correct data from instrumented defer response', async ({ page }) => {
+  await page.goto('/loader-defer-response/98765');
+
+  const renderedId = await page.waitForSelector('#data-render');
+
+  expect(await renderedId?.textContent()).toBe('98765');
+});