fix(api-key): read workspace key settings on the auth pool

Give the memory harness integration mock a dbFor so the billing pool resolves.

Author: waleedlatif1

apps/sim/executor/handlers/agent/memory-harness.integration.ts

@@ -30,8 +30,8 @@ const fixture = vi.hoisted(() => ({
   uploads: '',
 }))
 
-vi.mock('@sim/db', () => ({
-  db: new Proxy(
+vi.mock('@sim/db', () => {
+  const db = new Proxy(
     {},
     {
       get(_target, property) {
@@ -40,8 +40,9 @@ vi.mock('@sim/db', () => ({
         return typeof value === 'function' ? value.bind(fixture.database) : value
       },
     }
-  ),
-}))
+  )
+  return { db, dbFor: () => db }
+})
 vi.mock('@/lib/uploads/core/setup.server', () => ({
   get UPLOAD_DIR_SERVER() {
     return fixture.uploads

apps/sim/lib/api-key/service.ts

@@ -56,7 +56,7 @@ export async function authenticateApiKeyFromHeader(
     let workspaceSettings: WorkspaceBillingSettings | null = null
 
     if (options.workspaceId) {
-      workspaceSettings = await getWorkspaceBillingSettings(options.workspaceId)
+      workspaceSettings = await getWorkspaceBillingSettings(options.workspaceId, getAuthPool())
       if (!workspaceSettings) {
         return { success: false, error: 'Workspace not found' }
       }

apps/sim/lib/db/pool-isolation.integration.ts

@@ -125,4 +125,13 @@ describe('dedicated pools under a saturated process pool', () => {
     await expect(settlesWithin(lookup, ISOLATED_BUDGET_MS)).resolves.toBe('done')
     await expect(lookup).resolves.toEqual({ success: false, error: 'Invalid API key' })
   })
+
+  it('reads workspace key settings on the auth pool', async () => {
+    const lookup = authenticateApiKeyFromHeader(`sk-sim-${generateId()}`, {
+      workspaceId: generateId(),
+    })
+
+    await expect(settlesWithin(lookup, ISOLATED_BUDGET_MS)).resolves.toBe('done')
+    await expect(lookup).resolves.toEqual({ success: false, error: 'Workspace not found' })
+  })
 })

apps/sim/lib/workspaces/utils.ts

@@ -9,7 +9,7 @@ import {
   type ChangeWorkspaceStoragePayerParams,
   changeWorkspaceStoragePayerInTx,
 } from '@/lib/billing/storage/payer-transfer'
-import type { DbOrTx } from '@/lib/db/types'
+import type { DbClient, DbOrTx } from '@/lib/db/types'
 
 const logger = createLogger('WorkspaceUtils')
 
@@ -21,13 +21,14 @@ export interface WorkspaceBillingSettings {
 export type WorkspaceScope = 'active' | 'archived' | 'all'
 
 export async function getWorkspaceBillingSettings(
-  workspaceId: string
+  workspaceId: string,
+  executor: DbClient = db
 ): Promise<WorkspaceBillingSettings | null> {
   if (!workspaceId) {
     return null
   }
 
-  const rows = await db
+  const rows = await executor
     .select({
       billedAccountUserId: workspaceTable.billedAccountUserId,
       allowPersonalApiKeys: workspaceTable.allowPersonalApiKeys,