/**
* Unit tests for Extension Setup Prompt
*
* Tests the prompt handler returns correct content for each extension.
*/
import { describe, it, expect, beforeEach } from "vitest";
import { createExtensionSetupPrompt } from "../extensionSetup.js";
import type {
PromptDefinition,
RequestContext,
} from "../../../../types/index.js";
describe("Extension Setup Prompt", () => {
let prompt: PromptDefinition;
let mockContext: RequestContext;
beforeEach(() => {
prompt = createExtensionSetupPrompt();
mockContext = {
timestamp: new Date(),
requestId: "test-request-123",
};
});
describe("prompt definition", () => {
it("should have correct name", () => {
expect(prompt.name).toBe("pg_extension_setup");
});
it("should have description", () => {
expect(prompt.description).toBeDefined();
expect(prompt.description.length).toBeGreaterThan(0);
});
it("should have extensionName argument", () => {
expect(prompt.arguments).toBeDefined();
expect(
prompt.arguments?.some((arg) => arg.name === "extensionName"),
).toBe(true);
});
it("should have extensionName as required", () => {
const extensionArg = prompt.arguments?.find(
(arg) => arg.name === "extensionName",
);
expect(extensionArg?.required).toBe(true);
});
});
describe("pg_stat_statements extension", () => {
it("should return setup content", async () => {
const content = await prompt.handler(
{ extensionName: "pg_stat_statements" },
mockContext,
);
expect(content).toContain("pg_stat_statements");
});
it("should include shared_preload_libraries config", async () => {
const content = await prompt.handler(
{ extensionName: "pg_stat_statements" },
mockContext,
);
expect(content).toContain("shared_preload_libraries");
});
it("should mention restart requirement", async () => {
const content = await prompt.handler(
{ extensionName: "pg_stat_statements" },
mockContext,
);
expect(content).toContain("Restart");
});
});
describe("hypopg extension", () => {
it("should return setup content", async () => {
const content = await prompt.handler(
{ extensionName: "hypopg" },
mockContext,
);
expect(content).toContain("hypopg");
});
it("should include hypopg_create_index example", async () => {
const content = await prompt.handler(
{ extensionName: "hypopg" },
mockContext,
);
expect(content).toContain("hypopg_create_index");
});
it("should mention no configuration needed", async () => {
const content = await prompt.handler(
{ extensionName: "hypopg" },
mockContext,
);
expect(content).toContain("No configuration needed");
});
});
describe("pgvector extension", () => {
it("should return setup content", async () => {
const content = await prompt.handler(
{ extensionName: "pgvector" },
mockContext,
);
expect(content).toContain("pgvector");
});
it("should include vector column example", async () => {
const content = await prompt.handler(
{ extensionName: "pgvector" },
mockContext,
);
expect(content).toContain("vector");
});
it("should mention HNSW index", async () => {
const content = await prompt.handler(
{ extensionName: "pgvector" },
mockContext,
);
expect(content).toContain("hnsw");
});
it("should reference pg_setup_pgvector prompt", async () => {
const content = await prompt.handler(
{ extensionName: "pgvector" },
mockContext,
);
expect(content).toContain("pg_setup_pgvector");
});
});
describe("postgis extension", () => {
it("should return setup content", async () => {
const content = await prompt.handler(
{ extensionName: "postgis" },
mockContext,
);
expect(content).toContain("postgis");
});
it("should include PostGIS_Full_Version check", async () => {
const content = await prompt.handler(
{ extensionName: "postgis" },
mockContext,
);
expect(content).toContain("PostGIS_Full_Version");
});
it("should include GEOGRAPHY type example", async () => {
const content = await prompt.handler(
{ extensionName: "postgis" },
mockContext,
);
expect(content).toContain("GEOGRAPHY");
});
it("should mention GIST index", async () => {
const content = await prompt.handler(
{ extensionName: "postgis" },
mockContext,
);
expect(content).toContain("GIST");
});
});
describe("pg_cron extension", () => {
it("should return setup content", async () => {
const content = await prompt.handler(
{ extensionName: "pg_cron" },
mockContext,
);
expect(content).toContain("pg_cron");
});
it("should include cron.schedule example", async () => {
const content = await prompt.handler(
{ extensionName: "pg_cron" },
mockContext,
);
expect(content).toContain("cron.schedule");
});
it("should include cron.database_name config", async () => {
const content = await prompt.handler(
{ extensionName: "pg_cron" },
mockContext,
);
expect(content).toContain("cron.database_name");
});
});
describe("pg_partman extension", () => {
it("should return setup content", async () => {
const content = await prompt.handler(
{ extensionName: "pg_partman" },
mockContext,
);
expect(content).toContain("pg_partman");
});
it("should include partman.create_parent example", async () => {
const content = await prompt.handler(
{ extensionName: "pg_partman" },
mockContext,
);
expect(content).toContain("partman.create_parent");
});
});
describe("pg_stat_kcache extension", () => {
it("should return setup content", async () => {
const content = await prompt.handler(
{ extensionName: "pg_stat_kcache" },
mockContext,
);
expect(content).toContain("pg_stat_kcache");
});
it("should include shared_preload_libraries config", async () => {
const content = await prompt.handler(
{ extensionName: "pg_stat_kcache" },
mockContext,
);
expect(content).toContain("shared_preload_libraries");
});
});
describe("citext extension", () => {
it("should return setup content", async () => {
const content = await prompt.handler(
{ extensionName: "citext" },
mockContext,
);
expect(content).toContain("citext");
});
it("should include CITEXT type example", async () => {
const content = await prompt.handler(
{ extensionName: "citext" },
mockContext,
);
expect(content).toContain("CITEXT");
});
});
describe("ltree extension", () => {
it("should return setup content", async () => {
const content = await prompt.handler(
{ extensionName: "ltree" },
mockContext,
);
expect(content).toContain("ltree");
});
it("should include LTREE type example", async () => {
const content = await prompt.handler(
{ extensionName: "ltree" },
mockContext,
);
expect(content).toContain("LTREE");
});
it("should mention GIST index", async () => {
const content = await prompt.handler(
{ extensionName: "ltree" },
mockContext,
);
expect(content).toContain("GIST");
});
});
describe("pgcrypto extension", () => {
it("should return setup content", async () => {
const content = await prompt.handler(
{ extensionName: "pgcrypto" },
mockContext,
);
expect(content).toContain("pgcrypto");
});
it("should include crypt function example", async () => {
const content = await prompt.handler(
{ extensionName: "pgcrypto" },
mockContext,
);
expect(content).toContain("crypt");
});
it("should include gen_salt example", async () => {
const content = await prompt.handler(
{ extensionName: "pgcrypto" },
mockContext,
);
expect(content).toContain("gen_salt");
});
});
describe("unknown extension", () => {
it("should return generic content for unknown extension", async () => {
const content = await prompt.handler(
{ extensionName: "unknown_extension" },
mockContext,
);
expect(content).toContain("unknown_extension");
});
it("should include generic configuration message", async () => {
const content = await prompt.handler(
{ extensionName: "some_other_ext" },
mockContext,
);
expect(content).toContain("Extension-specific configuration may vary");
});
});
describe("default extension", () => {
it("should use pg_stat_statements when no extension specified", async () => {
const content = await prompt.handler({}, mockContext);
expect(content).toContain("pg_stat_statements");
});
});
describe("common content", () => {
it("should include best practices section", async () => {
const content = await prompt.handler(
{ extensionName: "pgvector" },
mockContext,
);
expect(content).toContain("Best Practices");
});
it("should include troubleshooting section", async () => {
const content = await prompt.handler(
{ extensionName: "postgis" },
mockContext,
);
expect(content).toContain("Troubleshooting");
});
it("should include CREATE EXTENSION command", async () => {
const content = await prompt.handler(
{ extensionName: "citext" },
mockContext,
);
expect(content).toContain("CREATE EXTENSION");
});
it("should include pg_available_extensions check", async () => {
const content = await prompt.handler(
{ extensionName: "ltree" },
mockContext,
);
expect(content).toContain("pg_available_extensions");
});
});
});
