emdash-cms--emdash/packages/core/tests/integration/search/suggest.test.ts

import type { Kysely } from "kysely";
import { describe, it, expect, beforeEach, afterEach } from "vitest";

import { ContentRepository } from "../../../src/database/repositories/content.js";
import type { Database } from "../../../src/database/types.js";
import { SchemaRegistry } from "../../../src/schema/registry.js";
import { FTSManager } from "../../../src/search/fts-manager.js";
import { getSuggestions } from "../../../src/search/query.js";
import { createPostFixture } from "../../utils/fixtures.js";
import { setupTestDatabaseWithCollections, teardownTestDatabase } from "../../utils/test-db.js";

describe("getSuggestions (Integration)", () => {
	let db: Kysely<Database>;
	let repo: ContentRepository;

	beforeEach(async () => {
		db = await setupTestDatabaseWithCollections();
		repo = new ContentRepository(db);

		const registry = new SchemaRegistry(db);
		const ftsManager = new FTSManager(db);
		await registry.updateField("post", "title", { searchable: true });
		await ftsManager.enableSearch("post");

		await repo.create(
			createPostFixture({
				slug: "designing-things",
				status: "published",
				data: { title: "Designing things" },
			}),
		);
	});

	afterEach(async () => {
		await teardownTestDatabase(db);
	});

	it("returns matching suggestions for a plain prefix query", async () => {
		const suggestions = await getSuggestions(db, "des", {
			collections: ["post"],
		});

		expect(suggestions).toHaveLength(1);
		expect(suggestions[0]).toMatchObject({
			collection: "post",
			slug: "designing-things",
			title: "Designing things",
		});
	});

	it("does not return draft content", async () => {
		await repo.create(
			createPostFixture({
				slug: "designing-secret-things",
				status: "draft",
				data: { title: "Designing secret things" },
			}),
		);

		const suggestions = await getSuggestions(db, "des", {
			collections: ["post"],
		});

		expect(suggestions).toHaveLength(1);
		expect(suggestions[0]?.title).toBe("Designing things");
	});

	it("returns empty array for a non-matching query", async () => {
		const suggestions = await getSuggestions(db, "zzz", {
			collections: ["post"],
		});

		expect(suggestions).toEqual([]);
	});
});