Function mockDb

apps/sim/lib/logs/execution/logger.test.ts:502–515

(billedRows: Array<Record<string, unknown>>)

  // (terminated by .limit), then the already-billed usage_log rows (terminated
  // by .groupBy). Return each in order.
  const mockDb = (billedRows: Array<Record<string, unknown>>) => {
    let call = 0
    dbSelectMock.mockImplementation(() => {
      call += 1
      const rows = call === 1 ? [{ id: 'workflow-1', workspaceId: 'ws-1' }] : billedRows
      const chain: any = {
        from: () => chain,
        where: () => chain,
        limit: () => Promise.resolve(rows),
        groupBy: () => Promise.resolve(rows),
      }
      return chain
    })
  }

  const run = (
    summary: ReturnType<typeof costSummary>,