mirror of
https://github.com/kikootwo/ReadMeABook.git
synced 2026-06-03 21:00:09 +00:00
5d8ac2f73d
Introduce centralized language configuration and wire locale-aware behavior across scraping and ranking. Adds src/lib/constants/language-config.ts with per-language scraping rules, stop words, and character replacements; replaces AudibleRegion.isEnglish with a language field in types and AUDIBLE_REGIONS. Update AudibleService, ebook scraper, processors, and API routes to use getLanguageForRegion so Anna's Archive searches, scraping selectors, runtime/rating parsing, and ranking use language-specific params and filters. Extend ranking algorithm to accept stopWords and characterReplacements and apply them during normalization and matching. Update UI selects to mark non-English regions and adjust tests accordingly.
130 lines
4.0 KiB
TypeScript
130 lines
4.0 KiB
TypeScript
/**
|
|
* Component: Search Indexers Processor Tests
|
|
* Documentation: documentation/backend/services/jobs.md
|
|
*/
|
|
|
|
import { beforeEach, describe, expect, it, vi } from 'vitest';
|
|
import { createPrismaMock } from '../helpers/prisma';
|
|
import { createJobQueueMock } from '../helpers/job-queue';
|
|
|
|
const prismaMock = createPrismaMock();
|
|
const configMock = vi.hoisted(() => ({ get: vi.fn(), getAudibleRegion: vi.fn().mockResolvedValue('us') }));
|
|
const jobQueueMock = createJobQueueMock();
|
|
const prowlarrMock = vi.hoisted(() => ({ search: vi.fn(), searchWithVariations: vi.fn() }));
|
|
|
|
vi.mock('@/lib/db', () => ({
|
|
prisma: prismaMock,
|
|
}));
|
|
|
|
vi.mock('@/lib/services/config.service', () => ({
|
|
getConfigService: () => configMock,
|
|
}));
|
|
|
|
vi.mock('@/lib/services/job-queue.service', () => ({
|
|
getJobQueueService: () => jobQueueMock,
|
|
}));
|
|
|
|
vi.mock('@/lib/integrations/prowlarr.service', () => ({
|
|
getProwlarrService: () => prowlarrMock,
|
|
}));
|
|
|
|
vi.mock('@/lib/integrations/audible.service', () => ({
|
|
getAudibleService: () => ({ getRuntime: vi.fn().mockResolvedValue(null) }),
|
|
}));
|
|
|
|
describe('processSearchIndexers', () => {
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
configMock.getAudibleRegion.mockResolvedValue('us');
|
|
});
|
|
|
|
it('marks request awaiting_search when no results found', async () => {
|
|
configMock.get.mockImplementation(async (key: string) => {
|
|
if (key === 'prowlarr_indexers') {
|
|
return JSON.stringify([{ id: 1, name: 'Indexer', protocol: 'torrent', priority: 10, categories: [3030] }]);
|
|
}
|
|
return null;
|
|
});
|
|
prowlarrMock.searchWithVariations.mockResolvedValue([]);
|
|
prismaMock.request.update.mockResolvedValue({});
|
|
|
|
const { processSearchIndexers } = await import('@/lib/processors/search-indexers.processor');
|
|
const result = await processSearchIndexers({
|
|
requestId: 'req-1',
|
|
audiobook: { id: 'a1', title: 'Book', author: 'Author' },
|
|
jobId: 'job-1',
|
|
});
|
|
|
|
expect(result.success).toBe(false);
|
|
expect(prismaMock.request.update).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
data: expect.objectContaining({ status: 'awaiting_search' }),
|
|
})
|
|
);
|
|
});
|
|
|
|
it('queues download job when results are ranked', async () => {
|
|
configMock.get.mockImplementation(async (key: string) => {
|
|
if (key === 'prowlarr_indexers') {
|
|
return JSON.stringify([{ id: 1, name: 'Indexer', protocol: 'torrent', priority: 10, categories: [3030] }]);
|
|
}
|
|
if (key === 'indexer_flag_config') {
|
|
return JSON.stringify([]);
|
|
}
|
|
return null;
|
|
});
|
|
|
|
prowlarrMock.searchWithVariations.mockResolvedValue([
|
|
{
|
|
indexer: 'Indexer',
|
|
indexerId: 1,
|
|
title: 'Book - Author',
|
|
size: 50 * 1024 * 1024,
|
|
seeders: 10,
|
|
publishDate: new Date(),
|
|
downloadUrl: 'magnet:?xt=urn:btih:abc',
|
|
guid: 'guid-1',
|
|
format: 'M4B',
|
|
},
|
|
]);
|
|
|
|
prismaMock.request.update.mockResolvedValue({});
|
|
|
|
const { processSearchIndexers } = await import('@/lib/processors/search-indexers.processor');
|
|
const result = await processSearchIndexers({
|
|
requestId: 'req-2',
|
|
audiobook: { id: 'a2', title: 'Book', author: 'Author' },
|
|
jobId: 'job-2',
|
|
});
|
|
|
|
expect(result.success).toBe(true);
|
|
expect(jobQueueMock.addDownloadJob).toHaveBeenCalledWith(
|
|
'req-2',
|
|
{ id: 'a2', title: 'Book', author: 'Author' },
|
|
expect.objectContaining({ title: 'Book - Author' })
|
|
);
|
|
});
|
|
|
|
it('fails when no indexers are configured', async () => {
|
|
configMock.get.mockResolvedValue(null);
|
|
prismaMock.request.update.mockResolvedValue({});
|
|
|
|
const { processSearchIndexers } = await import('@/lib/processors/search-indexers.processor');
|
|
await expect(
|
|
processSearchIndexers({
|
|
requestId: 'req-3',
|
|
audiobook: { id: 'a3', title: 'Book', author: 'Author' },
|
|
jobId: 'job-3',
|
|
})
|
|
).rejects.toThrow('No indexers configured');
|
|
|
|
expect(prismaMock.request.update).toHaveBeenCalledWith(
|
|
expect.objectContaining({
|
|
data: expect.objectContaining({ status: 'failed' }),
|
|
})
|
|
);
|
|
});
|
|
});
|
|
|
|
|