Function mockCandidates

apps/sim/lib/data-drains/dispatcher.test.ts:26–32

(rows: Array<{ id: string; organizationId: string }>)

import { dispatchDueDrains, reapOrphanedRuns } from '@/lib/data-drains/dispatcher'

function mockCandidates(rows: Array<{ id: string; organizationId: string }>) {
  // db.select().from().where() — override `from` so awaiting `.where(pred)`
  // resolves with the candidate rows.
  dbChainMockFns.from.mockReturnValueOnce({
    where: vi.fn().mockResolvedValueOnce(rows),
  } as never)
}

beforeEach(() => {
  vi.clearAllMocks()