ReadMeABook/tests/api/admin-downloads.routes.test.ts

/**
 * Component: Admin Downloads API Route Tests
 * Documentation: documentation/testing.md
 */

import { beforeEach, describe, expect, it, vi } from 'vitest';
import { createPrismaMock } from '../helpers/prisma';

let authRequest: any;

const prismaMock = createPrismaMock();
const requireAuthMock = vi.hoisted(() => vi.fn());
const requireAdminMock = vi.hoisted(() => vi.fn());
const configServiceMock = vi.hoisted(() => ({ get: vi.fn() }));
const qbittorrentMock = vi.hoisted(() => ({ getTorrent: vi.fn() }));
const sabnzbdMock = vi.hoisted(() => ({ getNZB: vi.fn() }));
const downloadClientManagerMock = vi.hoisted(() => ({
  getClientServiceForProtocol: vi.fn(),
}));

vi.mock('@/lib/db', () => ({
  prisma: prismaMock,
}));

vi.mock('@/lib/middleware/auth', () => ({
  requireAuth: requireAuthMock,
  requireAdmin: requireAdminMock,
}));

vi.mock('@/lib/services/config.service', () => ({
  getConfigService: () => configServiceMock,
}));

vi.mock('@/lib/integrations/qbittorrent.service', () => ({
  getQBittorrentService: async () => qbittorrentMock,
}));

vi.mock('@/lib/integrations/sabnzbd.service', () => ({
  getSABnzbdService: async () => sabnzbdMock,
}));

vi.mock('@/lib/services/download-client-manager.service', () => ({
  getDownloadClientManager: () => downloadClientManagerMock,
}));

describe('Admin downloads route', () => {
  beforeEach(() => {
    vi.clearAllMocks();
    authRequest = { user: { id: 'admin-1', role: 'admin' } };
    requireAuthMock.mockImplementation((_req: any, handler: any) => handler(authRequest));
    requireAdminMock.mockImplementation((_req: any, handler: any) => handler());
    downloadClientManagerMock.getClientServiceForProtocol.mockReset();
  });

  it('returns formatted active downloads', async () => {
    prismaMock.request.findMany.mockResolvedValueOnce([
      {
        id: 'req-1',
        status: 'downloading',
        progress: 50,
        updatedAt: new Date(),
        audiobook: { title: 'Title', author: 'Author' },
        user: { plexUsername: 'user' },
        downloadHistory: [{ torrentHash: 'hash', torrentName: 'Torrent', downloadStatus: 'downloading', downloadClient: 'qbittorrent' }],
      },
    ]);
    downloadClientManagerMock.getClientServiceForProtocol.mockResolvedValueOnce({
      getDownload: vi.fn().mockResolvedValue({
        downloadSpeed: 123,
        eta: 60,
      }),
    });

    const { GET } = await import('@/app/api/admin/downloads/active/route');
    const response = await GET({} as any);
    const payload = await response.json();

    expect(payload.downloads[0].speed).toBe(123);
    expect(payload.downloads[0].torrentName).toBe('Torrent');
  });

  it('returns formatted active downloads for SABnzbd', async () => {
    prismaMock.request.findMany.mockResolvedValueOnce([
      {
        id: 'req-2',
        status: 'downloading',
        progress: 20,
        updatedAt: new Date(),
        audiobook: { title: 'Title', author: 'Author' },
        user: { plexUsername: 'user' },
        downloadHistory: [{ nzbId: 'nzb-1', torrentName: 'NZB', downloadStatus: 'downloading', downloadClient: 'sabnzbd' }],
      },
    ]);
    downloadClientManagerMock.getClientServiceForProtocol.mockResolvedValueOnce({
      getDownload: vi.fn().mockResolvedValue({
        downloadSpeed: 555,
        eta: 120,
      }),
    });

    const { GET } = await import('@/app/api/admin/downloads/active/route');
    const response = await GET({} as any);
    const payload = await response.json();

    expect(payload.downloads[0].speed).toBe(555);
    expect(payload.downloads[0].eta).toBe(120);
  });

  it('returns defaults when download client lookup fails', async () => {
    prismaMock.request.findMany.mockResolvedValueOnce([
      {
        id: 'req-3',
        status: 'downloading',
        progress: 80,
        updatedAt: new Date(),
        audiobook: { title: 'Title', author: 'Author' },
        user: { plexUsername: 'user' },
        downloadHistory: [{ torrentHash: 'hash', torrentName: 'Torrent', downloadStatus: 'downloading', downloadClient: 'qbittorrent' }],
      },
    ]);
    downloadClientManagerMock.getClientServiceForProtocol.mockResolvedValueOnce({
      getDownload: vi.fn().mockRejectedValue(new Error('client down')),
    });

    const { GET } = await import('@/app/api/admin/downloads/active/route');
    const response = await GET({} as any);
    const payload = await response.json();

    expect(payload.downloads[0].speed).toBe(0);
    expect(payload.downloads[0].eta).toBeNull();
  });
});