import { vi } from 'vitest'

export type RecordedJob = {

  queue: string

  name: string

  data: unknown

  options?: unknown

}

// Every job enqueued through a mocked queue lands here, so specs can assert

// on what the controllers/services scheduled without a real Valkey

export const recordedJobs: RecordedJob[] = []

function createQueueMock(name: string) {

  return {

    name,

    add: vi.fn(async (jobName: string, data: unknown, options?: unknown) => {

      recordedJobs.push({ queue: name, name: jobName, data, options })

      return {}

    }),

    addBulk: vi.fn(async (jobs: { name: string; data: unknown; opts?: unknown }[] = []) => {

      for (const job of jobs) {

        recordedJobs.push({

          queue: name,

          name: job.name,

          data: job.data,

          options: job.opts

        })

      }

      return []

    }),

    getRepeatableJobs: vi.fn(async () => []),

    removeRepeatableByKey: vi.fn(async () => undefined),

  }

}

const queues = new Map<string, ReturnType<typeof createQueueMock>>()

export function getQueueMock(name: string) {

  if (!queues.has(name)) {

    queues.set(name, createQueueMock(name))

  }

  return queues.get(name)!

}

export function resetQueueMocks() {

  recordedJobs.length = 0

  for (const queue of queues.values()) {

    queue.add.mockClear()

    queue.addBulk.mockClear()

    queue.getRepeatableJobs.mockClear()

    queue.removeRepeatableByKey.mockClear()

  }

}