mirror of
https://github.com/kikootwo/ReadMeABook.git
synced 2026-09-01 05:08:38 +00:00
143 lines
4.9 KiB
TypeScript
143 lines
4.9 KiB
TypeScript
/**
|
|
* Component: Audiobooks Search Torrents API Route Tests
|
|
* Documentation: documentation/testing.md
|
|
*/
|
|
|
|
import { beforeEach, describe, expect, it, vi } from 'vitest';
|
|
|
|
let authRequest: any;
|
|
|
|
const requireAuthMock = vi.hoisted(() => vi.fn());
|
|
const configServiceMock = vi.hoisted(() => ({
|
|
get: vi.fn(),
|
|
getAudibleRegion: vi.fn().mockResolvedValue('us'),
|
|
}));
|
|
const prowlarrMock = vi.hoisted(() => ({
|
|
search: vi.fn(),
|
|
searchWithVariations: vi.fn(),
|
|
}));
|
|
const rankTorrentsMock = vi.hoisted(() => vi.fn());
|
|
const groupIndexersMock = vi.hoisted(() => vi.fn());
|
|
const groupDescriptionMock = vi.hoisted(() => vi.fn(() => 'Group'));
|
|
|
|
vi.mock('@/lib/db', () => ({
|
|
prisma: {},
|
|
}));
|
|
|
|
vi.mock('@/lib/middleware/auth', () => ({
|
|
requireAuth: requireAuthMock,
|
|
}));
|
|
|
|
vi.mock('@/lib/services/config.service', () => ({
|
|
getConfigService: () => configServiceMock,
|
|
}));
|
|
|
|
vi.mock('@/lib/integrations/prowlarr.service', () => ({
|
|
getProwlarrService: async () => prowlarrMock,
|
|
}));
|
|
|
|
vi.mock('@/lib/utils/ranking-algorithm', () => ({
|
|
rankTorrents: rankTorrentsMock,
|
|
}));
|
|
|
|
vi.mock('@/lib/utils/indexer-grouping', () => ({
|
|
groupIndexersByCategories: groupIndexersMock,
|
|
getGroupDescription: groupDescriptionMock,
|
|
}));
|
|
|
|
describe('Audiobooks search torrents route', () => {
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
configServiceMock.getAudibleRegion.mockResolvedValue('us');
|
|
authRequest = {
|
|
user: { id: 'user-1', role: 'user' },
|
|
json: vi.fn(),
|
|
};
|
|
requireAuthMock.mockImplementation((_req: any, handler: any) => handler(authRequest));
|
|
});
|
|
|
|
it('returns error when no indexers are configured', async () => {
|
|
authRequest.json.mockResolvedValue({ title: 'Title', author: 'Author' });
|
|
configServiceMock.get.mockResolvedValueOnce(null);
|
|
|
|
const { POST } = await import('@/app/api/audiobooks/search-torrents/route');
|
|
const response = await POST({} as any);
|
|
const payload = await response.json();
|
|
|
|
expect(response.status).toBe(400);
|
|
expect(payload.error).toBe('ConfigError');
|
|
});
|
|
|
|
it('returns ranked results with rank order', async () => {
|
|
authRequest.json.mockResolvedValue({ title: 'Title', author: 'Author' });
|
|
configServiceMock.get
|
|
.mockResolvedValueOnce(JSON.stringify([{ id: 1, name: 'Indexer', protocol: 'torrent', priority: 10 }]))
|
|
.mockResolvedValueOnce(null);
|
|
|
|
groupIndexersMock.mockReturnValue({ groups: [{ categories: [1], indexerIds: [1] }], skippedIndexers: [] });
|
|
prowlarrMock.searchWithVariations.mockResolvedValue([{ title: 'Result', size: 100, indexer: 'Indexer', indexerId: 1 }]);
|
|
rankTorrentsMock.mockReturnValue([
|
|
{
|
|
title: 'Result',
|
|
size: 100,
|
|
indexer: 'Indexer',
|
|
indexerId: 1,
|
|
score: 50,
|
|
breakdown: { matchScore: 50, formatScore: 0, sizeScore: 0, seederScore: 0, notes: [] },
|
|
bonusPoints: 0,
|
|
bonusModifiers: [],
|
|
finalScore: 50,
|
|
},
|
|
]);
|
|
|
|
const { POST } = await import('@/app/api/audiobooks/search-torrents/route');
|
|
const response = await POST({} as any);
|
|
const payload = await response.json();
|
|
|
|
expect(payload.success).toBe(true);
|
|
expect(payload.results[0].rank).toBe(1);
|
|
expect(rankTorrentsMock).toHaveBeenCalled();
|
|
});
|
|
|
|
it('ranks all candidates before returning the top 100 with truncation metadata', async () => {
|
|
authRequest.json.mockResolvedValue({ title: 'Title', author: 'Author' });
|
|
configServiceMock.get
|
|
.mockResolvedValueOnce(JSON.stringify([{ id: 1, name: 'Indexer', protocol: 'torrent', priority: 10 }]))
|
|
.mockResolvedValueOnce(null);
|
|
groupIndexersMock.mockReturnValue({ groups: [{ categories: [3030], indexerIds: [1] }], skippedIndexers: [] });
|
|
|
|
const candidates = Array.from({ length: 101 }, (_, index) => ({
|
|
title: index === 100 ? 'Best result' : `Result ${index}`,
|
|
size: 100,
|
|
indexer: 'Indexer',
|
|
indexerId: 1,
|
|
}));
|
|
const rankedCandidates = [candidates[100], ...candidates.slice(0, 100)].map((result, index) => ({
|
|
...result,
|
|
score: 100 - index,
|
|
breakdown: { matchScore: 50, formatScore: 0, sizeScore: 0, seederScore: 0, notes: [] },
|
|
bonusPoints: 0,
|
|
bonusModifiers: [],
|
|
finalScore: 100 - index,
|
|
}));
|
|
prowlarrMock.searchWithVariations.mockResolvedValue(candidates);
|
|
rankTorrentsMock.mockReturnValue(rankedCandidates);
|
|
|
|
const { POST } = await import('@/app/api/audiobooks/search-torrents/route');
|
|
const response = await POST({} as any);
|
|
const payload = await response.json();
|
|
|
|
expect(rankTorrentsMock.mock.calls[0][0]).toHaveLength(101);
|
|
expect(prowlarrMock.searchWithVariations).toHaveBeenCalledWith('Title', 'Author', {
|
|
categories: [3030],
|
|
indexerIds: [1],
|
|
});
|
|
expect(payload.results).toHaveLength(100);
|
|
expect(payload.results[0]).toEqual(expect.objectContaining({ title: 'Best result', rank: 1 }));
|
|
expect(payload.truncated).toBe(true);
|
|
expect(payload.rawCount).toBe(101);
|
|
});
|
|
});
|
|
|
|
|