queue.test.ts

import { SYSTEM_TENANT } from '@internal/queue'
import { vi } from 'vitest'

const { mockMoveJobsSend } = vi.hoisted(() => ({
  mockMoveJobsSend: vi.fn(),
}))

vi.mock('@storage/events', () => ({
  MoveJobs: {
    send: mockMoveJobsSend,
  },
}))

describe('admin queue routes', () => {
  beforeEach(() => {
    vi.clearAllMocks()
  })

  it('passes sbReqId to the move jobs task', async () => {
    vi.resetModules()

    const { getConfig, mergeConfig } = await import('../../../config')
    getConfig()
    mergeConfig({
      pgQueueEnable: true,
      adminApiKeys: 'test-admin-key',
    })

    const fastify = (await import('fastify')).default
    const { default: routes } = await import('./queue')

    const app = fastify()
    app.decorateRequest('sbReqId', undefined)
    app.addHook('onRequest', (request, _reply, done) => {
      request.sbReqId =
        typeof request.headers['sb-request-id'] === 'string'
          ? request.headers['sb-request-id']
          : undefined
      done()
    })
    app.register(routes, { prefix: '/queue' })

    try {
      const response = await app.inject({
        method: 'POST',
        url: '/queue/move',
        headers: {
          apikey: 'test-admin-key',
          'sb-request-id': 'sb-req-123',
        },
        payload: {
          fromQueue: 'source-queue',
          toQueue: 'target-queue',
          deleteJobsFromOriginalQueue: true,
        },
      })

      expect(response.statusCode).toBe(200)
      expect(response.json()).toEqual({ message: 'Move jobs scheduled' })
      expect(mockMoveJobsSend).toHaveBeenCalledWith({
        fromQueue: 'source-queue',
        toQueue: 'target-queue',
        deleteJobsFromOriginalQueue: true,
        sbReqId: 'sb-req-123',
        tenant: SYSTEM_TENANT,
      })
    } finally {
      await app.close()
    }
  })

  it('rejects move jobs requests without queue names', async () => {
    vi.resetModules()

    const { getConfig, mergeConfig } = await import('../../../config')
    getConfig()
    mergeConfig({
      pgQueueEnable: true,
      adminApiKeys: 'test-admin-key',
    })

    const fastify = (await import('fastify')).default
    const { default: routes } = await import('./queue')

    const app = fastify()
    app.decorateRequest('sbReqId', undefined)
    app.register(routes, { prefix: '/queue' })

    try {
      const response = await app.inject({
        method: 'POST',
        url: '/queue/move',
        headers: {
          apikey: 'test-admin-key',
        },
        payload: {},
      })

      expect(response.statusCode).toBe(400)
      expect(mockMoveJobsSend).not.toHaveBeenCalled()
    } finally {
      await app.close()
    }
  })

  it('rejects queue overflow restore on OrioleDB', async () => {
    vi.resetModules()

    const { getConfig, mergeConfig } = await import('../../../config')
    getConfig()
    mergeConfig({
      pgQueueEnable: true,
      adminApiKeys: 'test-admin-key',
      databaseEngine: 'oriole',
    })

    const fastify = (await import('fastify')).default
    const { default: routes } = await import('./queue')

    const app = fastify()
    app.register(routes, { prefix: '/queue' })

    try {
      const response = await app.inject({
        method: 'POST',
        url: '/queue/overflow/restore',
        headers: {
          apikey: 'test-admin-key',
        },
        payload: {
          name: 'webhooks',
        },
      })

      expect(response.statusCode).toBe(400)
      expect(response.json()).toEqual({
        message: 'Queue overflow restore is not supported on OrioleDB',
      })
    } finally {
      await app.close()
    }
  })
})