add double wrapping test

nicohrubec · nicohrubec · commit 330d7f89be08 · 2026-01-27T17:22:30.000+01:00
diff --git a/packages/tanstackstart-react/test/vite/autoInstrumentMiddleware.test.ts b/packages/tanstackstart-react/test/vite/autoInstrumentMiddleware.test.ts
@@ -179,6 +179,20 @@ createStart(() => ({
 
     consoleLogSpy.mockRestore();
   });
+
+  it('does not double-wrap already wrapped middleware', () => {
+    const code = `
+createStart(() => ({
+  requestMiddleware: wrapMiddlewaresWithSentry({ authMiddleware }),
+  functionMiddleware: wrapMiddlewaresWithSentry({ loggingMiddleware }),
+}));
+`;
+    const result = wrapGlobalMiddleware(code, '/app/start.ts', false);
+
+    expect(result.didWrap).toBe(false);
+    expect(result.code).toBe(code);
+    expect(result.skipped).toHaveLength(0);
+  });
 });
 
 describe('wrapRouteMiddleware', () => {
@@ -327,6 +341,22 @@ export const Route = createFileRoute('/foo')({
 
     consoleLogSpy.mockRestore();
   });
+
+  it('does not double-wrap already wrapped middleware', () => {
+    const code = `
+export const Route = createFileRoute('/foo')({
+  server: {
+    middleware: wrapMiddlewaresWithSentry({ authMiddleware }),
+    handlers: { GET: () => ({}) },
+  },
+});
+`;
+    const result = wrapRouteMiddleware(code, '/app/routes/foo.ts', false);
+
+    expect(result.didWrap).toBe(false);
+    expect(result.code).toBe(code);
+    expect(result.skipped).toHaveLength(0);
+  });
 });
 
 describe('addSentryImport', () => {
