tower-studio 0.2.7 → 0.2.9

package/src/mcp/tools/__tests__/label-tools.test.ts

@@ -1,140 +0,0 @@
-import { vi, describe, it, expect, beforeEach } from "vitest";
-
-// Mock db before any imports
-vi.mock("../../db", () => ({
-  db: {
-    label: {
-      findMany: vi.fn(),
-      create: vi.fn(),
-      delete: vi.fn(),
-    },
-    taskLabel: {
-      deleteMany: vi.fn(),
-      createMany: vi.fn(),
-    },
-    $transaction: vi.fn(),
-  },
-}));
-
-import { db } from "../../db";
-import { labelTools } from "../label-tools";
-
-const mockDb = db as {
-  label: {
-    findMany: ReturnType<typeof vi.fn>;
-    create: ReturnType<typeof vi.fn>;
-    delete: ReturnType<typeof vi.fn>;
-  };
-  taskLabel: {
-    deleteMany: ReturnType<typeof vi.fn>;
-    createMany: ReturnType<typeof vi.fn>;
-  };
-  $transaction: ReturnType<typeof vi.fn>;
-};
-
-describe("label-tools", () => {
-  beforeEach(() => {
-    vi.clearAllMocks();
-  });
-
-  describe("list_labels", () => {
-    it("calls db.label.findMany with OR condition for builtin and workspace-specific labels", async () => {
-      const mockLabels = [
-        { id: "l1", name: "Bug", color: "#ff0000", isBuiltin: true },
-        { id: "l2", name: "Custom", color: "#00ff00", isBuiltin: false, workspaceId: "ws1" },
-      ];
-      mockDb.label.findMany.mockResolvedValue(mockLabels);
-
-      const result = await labelTools.list_labels.handler({ workspaceId: "ws1" });
-
-      expect(mockDb.label.findMany).toHaveBeenCalledOnce();
-      const callArgs = mockDb.label.findMany.mock.calls[0][0];
-      expect(callArgs.where.OR).toEqual(
-        expect.arrayContaining([{ isBuiltin: true }, { workspaceId: "ws1" }])
-      );
-      expect(result).toEqual(mockLabels);
-    });
-  });
-
-  describe("create_label", () => {
-    it("calls db.label.create with name, color, and workspaceId", async () => {
-      const mockLabel = { id: "l1", name: "Feature", color: "#0000ff", workspaceId: "ws1" };
-      mockDb.label.create.mockResolvedValue(mockLabel);
-
-      const result = await labelTools.create_label.handler({
-        workspaceId: "ws1",
-        name: "Feature",
-        color: "#0000ff",
-      });
-
-      expect(mockDb.label.create).toHaveBeenCalledWith({
-        data: { name: "Feature", color: "#0000ff", workspaceId: "ws1" },
-      });
-      expect(result).toEqual(mockLabel);
-    });
-  });
-
-  describe("delete_label", () => {
-    it("calls db.label.delete and returns { deleted: true }", async () => {
-      mockDb.label.delete.mockResolvedValue({ id: "l1" });
-
-      const result = await labelTools.delete_label.handler({ labelId: "l1" });
-
-      expect(mockDb.label.delete).toHaveBeenCalledWith({ where: { id: "l1" } });
-      expect(result).toEqual({ deleted: true });
-    });
-  });
-
-  describe("set_task_labels", () => {
-    it("calls tx.taskLabel.deleteMany then tx.taskLabel.createMany in transaction for non-empty labelIds", async () => {
-      const mockTx = {
-        taskLabel: {
-          deleteMany: vi.fn().mockResolvedValue({}),
-          createMany: vi.fn().mockResolvedValue({}),
-        },
-      };
-      mockDb.$transaction.mockImplementation(
-        async (fn: (tx: typeof mockTx) => Promise<void>) => fn(mockTx)
-      );
-
-      const result = await labelTools.set_task_labels.handler({
-        taskId: "task1",
-        labelIds: ["l1", "l2"],
-      });
-
-      expect(mockTx.taskLabel.deleteMany).toHaveBeenCalledWith({
-        where: { taskId: "task1" },
-      });
-      expect(mockTx.taskLabel.createMany).toHaveBeenCalledWith({
-        data: [
-          { taskId: "task1", labelId: "l1" },
-          { taskId: "task1", labelId: "l2" },
-        ],
-      });
-      expect(result).toEqual({ taskId: "task1", labelIds: ["l1", "l2"] });
-    });
-
-    it("calls tx.taskLabel.deleteMany but NOT createMany for empty labelIds", async () => {
-      const mockTx = {
-        taskLabel: {
-          deleteMany: vi.fn().mockResolvedValue({}),
-          createMany: vi.fn().mockResolvedValue({}),
-        },
-      };
-      mockDb.$transaction.mockImplementation(
-        async (fn: (tx: typeof mockTx) => Promise<void>) => fn(mockTx)
-      );
-
-      const result = await labelTools.set_task_labels.handler({
-        taskId: "task1",
-        labelIds: [],
-      });
-
-      expect(mockTx.taskLabel.deleteMany).toHaveBeenCalledWith({
-        where: { taskId: "task1" },
-      });
-      expect(mockTx.taskLabel.createMany).not.toHaveBeenCalled();
-      expect(result).toEqual({ taskId: "task1", labelIds: [] });
-    });
-  });
-});

