fix tests

Author: TheodoreSpeaks

apps/sim/app/api/auth/forget-password/route.test.ts

@@ -34,6 +34,8 @@ vi.mock('@/lib/auth', () => ({
 }))
 vi.mock('@sim/logger', () => ({
   createLogger: vi.fn().mockReturnValue(mockLogger),
+  runWithRequestContext: <T>(_ctx: unknown, fn: () => T): T => fn(),
+  getRequestContext: () => undefined,
 }))
 
 import { POST } from '@/app/api/auth/forget-password/route'

apps/sim/app/api/auth/reset-password/route.test.ts

@@ -31,6 +31,8 @@ vi.mock('@/lib/auth', () => ({
 }))
 vi.mock('@sim/logger', () => ({
   createLogger: vi.fn().mockReturnValue(mockLogger),
+  runWithRequestContext: <T>(_ctx: unknown, fn: () => T): T => fn(),
+  getRequestContext: () => undefined,
 }))
 
 import { POST } from '@/app/api/auth/reset-password/route'

apps/sim/app/api/folders/[id]/route.test.ts

@@ -40,6 +40,8 @@ const mockGetUserEntityPermissions = permissionsMockFns.mockGetUserEntityPermiss
 vi.mock('@/lib/audit/log', () => auditMock)
 vi.mock('@sim/logger', () => ({
   createLogger: vi.fn().mockReturnValue(mockLogger),
+  runWithRequestContext: <T>(_ctx: unknown, fn: () => T): T => fn(),
+  getRequestContext: () => undefined,
 }))
 vi.mock('@/lib/workspaces/permissions/utils', () => permissionsMock)
 vi.mock('@sim/db', () => ({

apps/sim/app/api/folders/route.test.ts

@@ -37,6 +37,8 @@ vi.mock('drizzle-orm', () => ({
 }))
 vi.mock('@sim/logger', () => ({
   createLogger: vi.fn().mockReturnValue(mockLogger),
+  runWithRequestContext: <T>(_ctx: unknown, fn: () => T): T => fn(),
+  getRequestContext: () => undefined,
 }))
 vi.mock('@/lib/workspaces/permissions/utils', () => permissionsMock)
 

apps/sim/lib/core/utils/with-route-handler.ts

@@ -23,9 +23,9 @@ export function withRouteHandler<T>(handler: RouteHandler<T>): RouteHandler<T> {
   return async (request: NextRequest, context: T) => {
     const requestId = generateRequestId()
     const startTime = Date.now()
-    const method = request.method
+    const method = request?.method ?? 'UNKNOWN'
     const path =
-      request.nextUrl?.pathname ?? new URL(request.url ?? '/', 'http://localhost').pathname
+      request?.nextUrl?.pathname ?? new URL(request?.url ?? '/', 'http://localhost').pathname
 
     return runWithRequestContext({ requestId, method, path }, async () => {
       let response: NextResponse | Response
@@ -36,22 +36,22 @@ export function withRouteHandler<T>(handler: RouteHandler<T>): RouteHandler<T> {
         const message = error instanceof Error ? error.message : 'Unknown error'
         logger.error('Unhandled route error', { duration, error: message })
         response = NextResponse.json({ error: 'Internal server error', requestId }, { status: 500 })
-        response.headers.set('x-request-id', requestId)
+        response?.headers?.set('x-request-id', requestId)
         return response
       }
 
-      const status = response.status
+      const status = response?.status ?? 0
       const duration = Date.now() - startTime
 
       if (status >= 500) {
         logger.error('Server error response', { status, duration })
       } else if (status >= 400) {
         logger.warn('Client error response', { status, duration })
-      } else {
+      } else if (status > 0) {
         logger.info('OK', { status, duration })
       }
 
-      response.headers.set('x-request-id', requestId)
+      response?.headers?.set('x-request-id', requestId)
       return response
     })
   }