@polderlabs/bizar 6.2.4 → 6.3.0

package/plugins/bizar/tests/tools/plan-action.test.ts

@@ -1,599 +0,0 @@
-/**
- * plan-action.test.ts
- *
- * Tests for the `planAction` core function in src/tools/plan-action.ts.
- *
- * The tool factory (`createPlanActionTool`) is a thin Zod schema + JSON
- * wrapper around `planAction`. Testing the core function directly is
- * faster and lets us drive edge cases that the framework would reject
- * before reaching our code.
- *
- * Groups (13 tests):
- *   1. get_canvas (success + missing + corrupt)
- *   2. add_element (id generation + defaults)
- *   3. update_element (patch + missing)
- *   4. delete_element (cascade connections/comments)
- *   5. add_connection (id generation + shorthand normalization)
- *   6. delete_connection (success + missing)
- *   7. add_comment (id + timestamp)
- *   8. reply_to_comment (append to thread + missing)
- *   9. set_status (update meta.json + touch plan.json)
- *  10. invalid slug
- *  11. corrupt plan.json does not throw
- *  12. concurrent writes do not lose data
- *  13. unknown action
- */
-
-import { describe, test, expect, beforeAll, afterAll } from "bun:test";
-import {
-  mkdtempSync,
-  rmSync,
-  mkdirSync,
-  writeFileSync,
-  existsSync,
-  readFileSync,
-} from "node:fs";
-import { join } from "node:path";
-import { tmpdir } from "node:os";
-
-import { planAction } from "../../src/tools/plan-action.js";
-
-// ---------------------------------------------------------------------------
-// Fixtures
-// ---------------------------------------------------------------------------
-
-class MockLogger {
-  messages: Array<{ level: string; message: string }> = [];
-  log(opts: { level: string; message: string }) { this.messages.push(opts); }
-  debug(m: string) { this.messages.push({ level: "debug", message: m }); }
-  info(m: string) { this.messages.push({ level: "info", message: m }); }
-  warn(m: string) { this.messages.push({ level: "warn", message: m }); }
-  error(m: string) { this.messages.push({ level: "error", message: m }); }
-}
-
-let worktree: string;
-const logger = new MockLogger();
-const locks = new Map<string, Promise<unknown>>();
-
-beforeAll(() => {
-  worktree = mkdtempSync(join(tmpdir(), "bizar-plan-action-test-"));
-});
-
-afterAll(() => {
-  if (worktree) rmSync(worktree, { recursive: true, force: true });
-});
-
-function planDir(slug: string) {
-  return join(worktree, "plans", slug);
-}
-
-function planJsonPath(slug: string) {
-  return join(planDir(slug), "plan.json");
-}
-
-function metaJsonPath(slug: string) {
-  return join(planDir(slug), "meta.json");
-}
-
-function seedCanvas(slug: string, canvas: unknown) {
-  mkdirSync(planDir(slug), { recursive: true });
-  writeFileSync(planJsonPath(slug), JSON.stringify(canvas), "utf-8");
-}
-
-function seedMeta(slug: string, meta: unknown) {
-  mkdirSync(planDir(slug), { recursive: true });
-  writeFileSync(metaJsonPath(slug), JSON.stringify(meta), "utf-8");
-}
-
-function readCanvas(slug: string): unknown {
-  return JSON.parse(readFileSync(planJsonPath(slug), "utf-8"));
-}
-
-// ===========================================================================
-// Group 1 — get_canvas
-// ===========================================================================
-
-describe("planAction — get_canvas", () => {
-  test("returns the full plan.json when present", async () => {
-    const slug = "get-canvas-ok";
-    seedCanvas(slug, {
-      schemaVersion: 2,
-      title: "Test",
-      elements: [{ id: "el_a", title: "A" }],
-      comments: [],
-    });
-    const r = await planAction(worktree, logger, locks, {
-      action: "get_canvas",
-      planSlug: slug,
-    });
-    expect(r.ok).toBe(true);
-    if (r.ok) {
-      expect(r.action).toBe("get_canvas");
-      const canvas = r.canvas as { title?: string; elements?: unknown[] };
-      expect(canvas.title).toBe("Test");
-      expect(canvas.elements).toHaveLength(1);
-    }
-  });
-
-  test("returns error when plan.json does not exist", async () => {
-    const r = await planAction(worktree, logger, locks, {
-      action: "get_canvas",
-      planSlug: "nonexistent-plan",
-    });
-    expect(r.ok).toBe(false);
-    if (!r.ok) {
-      expect(r.error).toMatch(/Plan not found/);
-      expect(r.planSlug).toBe("nonexistent-plan");
-    }
-  });
-});
-
-// ===========================================================================
-// Group 2 — add_element
-// ===========================================================================
-
-describe("planAction — add_element", () => {
-  test("writes a new element with a generated id when none provided", async () => {
-    const slug = "add-element-gen";
-    await planAction(worktree, logger, locks, {
-      action: "add_element",
-      planSlug: slug,
-      element: { title: "Generated" },
-    });
-    const canvas = readCanvas(slug) as { elements: Array<{ id: string; title: string }> };
-    expect(canvas.elements).toHaveLength(1);
-    expect(canvas.elements[0]!.id).toMatch(/^el_/);
-    expect(canvas.elements[0]!.title).toBe("Generated");
-  });
-
-  test("preserves a provided id when present", async () => {
-    const slug = "add-element-id";
-    await planAction(worktree, logger, locks, {
-      action: "add_element",
-      planSlug: slug,
-      element: { id: "el_custom", title: "Custom" },
-    });
-    const canvas = readCanvas(slug) as { elements: Array<{ id: string }> };
-    expect(canvas.elements[0]!.id).toBe("el_custom");
-  });
-
-  test("applies default position/size when missing", async () => {
-    const slug = "add-element-defaults";
-    await planAction(worktree, logger, locks, {
-      action: "add_element",
-      planSlug: slug,
-      element: { title: "T" },
-    });
-    const canvas = readCanvas(slug) as {
-      elements: Array<{ x: number; y: number; width: number; height: number; type: string }>;
-    };
-    expect(canvas.elements[0]!.x).toBe(80);
-    expect(canvas.elements[0]!.y).toBe(80);
-    expect(canvas.elements[0]!.width).toBe(240);
-    expect(canvas.elements[0]!.height).toBe(160);
-    expect(canvas.elements[0]!.type).toBe("text");
-  });
-
-  test("returns an error when 'element' arg is missing", async () => {
-    const r = await planAction(worktree, logger, locks, {
-      action: "add_element",
-      planSlug: "missing-element-arg",
-    });
-    expect(r.ok).toBe(false);
-    if (!r.ok) {
-      expect(r.error).toMatch(/Missing required argument: "element"/);
-    }
-  });
-});
-
-// ===========================================================================
-// Group 3 — update_element
-// ===========================================================================
-
-describe("planAction — update_element", () => {
-  test("patches an existing element", async () => {
-    const slug = "update-elem-ok";
-    seedCanvas(slug, {
-      schemaVersion: 2,
-      elements: [{ id: "el_x", title: "Old", x: 10 }],
-    });
-    const r = await planAction(worktree, logger, locks, {
-      action: "update_element",
-      planSlug: slug,
-      elementId: "el_x",
-      element: { title: "New", x: 20 },
-    });
-    expect(r.ok).toBe(true);
-    const canvas = readCanvas(slug) as { elements: Array<{ title: string; x: number; id: string }> };
-    expect(canvas.elements[0]!.title).toBe("New");
-    expect(canvas.elements[0]!.x).toBe(20);
-    expect(canvas.elements[0]!.id).toBe("el_x"); // id preserved
-  });
-
-  test("returns error when element not found", async () => {
-    const slug = "update-elem-missing";
-    seedCanvas(slug, { schemaVersion: 2, elements: [] });
-    const r = await planAction(worktree, logger, locks, {
-      action: "update_element",
-      planSlug: slug,
-      elementId: "el_nope",
-      element: { title: "X" },
-    });
-    expect(r.ok).toBe(false);
-    if (!r.ok) expect(r.error).toMatch(/Element not found/);
-  });
-
-  test("returns error when elementId is missing", async () => {
-    const r = await planAction(worktree, logger, locks, {
-      action: "update_element",
-      planSlug: "update-elem-no-id",
-      element: { title: "X" },
-    });
-    expect(r.ok).toBe(false);
-    if (!r.ok) expect(r.error).toMatch(/Missing required argument: "elementId"/);
-  });
-});
-
-// ===========================================================================
-// Group 4 — delete_element (cascade)
-// ===========================================================================
-
-describe("planAction — delete_element", () => {
-  test("removes the element and its connections + comments", async () => {
-    const slug = "delete-elem-cascade";
-    seedCanvas(slug, {
-      schemaVersion: 2,
-      elements: [
-        { id: "el_a" },
-        { id: "el_b" },
-      ],
-      connections: [
-        { id: "conn_1", fromElementId: "el_a", toElementId: "el_b" },
-        { id: "conn_2", fromElementId: "el_b", toElementId: "el_a" },
-      ],
-      comments: [
-        { id: "c_1", elementId: "el_a", text: "Pinned to a" },
-        { id: "c_2", elementId: "el_b", text: "Pinned to b" },
-        { id: "c_3", elementId: null, text: "Canvas pinned" },
-      ],
-    });
-    const r = await planAction(worktree, logger, locks, {
-      action: "delete_element",
-      planSlug: slug,
-      elementId: "el_a",
-    });
-    expect(r.ok).toBe(true);
-    if (r.ok) {
-      expect(r.removed).toBe(1);
-      expect(r.removedConnections).toBe(2);
-      expect(r.removedComments).toBe(1); // only c_1; canvas-pinned c_3 kept
-    }
-    const canvas = readCanvas(slug) as {
-      elements: unknown[];
-      connections: unknown[];
-      comments: Array<{ elementId: string | null }>;
-    };
-    expect(canvas.elements).toHaveLength(1);
-    expect(canvas.connections).toHaveLength(0);
-    expect(canvas.comments).toHaveLength(2);
-    expect(canvas.comments.find((c) => c.elementId === null)).toBeDefined();
-  });
-});
-
-// ===========================================================================
-// Group 5 — add_connection
-// ===========================================================================
-
-describe("planAction — add_connection", () => {
-  test("writes a new connection with a generated id", async () => {
-    const slug = "add-conn-gen";
-    await planAction(worktree, logger, locks, {
-      action: "add_connection",
-      planSlug: slug,
-      connection: { from: "el_a", to: "el_b" },
-    });
-    const canvas = readCanvas(slug) as { connections: Array<{ id: string; fromElementId: string; toElementId: string }> };
-    expect(canvas.connections).toHaveLength(1);
-    expect(canvas.connections[0]!.id).toMatch(/^conn_/);
-    expect(canvas.connections[0]!.fromElementId).toBe("el_a");
-    expect(canvas.connections[0]!.toElementId).toBe("el_b");
-  });
-
-  test("preserves a provided id", async () => {
-    const slug = "add-conn-id";
-    await planAction(worktree, logger, locks, {
-      action: "add_connection",
-      planSlug: slug,
-      connection: { id: "conn_custom", fromElementId: "el_a", toElementId: "el_b" },
-    });
-    const canvas = readCanvas(slug) as { connections: Array<{ id: string }> };
-    expect(canvas.connections[0]!.id).toBe("conn_custom");
-  });
-});
-
-// ===========================================================================
-// Group 6 — delete_connection
-// ===========================================================================
-
-describe("planAction — delete_connection", () => {
-  test("removes the connection", async () => {
-    const slug = "del-conn-ok";
-    seedCanvas(slug, {
-      schemaVersion: 2,
-      connections: [
-        { id: "conn_x", fromElementId: "el_a", toElementId: "el_b" },
-      ],
-    });
-    const r = await planAction(worktree, logger, locks, {
-      action: "delete_connection",
-      planSlug: slug,
-      connectionId: "conn_x",
-    });
-    expect(r.ok).toBe(true);
-    const canvas = readCanvas(slug) as { connections: unknown[] };
-    expect(canvas.connections).toHaveLength(0);
-  });
-
-  test("returns error when connection not found", async () => {
-    const slug = "del-conn-missing";
-    seedCanvas(slug, { schemaVersion: 2, connections: [] });
-    const r = await planAction(worktree, logger, locks, {
-      action: "delete_connection",
-      planSlug: slug,
-      connectionId: "conn_nope",
-    });
-    expect(r.ok).toBe(false);
-    if (!r.ok) expect(r.error).toMatch(/Connection not found/);
-  });
-});
-
-// ===========================================================================
-// Group 7 — add_comment
-// ===========================================================================
-
-describe("planAction — add_comment", () => {
-  test("writes a comment with generated id and timestamp", async () => {
-    const slug = "add-comment-gen";
-    await planAction(worktree, logger, locks, {
-      action: "add_comment",
-      planSlug: slug,
-      comment: { elementId: "el_a", text: "Hi", author: "DrB0rk" },
-    });
-    const canvas = readCanvas(slug) as { comments: Array<{ id: string; created: string; thread: unknown[] }> };
-    expect(canvas.comments).toHaveLength(1);
-    expect(canvas.comments[0]!.id).toMatch(/^c_/);
-    expect(canvas.comments[0]!.created).toMatch(/T/); // ISO-ish
-    expect(canvas.comments[0]!.thread).toEqual([]);
-  });
-});
-
-// ===========================================================================
-// Group 8 — reply_to_comment
-// ===========================================================================
-
-describe("planAction — reply_to_comment", () => {
-  test("appends a reply to an existing comment's thread", async () => {
-    const slug = "reply-comment-ok";
-    seedCanvas(slug, {
-      schemaVersion: 2,
-      comments: [{ id: "c_target", elementId: "el_a", text: "Hello" }],
-    });
-    const r = await planAction(worktree, logger, locks, {
-      action: "reply_to_comment",
-      planSlug: slug,
-      commentId: "c_target",
-      reply: { author: "ai", text: "Done" },
-    });
-    expect(r.ok).toBe(true);
-    if (r.ok) {
-      expect(r.commentId).toBe("c_target");
-      expect(r.replyId).toMatch(/^r_/);
-    }
-    const canvas = readCanvas(slug) as {
-      comments: Array<{ thread: Array<{ id: string; author: string; text: string }> }>;
-    };
-    expect(canvas.comments[0]!.thread).toHaveLength(1);
-    expect(canvas.comments[0]!.thread[0]!.author).toBe("ai");
-    expect(canvas.comments[0]!.thread[0]!.text).toBe("Done");
-  });
-
-  test("returns error when comment not found", async () => {
-    const slug = "reply-comment-missing";
-    seedCanvas(slug, { schemaVersion: 2, comments: [] });
-    const r = await planAction(worktree, logger, locks, {
-      action: "reply_to_comment",
-      planSlug: slug,
-      commentId: "c_nope",
-      reply: { text: "Hi" },
-    });
-    expect(r.ok).toBe(false);
-    if (!r.ok) expect(r.error).toMatch(/Comment not found/);
-  });
-});
-
-// ===========================================================================
-// Group 9 — set_status
-// ===========================================================================
-
-describe("planAction — set_status", () => {
-  test("updates meta.json status", async () => {
-    const slug = "set-status-ok";
-    seedMeta(slug, { status: "draft" });
-    const r = await planAction(worktree, logger, locks, {
-      action: "set_status",
-      planSlug: slug,
-      status: "approved",
-    });
-    expect(r.ok).toBe(true);
-    if (r.ok) expect(r.status).toBe("approved");
-    const meta = JSON.parse(readFileSync(metaJsonPath(slug), "utf-8"));
-    expect(meta.status).toBe("approved");
-    expect(meta.lastEdited).toMatch(/T/);
-  });
-
-  test("creates meta.json if missing", async () => {
-    const slug = "set-status-create";
-    mkdirSync(planDir(slug), { recursive: true });
-    const r = await planAction(worktree, logger, locks, {
-      action: "set_status",
-      planSlug: slug,
-      status: "rejected",
-    });
-    expect(r.ok).toBe(true);
-    expect(existsSync(metaJsonPath(slug))).toBe(true);
-  });
-
-  test("rejects invalid status value", async () => {
-    const r = await planAction(worktree, logger, locks, {
-      action: "set_status",
-      planSlug: "set-status-bad",
-      status: "wat" as never,
-    });
-    expect(r.ok).toBe(false);
-    if (!r.ok) expect(r.error).toMatch(/Invalid status/);
-  });
-});
-
-// ===========================================================================
-// Group 10 — invalid slug
-// ===========================================================================
-
-describe("planAction — invalid slug", () => {
-  test("rejects UPPERCASE slug", async () => {
-    const r = await planAction(worktree, logger, locks, {
-      action: "get_canvas",
-      planSlug: "UPPERCASE",
-    });
-    expect(r.ok).toBe(false);
-    if (!r.ok) expect(r.error).toMatch(/Invalid planSlug/);
-  });
-
-  test("rejects empty slug", async () => {
-    const r = await planAction(worktree, logger, locks, {
-      action: "get_canvas",
-      planSlug: "",
-    });
-    expect(r.ok).toBe(false);
-  });
-
-  test("rejects path-traversal slug", async () => {
-    const r = await planAction(worktree, logger, locks, {
-      action: "get_canvas",
-      planSlug: "../etc",
-    });
-    expect(r.ok).toBe(false);
-    if (!r.ok) expect(r.error).toMatch(/Invalid planSlug/);
-  });
-});
-
-// ===========================================================================
-// Group 11 — corrupt plan.json does not throw
-// ===========================================================================
-
-describe("planAction — corrupt plan.json", () => {
-  test("does not throw and returns an error result on invalid JSON", async () => {
-    const slug = "corrupt-plan";
-    mkdirSync(planDir(slug), { recursive: true });
-    writeFileSync(planJsonPath(slug), "{ not valid json :: !!", "utf-8");
-    const r = await planAction(worktree, logger, locks, {
-      action: "add_element",
-      planSlug: slug,
-      element: { title: "Should fail" },
-    });
-    expect(r.ok).toBe(false);
-    if (!r.ok) {
-      expect(r.error).toMatch(/corrupt/i);
-    }
-    // Original corrupt file is preserved
-    expect(readFileSync(planJsonPath(slug), "utf-8")).toBe("{ not valid json :: !!");
-  });
-
-  test("does not throw when plan.json is null", async () => {
-    const slug = "null-plan";
-    mkdirSync(planDir(slug), { recursive: true });
-    writeFileSync(planJsonPath(slug), "null", "utf-8");
-    const r = await planAction(worktree, logger, locks, {
-      action: "add_element",
-      planSlug: slug,
-      element: { title: "X" },
-    });
-    expect(r.ok).toBe(false);
-  });
-});
-
-// ===========================================================================
-// Group 12 — concurrent writes do not lose data
-// ===========================================================================
-
-describe("planAction — concurrency", () => {
-  test("parallel add_element calls all land on disk", async () => {
-    const slug = "concurrent-add";
-    const N = 30;
-    const results = await Promise.all(
-      Array.from({ length: N }, (_, i) =>
-        planAction(worktree, logger, locks, {
-          action: "add_element",
-          planSlug: slug,
-          element: { title: `e-${i}` },
-        }),
-      ),
-    );
-    const oks = results.filter((r) => r.ok);
-    expect(oks).toHaveLength(N);
-    const canvas = readCanvas(slug) as { elements: unknown[] };
-    expect(canvas.elements).toHaveLength(N);
-    const ids = new Set(
-      ((canvas.elements as Array<{ id: string }>)).map((e) => e.id),
-    );
-    expect(ids.size).toBe(N); // all ids unique
-  });
-
-  test("parallel delete + add does not corrupt the file", async () => {
-    const slug = "concurrent-mixed";
-    // Seed 10 elements
-    seedCanvas(slug, {
-      schemaVersion: 2,
-      elements: Array.from({ length: 10 }, (_, i) => ({ id: `el_seed_${i}` })),
-    });
-    // Fire 10 deletes + 10 adds in parallel
-    const ops = [
-      ...Array.from({ length: 10 }, (_, i) =>
-        planAction(worktree, logger, locks, {
-          action: "delete_element",
-          planSlug: slug,
-          elementId: `el_seed_${i}`,
-        }),
-      ),
-      ...Array.from({ length: 10 }, (_, i) =>
-        planAction(worktree, logger, locks, {
-          action: "add_element",
-          planSlug: slug,
-          element: { title: `new-${i}` },
-        }),
-      ),
-    ];
-    const results = await Promise.all(ops);
-    expect(results.every((r) => r.ok)).toBe(true);
-    // File is still valid JSON
-    const canvas = JSON.parse(readFileSync(planJsonPath(slug), "utf-8")) as {
-      elements: unknown[];
-    };
-    expect(Array.isArray(canvas.elements)).toBe(true);
-    expect(canvas.elements).toHaveLength(10); // 10 deletes + 10 adds = 10
-  });
-});
-
-// ===========================================================================
-// Group 13 — unknown action
-// ===========================================================================
-
-describe("planAction — unknown action", () => {
-  test("returns error for unknown action", async () => {
-    const r = await planAction(worktree, logger, locks, {
-      action: "nuke_plan",
-      planSlug: "unknown-action",
-    });
-    expect(r.ok).toBe(false);
-    if (!r.ok) expect(r.error).toMatch(/Unknown action/);
-  });
-});