package/src/mcp/tools/__tests__/project-tools.test.ts

@@ -1,182 +0,0 @@
-// @vitest-environment node
-import { vi, describe, it, expect, beforeEach } from "vitest";
-
-// Mock db BEFORE imports
-vi.mock("../../db", () => ({
-  db: {
-    project: {
-      findMany: vi.fn(),
-      create: vi.fn(),
-      update: vi.fn(),
-      delete: vi.fn(),
-    },
-  },
-}));
-
-import { db } from "../../db";
-import { projectTools } from "../project-tools";
-
-const mockDb = db as {
-  project: {
-    findMany: ReturnType<typeof vi.fn>;
-    create: ReturnType<typeof vi.fn>;
-    update: ReturnType<typeof vi.fn>;
-    delete: ReturnType<typeof vi.fn>;
-  };
-};
-
-describe("project-tools", () => {
-  beforeEach(() => {
-    vi.clearAllMocks();
-  });
-
-  // ─── list_projects ──────────────────────────────────────────────────────
-
-  describe("list_projects", () => {
-    it("returns projects with taskCount and repositoryCount derived from array lengths", async () => {
-      const mockProjects = [
-        {
-          id: "proj1",
-          name: "Project Alpha",
-          workspaceId: "ws1",
-          tasks: [{ id: "t1" }, { id: "t2" }, { id: "t3" }],
-          repositories: [{ id: "r1" }],
-          updatedAt: new Date("2026-04-20"),
-        },
-        {
-          id: "proj2",
-          name: "Project Beta",
-          workspaceId: "ws1",
-          tasks: [],
-          repositories: [],
-          updatedAt: new Date("2026-04-19"),
-        },
-      ];
-      mockDb.project.findMany.mockResolvedValue(mockProjects);
-
-      const result = await projectTools.list_projects.handler({ workspaceId: "ws1" });
-
-      expect(mockDb.project.findMany).toHaveBeenCalledWith(
-        expect.objectContaining({
-          where: { workspaceId: "ws1" },
-        })
-      );
-      expect(result).toHaveLength(2);
-      // tasks and repositories arrays should not appear in output
-      expect(result[0]).not.toHaveProperty("tasks");
-      expect(result[0]).not.toHaveProperty("repositories");
-      // counts derived from array lengths
-      expect(result[0].taskCount).toBe(3);
-      expect(result[0].repositoryCount).toBe(1);
-      expect(result[1].taskCount).toBe(0);
-      expect(result[1].repositoryCount).toBe(0);
-    });
-  });
-
-  // ─── create_project ──────────────────────────────────────────────────────
-
-  describe("create_project", () => {
-    it("sets type to GIT when gitUrl is provided", async () => {
-      const createdProject = {
-        id: "proj1",
-        name: "Git Project",
-        type: "GIT",
-        gitUrl: "https://github.com/org/repo",
-        workspaceId: "ws1",
-      };
-      mockDb.project.create.mockResolvedValue(createdProject);
-
-      const result = await projectTools.create_project.handler({
-        workspaceId: "ws1",
-        name: "Git Project",
-        gitUrl: "https://github.com/org/repo",
-      });
-
-      expect(mockDb.project.create).toHaveBeenCalledWith({
-        data: expect.objectContaining({
-          type: "GIT",
-          gitUrl: "https://github.com/org/repo",
-        }),
-      });
-      expect(result.type).toBe("GIT");
-    });
-
-    it("sets type to NORMAL when gitUrl is absent", async () => {
-      const createdProject = {
-        id: "proj2",
-        name: "Normal Project",
-        type: "NORMAL",
-        workspaceId: "ws1",
-      };
-      mockDb.project.create.mockResolvedValue(createdProject);
-
-      const result = await projectTools.create_project.handler({
-        workspaceId: "ws1",
-        name: "Normal Project",
-      });
-
-      expect(mockDb.project.create).toHaveBeenCalledWith({
-        data: expect.objectContaining({
-          type: "NORMAL",
-        }),
-      });
-      // Ensure type is NORMAL not GIT — caller cannot override this
-      expect(result.type).toBe("NORMAL");
-    });
-
-    it("passes alias to db.project.create when provided", async () => {
-      const createdProject = { id: "proj3", name: "Aliased Project", alias: "myproj", type: "NORMAL", workspaceId: "ws1" };
-      mockDb.project.create.mockResolvedValue(createdProject);
-
-      await projectTools.create_project.handler({
-        workspaceId: "ws1",
-        name: "Aliased Project",
-        alias: "myproj",
-      });
-
-      expect(mockDb.project.create).toHaveBeenCalledWith({
-        data: expect.objectContaining({
-          alias: "myproj",
-          type: "NORMAL",
-        }),
-      });
-    });
-  });
-
-  // ─── update_project ──────────────────────────────────────────────────────
-
-  describe("update_project", () => {
-    it("excludes projectId from update data and spreads remaining fields", async () => {
-      const updatedProject = { id: "proj1", name: "New Name", localPath: "/new/path" };
-      mockDb.project.update.mockResolvedValue(updatedProject);
-
-      const result = await projectTools.update_project.handler({
-        projectId: "proj1",
-        name: "New Name",
-        localPath: "/new/path",
-      });
-
-      expect(mockDb.project.update).toHaveBeenCalledWith({
-        where: { id: "proj1" },
-        data: { name: "New Name", localPath: "/new/path" },
-      });
-      // projectId must not appear in the data
-      const updateCall = mockDb.project.update.mock.calls[0][0];
-      expect(updateCall.data).not.toHaveProperty("projectId");
-      expect(result).toEqual(updatedProject);
-    });
-  });
-
-  // ─── delete_project ──────────────────────────────────────────────────────
-
-  describe("delete_project", () => {
-    it("deletes project and returns confirmation with projectId", async () => {
-      mockDb.project.delete.mockResolvedValue({ id: "proj1" });
-
-      const result = await projectTools.delete_project.handler({ projectId: "proj1" });
-
-      expect(mockDb.project.delete).toHaveBeenCalledWith({ where: { id: "proj1" } });
-      expect(result).toEqual({ deleted: true, projectId: "proj1" });
-    });
-  });
-});

