add counter?

Author: alvarosevilla95

e2e/react-start/server-functions/src/routes/middleware/retry-next.tsx

@@ -1,20 +1,26 @@
 import { createFileRoute } from '@tanstack/react-router'
 import { createMiddleware, createServerFn } from '@tanstack/react-start'
 
+let count = 0
+
 const retryMiddleware = createMiddleware({ type: 'function' })
   .client(async ({ next }) => {
     await next()
     return await next()
   })
   .server(async ({ next }) => {
+    // Reset count when called a second time
+    if (count == 4) {
+      count = 0
+    }
     await next()
     return await next()
   })
 
 const serverFn = createServerFn()
   .middleware([retryMiddleware])
   .handler(() => {
-    return 'success'
+    return `called ${++count} times`
   })
 
 export const Route = createFileRoute('/middleware/retry-next')({

e2e/react-start/server-functions/tests/server-functions.spec.ts

@@ -376,7 +376,7 @@ test.describe('middleware', () => {
     async function runRetryTest(page: Page) {
       await page.waitForLoadState('networkidle')
       await expect(page.getByTestId('retry-success-result')).toContainText(
-        'success',
+        'called 4 times',
       )
     }