/**
* Unit tests for Tool Annotations Presets
*
* Tests the annotation presets and helper functions.
*/
import { describe, it, expect } from "vitest";
import {
READ_ONLY,
WRITE,
DESTRUCTIVE,
IDEMPOTENT,
ADMIN,
withTitle,
readOnly,
write,
destructive,
idempotent,
admin,
} from "../annotations.js";
describe("Tool Annotations Presets", () => {
describe("Base Presets", () => {
it("READ_ONLY should have correct flags", () => {
expect(READ_ONLY.readOnlyHint).toBe(true);
expect(READ_ONLY.destructiveHint).toBe(false);
});
it("WRITE should have correct flags", () => {
expect(WRITE.readOnlyHint).toBe(false);
expect(WRITE.destructiveHint).toBe(false);
});
it("DESTRUCTIVE should have correct flags", () => {
expect(DESTRUCTIVE.readOnlyHint).toBe(false);
expect(DESTRUCTIVE.destructiveHint).toBe(true);
});
it("IDEMPOTENT should have correct flags", () => {
expect(IDEMPOTENT.readOnlyHint).toBe(false);
expect(IDEMPOTENT.destructiveHint).toBe(false);
expect(IDEMPOTENT.idempotentHint).toBe(true);
});
it("ADMIN should have correct flags", () => {
expect(ADMIN.readOnlyHint).toBe(false);
expect(ADMIN.destructiveHint).toBe(false);
});
});
describe("Helper Functions", () => {
describe("withTitle", () => {
it("should add title to default READ_ONLY base", () => {
const result = withTitle("List Tables");
expect(result.title).toBe("List Tables");
expect(result.readOnlyHint).toBe(true);
expect(result.destructiveHint).toBe(false);
});
it("should add title to custom base annotations", () => {
const result = withTitle("Delete Row", DESTRUCTIVE);
expect(result.title).toBe("Delete Row");
expect(result.destructiveHint).toBe(true);
});
it("should add title to WRITE base", () => {
const result = withTitle("Insert Data", WRITE);
expect(result.title).toBe("Insert Data");
expect(result.readOnlyHint).toBe(false);
});
});
describe("readOnly", () => {
it("should create read-only annotations with title", () => {
const result = readOnly("Query Data");
expect(result.title).toBe("Query Data");
expect(result.readOnlyHint).toBe(true);
expect(result.destructiveHint).toBe(false);
});
});
describe("write", () => {
it("should create write annotations with title", () => {
const result = write("Create Table");
expect(result.title).toBe("Create Table");
expect(result.readOnlyHint).toBe(false);
expect(result.destructiveHint).toBe(false);
});
});
describe("destructive", () => {
it("should create destructive annotations with title", () => {
const result = destructive("Drop Table");
expect(result.title).toBe("Drop Table");
expect(result.readOnlyHint).toBe(false);
expect(result.destructiveHint).toBe(true);
});
});
describe("idempotent", () => {
it("should create idempotent annotations with title", () => {
const result = idempotent("Upsert Record");
expect(result.title).toBe("Upsert Record");
expect(result.readOnlyHint).toBe(false);
expect(result.destructiveHint).toBe(false);
expect(result.idempotentHint).toBe(true);
});
});
describe("admin", () => {
it("should create admin annotations with title", () => {
const result = admin("VACUUM Table");
expect(result.title).toBe("VACUUM Table");
expect(result.readOnlyHint).toBe(false);
expect(result.destructiveHint).toBe(false);
});
});
});
});