package/src/mcp/tools/__tests__/report-tools.test.ts

@@ -1,212 +0,0 @@
-import { describe, it, expect, vi, beforeEach } from "vitest";
-
-vi.mock("../../db", () => ({
-  db: {
-    task: { findMany: vi.fn() },
-    projectNote: { findMany: vi.fn() },
-  },
-}));
-
-import { db } from "../../db";
-import { reportTools } from "../report-tools";
-
-const mockDb = db as {
-  task: { findMany: ReturnType<typeof vi.fn> };
-  projectNote: { findMany: ReturnType<typeof vi.fn> };
-};
-
-/** Build a minimal task object matching the daily_summary include clause */
-function makeSummaryTask(overrides: {
-  id?: string;
-  title?: string;
-  status?: string;
-  priority?: string;
-  updatedAt?: Date;
-  project?: object;
-  messages?: { content: string; createdAt: Date }[];
-  executions?: { startedAt?: Date; status?: string; summary?: string | null }[];
-}) {
-  const now = new Date("2026-01-15T10:00:00Z");
-  const ws = { id: "ws1", name: "Workspace 1" };
-  const proj = { id: "p1", name: "Project 1", alias: null, workspace: ws };
-
-  return {
-    id: overrides.id ?? "task1",
-    title: overrides.title ?? "Test task",
-    status: overrides.status ?? "TODO",
-    priority: overrides.priority ?? "MEDIUM",
-    updatedAt: overrides.updatedAt ?? now,
-    createdAt: now,
-    project: overrides.project ?? proj,
-    messages: overrides.messages ?? [],
-    executions: overrides.executions ?? [],
-    labels: [],
-  };
-}
-
-/** Build a minimal task object matching the daily_todo include clause */
-function makeTodoTask(overrides: {
-  id?: string;
-  title?: string;
-  status?: string;
-  priority?: string;
-  updatedAt?: Date;
-  project?: object;
-  labels?: { label: { name: string } }[];
-  executions?: { sessionId?: string | null }[];
-}) {
-  const now = new Date("2026-01-15T10:00:00Z");
-  const ws = { id: "ws1", name: "Workspace 1" };
-  const proj = { id: "p1", name: "Project 1", alias: null, workspace: ws };
-
-  return {
-    id: overrides.id ?? "task1",
-    title: overrides.title ?? "Test task",
-    status: overrides.status ?? "TODO",
-    priority: overrides.priority ?? "MEDIUM",
-    updatedAt: overrides.updatedAt ?? now,
-    createdAt: now,
-    project: overrides.project ?? proj,
-    labels: overrides.labels ?? [],
-    executions: overrides.executions ?? [],
-  };
-}
-
-describe("report-tools (MCP)", () => {
-  beforeEach(() => {
-    vi.clearAllMocks();
-    mockDb.projectNote.findMany.mockResolvedValue([]);
-  });
-
-  // ── daily_summary ─────────────────────────────────────────────────────────
-
-  describe("daily_summary", () => {
-    it("with date param: uses date range [00:00, next day 00:00) and groups DONE/IN_PROGRESS tasks", async () => {
-      const doneTask = makeSummaryTask({ id: "t1", status: "DONE", title: "Done task" });
-      const inProgressTask = makeSummaryTask({ id: "t2", status: "IN_PROGRESS", title: "WIP task" });
-      mockDb.task.findMany.mockResolvedValue([doneTask, inProgressTask]);
-
-      const result = await reportTools.daily_summary.handler({ date: "2026-01-15" });
-
-      // Verify date range filter passed to findMany
-      const callArgs = mockDb.task.findMany.mock.calls[0][0];
-      const orClauses = callArgs.where.OR;
-      const updatedAtClause = orClauses.find((c: Record<string, unknown>) => "updatedAt" in c);
-      expect(updatedAtClause).toBeDefined();
-      const gte = updatedAtClause.updatedAt.gte as Date;
-      const lt = updatedAtClause.updatedAt.lt as Date;
-      expect(gte.getHours()).toBe(0);
-      expect(gte.getMinutes()).toBe(0);
-      const diffMs = lt.getTime() - gte.getTime();
-      expect(diffMs).toBe(24 * 60 * 60 * 1000);
-
-      // Verify grouping
-      expect(result.workspaces).toHaveLength(1);
-      const ws = result.workspaces[0];
-      expect(ws.projects).toHaveLength(1);
-      const proj = ws.projects[0];
-      expect(proj.completed).toHaveLength(1);
-      expect(proj.completed[0].id).toBe("t1");
-      expect(proj.inProgress).toHaveLength(1);
-      expect(proj.inProgress[0].id).toBe("t2");
-      expect(result.stats.totalCompleted).toBe(1);
-      expect(result.stats.totalInProgress).toBe(1);
-    });
-
-    it("without date param: defaults to today (passes Date objects to findMany)", async () => {
-      mockDb.task.findMany.mockResolvedValue([]);
-
-      const before = new Date();
-      before.setHours(0, 0, 0, 0);
-      await reportTools.daily_summary.handler({});
-      const after = new Date();
-      after.setHours(0, 0, 0, 0);
-
-      const callArgs = mockDb.task.findMany.mock.calls[0][0];
-      const orClauses = callArgs.where.OR;
-      const updatedAtClause = orClauses.find((c: Record<string, unknown>) => "updatedAt" in c);
-      const gte = updatedAtClause.updatedAt.gte as Date;
-      // gte should be today's midnight (between before and after)
-      expect(gte.getTime()).toBeGreaterThanOrEqual(before.getTime());
-      expect(gte.getTime()).toBeLessThanOrEqual(after.getTime());
-    });
-
-    it("CANCELLED tasks are excluded from completed and inProgress", async () => {
-      const cancelledTask = makeSummaryTask({ id: "t1", status: "CANCELLED" });
-      mockDb.task.findMany.mockResolvedValue([cancelledTask]);
-
-      const result = await reportTools.daily_summary.handler({});
-
-      expect(result.stats.totalCompleted).toBe(0);
-      expect(result.stats.totalInProgress).toBe(0);
-    });
-
-    it("empty result: returns empty workspaces array with zero stats", async () => {
-      mockDb.task.findMany.mockResolvedValue([]);
-
-      const result = await reportTools.daily_summary.handler({});
-
-      expect(result.workspaces).toHaveLength(0);
-      expect(result.stats.totalCompleted).toBe(0);
-      expect(result.stats.totalInProgress).toBe(0);
-      expect(result.stats.totalActive).toBe(0);
-    });
-  });
-
-  // ── daily_todo ────────────────────────────────────────────────────────────
-
-  describe("daily_todo", () => {
-    it("default call: status filter is [TODO, IN_PROGRESS, IN_REVIEW]", async () => {
-      mockDb.task.findMany.mockResolvedValue([]);
-
-      await reportTools.daily_todo.handler({});
-
-      const where = mockDb.task.findMany.mock.calls[0][0].where;
-      expect(where.status).toEqual({ in: ["TODO", "IN_PROGRESS", "IN_REVIEW"] });
-    });
-
-    it("with workspaceId: where clause includes project.workspaceId", async () => {
-      mockDb.task.findMany.mockResolvedValue([]);
-
-      await reportTools.daily_todo.handler({ workspaceId: "ws1" });
-
-      const where = mockDb.task.findMany.mock.calls[0][0].where;
-      expect(where.project).toEqual({ workspaceId: "ws1" });
-    });
-
-    it("with projectId: where clause includes projectId", async () => {
-      mockDb.task.findMany.mockResolvedValue([]);
-
-      await reportTools.daily_todo.handler({ projectId: "p1" });
-
-      const where = mockDb.task.findMany.mock.calls[0][0].where;
-      expect(where.projectId).toBe("p1");
-    });
-
-    it("with priority filter: where clause includes priority.in", async () => {
-      mockDb.task.findMany.mockResolvedValue([]);
-
-      await reportTools.daily_todo.handler({ priority: ["CRITICAL", "HIGH"] });
-
-      const where = mockDb.task.findMany.mock.calls[0][0].where;
-      expect(where.priority).toEqual({ in: ["CRITICAL", "HIGH"] });
-    });
-
-    it("stats computation: byStatus and byPriority match returned tasks", async () => {
-      const t1 = makeTodoTask({ id: "t1", status: "TODO", priority: "CRITICAL" });
-      const t2 = makeTodoTask({ id: "t2", status: "IN_PROGRESS", priority: "HIGH" });
-      const t3 = makeTodoTask({ id: "t3", status: "IN_REVIEW", priority: "CRITICAL" });
-      mockDb.task.findMany.mockResolvedValue([t1, t2, t3]);
-
-      const result = await reportTools.daily_todo.handler({});
-
-      expect(result.stats.total).toBe(3);
-      expect(result.stats.byStatus.TODO).toBe(1);
-      expect(result.stats.byStatus.IN_PROGRESS).toBe(1);
-      expect(result.stats.byStatus.IN_REVIEW).toBe(1);
-      expect(result.stats.byPriority.CRITICAL).toBe(2);
-      expect(result.stats.byPriority.HIGH).toBe(1);
-      expect(result.stats.byPriority.MEDIUM).toBe(0);
-    });
-  });
-});