Function createMockTx

apps/sim/lib/workflows/persistence/duplicate.test.ts:29–69

(
  selectResults: unknown[],
  onWorkflowInsert?: (values: Record<string, unknown>) => void,
  onInsert?: (values: unknown) => void
)

import { duplicateWorkflow } from './duplicate'

function createMockTx(
  selectResults: unknown[],
  onWorkflowInsert?: (values: Record<string, unknown>) => void,
  onInsert?: (values: unknown) => void
) {
  let selectCallCount = 0

  const select = vi.fn().mockImplementation(() => ({
    from: vi.fn().mockReturnValue({
      where: vi.fn().mockImplementation(() => {
        const result = selectResults[selectCallCount++] ?? []
        if (selectCallCount === 1) {
          return {
            limit: vi.fn().mockResolvedValue(result),
          }
        }
        return Promise.resolve(result)
      }),
    }),
  }))

  const insert = vi.fn().mockReturnValue({
    values: vi.fn().mockImplementation((values: Record<string, unknown>) => {
      onWorkflowInsert?.(values)
      onInsert?.(values)
      return Promise.resolve(undefined)
    }),
  })

  const update = vi.fn().mockReturnValue({
    set: vi.fn().mockReturnValue({
      where: vi.fn().mockResolvedValue(undefined),
    }),
  })

  return {
    select,
    insert,
    update,
  }
}

describe('duplicateWorkflow ordering', () => {
  beforeEach(() => {