trueref/src/routes/api/v1/libs/[id]/index/server.test.ts

/**
 * Unit tests for POST /api/v1/libs/:id/index
 *
 * Verifies:
 * - Default-branch re-index also enqueues jobs for all registered versions
 * - versionJobs array is returned in the response
 * - Explicit versionId request does NOT trigger extra version jobs
 * - Returns 404 when repo does not exist
 */

import { beforeEach, describe, expect, it, vi } from 'vitest';
import Database from 'better-sqlite3';
import { readFileSync } from 'node:fs';
import { join } from 'node:path';
import { RepositoryService } from '$lib/server/services/repository.service';
import { VersionService } from '$lib/server/services/version.service';

let db: Database.Database;
let mockQueue: { enqueue: ReturnType<typeof vi.fn> } | null = null;

vi.mock('$lib/server/db/client', () => ({
	getClient: () => db
}));

vi.mock('$lib/server/db/client.js', () => ({
	getClient: () => db
}));

vi.mock('$lib/server/pipeline/startup', () => ({
	getQueue: () => mockQueue
}));

vi.mock('$lib/server/pipeline/startup.js', () => ({
	getQueue: () => mockQueue
}));

vi.mock('$lib/server/embeddings/registry', () => ({
	createProviderFromProfile: () => null
}));

vi.mock('$lib/server/embeddings/registry.js', () => ({
	createProviderFromProfile: () => null
}));

import { POST as postIndex } from './+server.js';

const NOW_S = Math.floor(Date.now() / 1000);

function createTestDb(): Database.Database {
	const client = new Database(':memory:');
	client.pragma('foreign_keys = ON');

	const migrationsFolder = join(import.meta.dirname, '../../../../../../lib/server/db/migrations');
	const ftsFile = join(import.meta.dirname, '../../../../../../lib/server/db/fts.sql');

	const migration0 = readFileSync(join(migrationsFolder, '0000_large_master_chief.sql'), 'utf-8');
	const migration1 = readFileSync(join(migrationsFolder, '0001_quick_nighthawk.sql'), 'utf-8');
	const migration2 = readFileSync(join(migrationsFolder, '0002_silky_stellaris.sql'), 'utf-8');

	for (const migration of [migration0, migration1, migration2]) {
		for (const stmt of migration
			.split('--> statement-breakpoint')
			.map((s) => s.trim())
			.filter(Boolean)) {
			client.exec(stmt);
		}
	}

	client.exec(readFileSync(ftsFile, 'utf-8'));
	return client;
}

function makeEnqueueJob(repositoryId: string, versionId?: string) {
	return {
		id: `job-${Math.random().toString(36).slice(2)}`,
		repositoryId,
		versionId: versionId ?? null,
		status: 'queued' as const,
		processedFiles: 0,
		totalFiles: 0,
		error: null,
		startedAt: null,
		completedAt: null,
		createdAt: new Date(NOW_S * 1000)
	};
}

describe('POST /api/v1/libs/:id/index', () => {
	beforeEach(() => {
		db = createTestDb();
		mockQueue = null;
	});

	it('returns 404 when repo does not exist', async () => {
		const response = await postIndex({
			params: { id: encodeURIComponent('/nonexistent/repo') },
			request: new Request('http://test', { method: 'POST' })
		} as never);

		expect(response.status).toBe(404);
	});

	it('returns job and empty versionJobs when no versions are registered', async () => {
		const repoService = new RepositoryService(db);
		repoService.add({ source: 'github', sourceUrl: 'https://github.com/facebook/react' });

		const response = await postIndex({
			params: { id: encodeURIComponent('/facebook/react') },
			request: new Request('http://test', { method: 'POST' })
		} as never);

		expect(response.status).toBe(202);
		const body = await response.json();
		expect(body.job).toBeDefined();
		expect(body.job.repositoryId).toBe('/facebook/react');
		expect(body.versionJobs).toEqual([]);
	});

	it('enqueues jobs for all registered versions on default-branch re-index', async () => {
		const repoService = new RepositoryService(db);
		const versionService = new VersionService(db);

		repoService.add({ source: 'github', sourceUrl: 'https://github.com/facebook/react' });
		versionService.add('/facebook/react', 'v18.3.0', 'React v18.3.0');
		versionService.add('/facebook/react', 'v17.0.0', 'React v17.0.0');

		const enqueue = vi.fn().mockImplementation(
			(repositoryId: string, versionId?: string) => makeEnqueueJob(repositoryId, versionId)
		);
		mockQueue = { enqueue };

		const response = await postIndex({
			params: { id: encodeURIComponent('/facebook/react') },
			request: new Request('http://test', { method: 'POST' })
		} as never);

		expect(response.status).toBe(202);
		const body = await response.json();

		// Main job enqueued (no versionId)
		expect(body.job).toBeDefined();
		expect(body.job.repositoryId).toBe('/facebook/react');

		// Two version jobs enqueued
		expect(body.versionJobs).toHaveLength(2);
		expect(enqueue).toHaveBeenCalledTimes(3); // 1 main + 2 versions

		// Version IDs should be the registered version IDs
		const enqueuedVersionIds = enqueue.mock.calls.slice(1).map((call) => call[1]);
		expect(enqueuedVersionIds).toContain('/facebook/react/v18.3.0');
		expect(enqueuedVersionIds).toContain('/facebook/react/v17.0.0');
	});

	it('does NOT enqueue version jobs when an explicit versionId is provided', async () => {
		const repoService = new RepositoryService(db);
		const versionService = new VersionService(db);

		repoService.add({ source: 'github', sourceUrl: 'https://github.com/facebook/react' });
		versionService.add('/facebook/react', 'v18.3.0', 'React v18.3.0');

		const enqueue = vi.fn().mockImplementation(
			(repositoryId: string, versionId?: string) => makeEnqueueJob(repositoryId, versionId)
		);
		mockQueue = { enqueue };

		const response = await postIndex({
			params: { id: encodeURIComponent('/facebook/react') },
			request: new Request('http://test', {
				method: 'POST',
				headers: { 'Content-Type': 'application/json' },
				body: JSON.stringify({ version: '/facebook/react/v18.3.0' })
			})
		} as never);

		expect(response.status).toBe(202);
		const body = await response.json();

		// Only one call — the explicit version, no extra version enumeration
		expect(enqueue).toHaveBeenCalledTimes(1);
		expect(body.versionJobs).toEqual([]);
	});
});