simstudioai · waleedlatif1 · Feb 19, 2026 · Feb 19, 2026
diff --git a/apps/sim/app/api/auth/oauth/token/route.test.ts b/apps/sim/app/api/auth/oauth/token/route.test.ts
@@ -3,7 +3,7 @@
  *
  * @vitest-environment node
  */
-import { createMockLogger, createMockRequest } from '@sim/testing'
+import { createMockLogger, createMockRequest, mockHybridAuth } from '@sim/testing'
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
 
 describe('OAuth Token API Routes', () => {
@@ -12,7 +12,7 @@ describe('OAuth Token API Routes', () => {
  const mockRefreshTokenIfNeeded = vi.fn()
  const mockGetOAuthToken = vi.fn()
  const mockAuthorizeCredentialUse = vi.fn()
-const mockCheckSessionOrInternalAuth = vi.fn()
+let mockCheckSessionOrInternalAuth: ReturnType<typeof vi.fn>
 
  const mockLogger = createMockLogger()
 
@@ -41,9 +41,7 @@ describe('OAuth Token API Routes', () => {
  authorizeCredentialUse: mockAuthorizeCredentialUse,
  }))
 
- vi.doMock('@/lib/auth/hybrid', () => ({
- checkSessionOrInternalAuth: mockCheckSessionOrInternalAuth,
- }))
+ ;({ mockCheckSessionOrInternalAuth } = mockHybridAuth())
  })
 
  afterEach(() => {
@@ -73,23 +71,18 @@ describe('OAuth Token API Routes', () => {
  refreshed: false,
  })
 
- // Create mock request
  const req = createMockRequest('POST', {
  credentialId: 'credential-id',
  })
 
- // Import handler after setting up mocks
  const { POST } = await import('@/app/api/auth/oauth/token/route')
 
- // Call handler
  const response = await POST(req)
  const data = await response.json()
 
- // Verify request was handled correctly
  expect(response.status).toBe(200)
  expect(data).toHaveProperty('accessToken', 'fresh-token')
 
- // Verify mocks were called correctly
  expect(mockAuthorizeCredentialUse).toHaveBeenCalled()
  expect(mockGetCredential).toHaveBeenCalled()
  expect(mockRefreshTokenIfNeeded).toHaveBeenCalled()

diff --git a/apps/sim/app/api/chat/[identifier]/route.test.ts b/apps/sim/app/api/chat/[identifier]/route.test.ts
@@ -3,7 +3,7 @@
  *
  * @vitest-environment node
  */
-import { loggerMock } from '@sim/testing'
+import { loggerMock, requestUtilsMock } from '@sim/testing'
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
 
 /**
@@ -94,9 +94,7 @@ vi.mock('@/lib/core/utils/sse', () => ({
  },
 }))
 
-vi.mock('@/lib/core/utils/request', () => ({
- generateRequestId: vi.fn().mockReturnValue('test-request-id'),
-}))
+vi.mock('@/lib/core/utils/request', () => requestUtilsMock)
 
 vi.mock('@/lib/core/security/encryption', () => ({
  decryptSecret: vi.fn().mockResolvedValue({ decrypted: 'test-password' }),

diff --git a/apps/sim/app/api/chat/utils.test.ts b/apps/sim/app/api/chat/utils.test.ts
@@ -1,4 +1,4 @@
-import { databaseMock, loggerMock } from '@sim/testing'
+import { databaseMock, loggerMock, requestUtilsMock } from '@sim/testing'
 import type { NextResponse } from 'next/server'
 /**
  * Tests for chat API utils
@@ -37,9 +37,7 @@ vi.mock('@/lib/core/security/encryption', () => ({
  decryptSecret: mockDecryptSecret,
 }))
 
-vi.mock('@/lib/core/utils/request', () => ({
- generateRequestId: vi.fn(),
-}))
+vi.mock('@/lib/core/utils/request', () => requestUtilsMock)
 
 vi.mock('@/lib/core/config/feature-flags', () => ({
  isDev: true,

diff --git a/apps/sim/app/api/files/delete/route.test.ts b/apps/sim/app/api/files/delete/route.test.ts
@@ -2,6 +2,7 @@ import {
  createMockRequest,
  mockAuth,
  mockCryptoUuid,
+ mockHybridAuth,
  mockUuid,
  setupCommonApiMocks,
 } from '@sim/testing'
@@ -28,13 +29,12 @@ function setupFileApiMocks(
  authMocks.setUnauthenticated()
  }
 
- vi.doMock('@/lib/auth/hybrid', () => ({
- checkSessionOrInternalAuth: vi.fn().mockResolvedValue({
- success: authenticated,
- userId: authenticated ? 'test-user-id' : undefined,
- error: authenticated ? undefined : 'Unauthorized',
- }),
- }))
+ const { mockCheckSessionOrInternalAuth } = mockHybridAuth()
+ mockCheckSessionOrInternalAuth.mockResolvedValue({
+ success: authenticated,
+ userId: authenticated ? 'test-user-id' : undefined,
+ error: authenticated ? undefined : 'Unauthorized',
+ })
 
  vi.doMock('@/app/api/files/authorization', () => ({
  verifyFileAccess: vi.fn().mockResolvedValue(true),

diff --git a/apps/sim/app/api/files/parse/route.test.ts b/apps/sim/app/api/files/parse/route.test.ts
@@ -8,6 +8,7 @@ import {
  createMockRequest,
  mockAuth,
  mockCryptoUuid,
+ mockHybridAuth,
  mockUuid,
  setupCommonApiMocks,
 } from '@sim/testing'
@@ -34,13 +35,12 @@ function setupFileApiMocks(
  authMocks.setUnauthenticated()
  }
 
- vi.doMock('@/lib/auth/hybrid', () => ({
- checkInternalAuth: vi.fn().mockResolvedValue({
- success: authenticated,
- userId: authenticated ? 'test-user-id' : undefined,
- error: authenticated ? undefined : 'Unauthorized',
- }),
- }))
+ const { mockCheckInternalAuth } = mockHybridAuth()
+ mockCheckInternalAuth.mockResolvedValue({
+ success: authenticated,
+ userId: authenticated ? 'test-user-id' : undefined,
+ error: authenticated ? undefined : 'Unauthorized',
+ })
 
  vi.doMock('@/app/api/files/authorization', () => ({
  verifyFileAccess: vi.fn().mockResolvedValue(true),

diff --git a/apps/sim/app/api/files/presigned/route.test.ts b/apps/sim/app/api/files/presigned/route.test.ts
@@ -1,4 +1,10 @@
-import { mockAuth, mockCryptoUuid, mockUuid, setupCommonApiMocks } from '@sim/testing'
+import {
+ mockAuth,
+ mockCryptoUuid,
+ mockHybridAuth,
+ mockUuid,
+ setupCommonApiMocks,
+} from '@sim/testing'
 import { NextRequest } from 'next/server'
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
 
@@ -28,13 +34,12 @@ function setupFileApiMocks(
  authMocks.setUnauthenticated()
  }
 
- vi.doMock('@/lib/auth/hybrid', () => ({
- checkHybridAuth: vi.fn().mockResolvedValue({
- success: authenticated,
- userId: authenticated ? 'test-user-id' : undefined,
- error: authenticated ? undefined : 'Unauthorized',
- }),
- }))
+ const { mockCheckHybridAuth } = mockHybridAuth()
+ mockCheckHybridAuth.mockResolvedValue({
+ success: authenticated,
+ userId: authenticated ? 'test-user-id' : undefined,
+ error: authenticated ? undefined : 'Unauthorized',
+ })
 
  vi.doMock('@/app/api/files/authorization', () => ({
  verifyFileAccess: vi.fn().mockResolvedValue(true),

diff --git a/apps/sim/app/api/files/serve/[...path]/route.test.ts b/apps/sim/app/api/files/serve/[...path]/route.test.ts
@@ -7,6 +7,7 @@ import {
  defaultMockUser,
  mockAuth,
  mockCryptoUuid,
+ mockHybridAuth,
  mockUuid,
  setupCommonApiMocks,
 } from '@sim/testing'
@@ -54,12 +55,11 @@ describe('File Serve API Route', () => {
  withUploadUtils: true,
  })
 
- vi.doMock('@/lib/auth/hybrid', () => ({
- checkSessionOrInternalAuth: vi.fn().mockResolvedValue({
- success: true,
- userId: 'test-user-id',
- }),
- }))
+ const { mockCheckSessionOrInternalAuth: serveAuthMock } = mockHybridAuth()
+ serveAuthMock.mockResolvedValue({
+ success: true,
+ userId: 'test-user-id',
+ })
 
  vi.doMock('@/app/api/files/authorization', () => ({
  verifyFileAccess: vi.fn().mockResolvedValue(true),
@@ -164,12 +164,11 @@ describe('File Serve API Route', () => {
  findLocalFile: vi.fn().mockReturnValue('/test/uploads/nested/path/file.txt'),
  }))
 
- vi.doMock('@/lib/auth/hybrid', () => ({
- checkSessionOrInternalAuth: vi.fn().mockResolvedValue({
- success: true,
- userId: 'test-user-id',
- }),
- }))
+ const { mockCheckSessionOrInternalAuth: serveAuthMock } = mockHybridAuth()
+ serveAuthMock.mockResolvedValue({
+ success: true,
+ userId: 'test-user-id',
+ })
 
  vi.doMock('@/app/api/files/authorization', () => ({
  verifyFileAccess: vi.fn().mockResolvedValue(true),
@@ -225,12 +224,11 @@ describe('File Serve API Route', () => {
  USE_BLOB_STORAGE: false,
  }))
 
- vi.doMock('@/lib/auth/hybrid', () => ({
- checkSessionOrInternalAuth: vi.fn().mockResolvedValue({
- success: true,
- userId: 'test-user-id',
- }),
- }))
+ const { mockCheckSessionOrInternalAuth: serveAuthMock } = mockHybridAuth()
+ serveAuthMock.mockResolvedValue({
+ success: true,
+ userId: 'test-user-id',
+ })
 
  vi.doMock('@/app/api/files/authorization', () => ({
  verifyFileAccess: vi.fn().mockResolvedValue(true),
@@ -290,12 +288,11 @@ describe('File Serve API Route', () => {
  readFile: vi.fn().mockRejectedValue(new Error('ENOENT: no such file or directory')),
  }))
 
- vi.doMock('@/lib/auth/hybrid', () => ({
- checkSessionOrInternalAuth: vi.fn().mockResolvedValue({
- success: true,
- userId: 'test-user-id',
- }),
- }))
+ const { mockCheckSessionOrInternalAuth: serveAuthMock } = mockHybridAuth()
+ serveAuthMock.mockResolvedValue({
+ success: true,
+ userId: 'test-user-id',
+ })
 
  vi.doMock('@/app/api/files/authorization', () => ({
  verifyFileAccess: vi.fn().mockResolvedValue(false), // File not found = no access
@@ -349,12 +346,11 @@ describe('File Serve API Route', () => {
 
  for (const test of contentTypeTests) {
  it(`should serve ${test.ext} file with correct content type`, async () => {
- vi.doMock('@/lib/auth/hybrid', () => ({
- checkSessionOrInternalAuth: vi.fn().mockResolvedValue({
- success: true,
- userId: 'test-user-id',
- }),
- }))
+ const { mockCheckSessionOrInternalAuth: ctAuthMock } = mockHybridAuth()
+ ctAuthMock.mockResolvedValue({
+ success: true,
+ userId: 'test-user-id',
+ })
 
  vi.doMock('@/app/api/files/authorization', () => ({
  verifyFileAccess: vi.fn().mockResolvedValue(true),

diff --git a/apps/sim/app/api/files/upload/route.test.ts b/apps/sim/app/api/files/upload/route.test.ts
@@ -3,7 +3,13 @@
  *
  * @vitest-environment node
  */
-import { mockAuth, mockCryptoUuid, mockUuid, setupCommonApiMocks } from '@sim/testing'
+import {
+ mockAuth,
+ mockCryptoUuid,
+ mockHybridAuth,
+ mockUuid,
+ setupCommonApiMocks,
+} from '@sim/testing'
 import { NextRequest } from 'next/server'
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
 
@@ -27,13 +33,12 @@ function setupFileApiMocks(
  authMocks.setUnauthenticated()
  }
 
- vi.doMock('@/lib/auth/hybrid', () => ({
- checkHybridAuth: vi.fn().mockResolvedValue({
- success: authenticated,
- userId: authenticated ? 'test-user-id' : undefined,
- error: authenticated ? undefined : 'Unauthorized',
- }),
- }))
+ const { mockCheckHybridAuth } = mockHybridAuth()
+ mockCheckHybridAuth.mockResolvedValue({
+ success: authenticated,
+ userId: authenticated ? 'test-user-id' : undefined,
+ error: authenticated ? undefined : 'Unauthorized',
+ })
 
  vi.doMock('@/app/api/files/authorization', () => ({
  verifyFileAccess: vi.fn().mockResolvedValue(true),

diff --git a/apps/sim/app/api/knowledge/search/route.test.ts b/apps/sim/app/api/knowledge/search/route.test.ts
@@ -10,6 +10,7 @@ import {
  createMockRequest,
  mockConsoleLogger,
  mockKnowledgeSchemas,
+ requestUtilsMock,
 } from '@sim/testing'
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
 
@@ -29,9 +30,7 @@ mockKnowledgeSchemas()
 
 vi.mock('@/lib/core/config/env', () => createEnvMock({ OPENAI_API_KEY: 'test-api-key' }))
 
-vi.mock('@/lib/core/utils/request', () => ({
- generateRequestId: vi.fn(() => 'test-request-id'),
-}))
+vi.mock('@/lib/core/utils/request', () => requestUtilsMock)
 
 vi.mock('@/lib/documents/utils', () => ({
  retryWithExponentialBackoff: vi.fn().mockImplementation((fn) => fn()),

diff --git a/apps/sim/app/api/mcp/serve/[serverId]/route.test.ts b/apps/sim/app/api/mcp/serve/[serverId]/route.test.ts
@@ -3,10 +3,11 @@
  *
  * @vitest-environment node
  */
+import { mockHybridAuth } from '@sim/testing'
 import { NextRequest } from 'next/server'
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
 
-const mockCheckHybridAuth = vi.fn()
+let mockCheckHybridAuth: ReturnType<typeof vi.fn>
 const mockGetUserEntityPermissions = vi.fn()
 const mockGenerateInternalToken = vi.fn()
 const mockDbSelect = vi.fn()
@@ -61,9 +62,7 @@ describe('MCP Serve Route', () => {
  isDeployed: 'isDeployed',
  },
  }))
- vi.doMock('@/lib/auth/hybrid', () => ({
- checkHybridAuth: mockCheckHybridAuth,
- }))
+ ;({ mockCheckHybridAuth } = mockHybridAuth())
  vi.doMock('@/lib/workspaces/permissions/utils', () => ({
  getUserEntityPermissions: mockGetUserEntityPermissions,
  }))

diff --git a/apps/sim/app/api/schedules/[id]/route.test.ts b/apps/sim/app/api/schedules/[id]/route.test.ts
@@ -3,7 +3,7 @@
  *
  * @vitest-environment node
  */
-import { auditMock, databaseMock, loggerMock } from '@sim/testing'
+import { auditMock, databaseMock, loggerMock, requestUtilsMock } from '@sim/testing'
 import { NextRequest } from 'next/server'
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
 
@@ -31,9 +31,7 @@ vi.mock('drizzle-orm', () => ({
  eq: vi.fn(),
 }))
 
-vi.mock('@/lib/core/utils/request', () => ({
- generateRequestId: () => 'test-request-id',
-}))
+vi.mock('@/lib/core/utils/request', () => requestUtilsMock)
 
 vi.mock('@sim/logger', () => loggerMock)
 

diff --git a/apps/sim/app/api/schedules/route.test.ts b/apps/sim/app/api/schedules/route.test.ts
@@ -3,7 +3,7 @@
  *
  * @vitest-environment node
  */
-import { databaseMock, loggerMock } from '@sim/testing'
+import { databaseMock, loggerMock, requestUtilsMock } from '@sim/testing'
 import { NextRequest } from 'next/server'
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
 
@@ -43,9 +43,7 @@ vi.mock('drizzle-orm', () => ({
  isNull: vi.fn(),
 }))
 
-vi.mock('@/lib/core/utils/request', () => ({
- generateRequestId: () => 'test-request-id',
-}))
+vi.mock('@/lib/core/utils/request', () => requestUtilsMock)
 
 vi.mock('@sim/logger', () => loggerMock)