kandev/apps/web/lib/state/slices/gitlab/gitlab-slice.test.ts

import { describe, it, expect } from "vitest";
import { create } from "zustand";
import { immer } from "zustand/middleware/immer";
import { createGitLabSlice } from "./gitlab-slice";
import type { GitLabSlice } from "./types";
import type { TaskMR, TaskMRAutomationOptions } from "@/lib/types/gitlab";

function makeMR(overrides: Partial<TaskMR> = {}): TaskMR {
  return {
    id: "id",
    task_id: "task-1",
    host: "https://gitlab.com",
    project_path: "acme/api",
    mr_iid: 1,
    mr_url: "https://gitlab.com/acme/api/-/merge_requests/1",
    mr_title: "Test MR",
    head_branch: "feat",
    base_branch: "main",
    author_username: "alice",
    state: "open",
    approval_state: "",
    pipeline_state: "",
    merge_status: "",
    draft: false,
    approval_count: 0,
    required_approvals: 0,
    pipeline_jobs_total: 0,
    pipeline_jobs_pass: 0,
    reviewer_count: 0,
    unapproved_reviewers: 0,
    unresolved_discussions: 0,
    created_at: "",
    updated_at: "",
    ...overrides,
  };
}

function makeOptions(overrides: Partial<TaskMRAutomationOptions> = {}): TaskMRAutomationOptions {
  return {
    task_id: "task-a",
    auto_fix_enabled: false,
    auto_merge_enabled: false,
    auto_fix_max_rounds: 10,
    effective_auto_fix_prompt: "",
    using_default_prompt: true,
    prompt_on_review_requested: false,
    prompt_on_merged: false,
    prompt_on_closed: false,
    review_reviewer_username: "",
    updated_at: "2026-01-01T00:00:00Z",
    mr_states: [],
    mr_options: [],
    ...overrides,
  };
}

function makeStore() {
  return create<GitLabSlice>()(immer((...a) => createGitLabSlice(...a)));
}

function workspaceMRs(store: ReturnType<typeof makeStore>, workspaceId = "ws-1") {
  return store.getState().taskMRs.byWorkspaceId[workspaceId] ?? {};
}

describe("setTaskMRs", () => {
  it("replaces byTaskId wholesale", () => {
    const store = makeStore();
    const mrA = makeMR({ id: "a" });
    store.getState().setTaskMRs("ws-1", { "task-1": [mrA] });
    expect(workspaceMRs(store)["task-1"]).toEqual([mrA]);

    store.getState().setTaskMRs("ws-2", { "task-2": [makeMR({ id: "b" })] });
    expect(workspaceMRs(store)).toHaveProperty("task-1");
    expect(workspaceMRs(store, "ws-2")["task-2"]).toHaveLength(1);
  });
});

describe("setTaskMR", () => {
  it("appends an MR when the task has no rows yet", () => {
    const store = makeStore();
    const mr = makeMR({ repository_id: "repo-a" });

    store.getState().setTaskMR("ws-1", "task-1", mr);

    expect(workspaceMRs(store)["task-1"]).toEqual([mr]);
  });

  it("keeps a delayed workspace A link write isolated from workspace B", () => {
    const store = makeStore();
    const workspaceBMR = makeMR({ id: "b", task_id: "task-b" });
    const delayedWorkspaceAMR = makeMR({ id: "a", task_id: "task-a" });
    store.getState().setTaskMRs("ws-b", { "task-b": [workspaceBMR] });

    store.getState().setTaskMR("ws-a", "task-a", delayedWorkspaceAMR);

    expect(store.getState().taskMRs.byWorkspaceId["ws-b"]?.["task-b"]).toEqual([workspaceBMR]);
    expect(store.getState().taskMRs.byWorkspaceId["ws-a"]?.["task-a"]).toEqual([
      delayedWorkspaceAMR,
    ]);
  });

  it("upserts the same (repo, project, iid) key in place", () => {
    const store = makeStore();
    const original = makeMR({ id: "a", repository_id: "repo-a", mr_iid: 5 });
    const updated = makeMR({
      id: "a",
      repository_id: "repo-a",
      host: "https://gitlab.com/",
      mr_iid: 5,
      mr_title: "renamed",
    });

    store.getState().setTaskMR("ws-1", "task-1", original);
    store.getState().setTaskMR("ws-1", "task-1", updated);

    const list = workspaceMRs(store)["task-1"];
    expect(list).toHaveLength(1);
    expect(list[0]!.mr_title).toBe("renamed");
  });

  it("keeps multi-repo MRs distinct on (repository_id, project_path, mr_iid)", () => {
    const store = makeStore();
    // Same project + iid, different repositories — must coexist.
    const repoA = makeMR({ id: "a", repository_id: "repo-a", mr_iid: 1 });
    const repoB = makeMR({ id: "b", repository_id: "repo-b", mr_iid: 1 });
    // Same repo, different project — must also coexist.
    const repoAOtherProject = makeMR({
      id: "c",
      repository_id: "repo-a",
      project_path: "acme/web",
      mr_iid: 1,
    });
    // Same repo + project, different iid — must coexist (multiple open MRs).
    const repoASecondMR = makeMR({ id: "d", repository_id: "repo-a", mr_iid: 99 });

    store.getState().setTaskMR("ws-1", "task-1", repoA);
    store.getState().setTaskMR("ws-1", "task-1", repoB);
    store.getState().setTaskMR("ws-1", "task-1", repoAOtherProject);
    store.getState().setTaskMR("ws-1", "task-1", repoASecondMR);

    const list = workspaceMRs(store)["task-1"];
    expect(list).toHaveLength(4);
    expect(list.map((m) => m.id).sort()).toEqual(["a", "b", "c", "d"]);
  });

  it("keeps identical repository/project/IID associations distinct across hosts", () => {
    const store = makeStore();
    store
      .getState()
      .setTaskMR(
        "ws-1",
        "task-1",
        makeMR({ id: "public", repository_id: "repo-a", host: "https://gitlab.com" }),
      );
    store
      .getState()
      .setTaskMR(
        "ws-1",
        "task-1",
        makeMR({ id: "private", repository_id: "repo-a", host: "https://gitlab.internal" }),
      );

    expect(workspaceMRs(store)["task-1"]?.map((mr) => mr.id)).toEqual(["public", "private"]);
  });

  it("treats missing repository_id as the empty key (single-repo tasks)", () => {
    // A multi-repo MR with repository_id and a single-repo MR with the same
    // (project_path, mr_iid) but no repository_id must still coexist — the
    // empty repo key is its own slot.
    const store = makeStore();
    const noRepo = makeMR({ id: "x", mr_iid: 1 });
    const withRepo = makeMR({ id: "y", repository_id: "repo-a", mr_iid: 1 });

    store.getState().setTaskMR("ws-1", "task-1", noRepo);
    store.getState().setTaskMR("ws-1", "task-1", withRepo);

    const list = workspaceMRs(store)["task-1"];
    expect(list).toHaveLength(2);
    expect(list.map((m) => m.id).sort()).toEqual(["x", "y"]);
  });
});

describe("resetTaskMRs", () => {
  it("clears byTaskId so a workspace switch doesn't leak previous MRs", () => {
    const store = makeStore();
    store.getState().setTaskMR("ws-1", "task-1", makeMR({ id: "a" }));
    expect(workspaceMRs(store)["task-1"]).toHaveLength(1);

    store.getState().resetTaskMRs();

    expect(store.getState().taskMRs.byWorkspaceId).toEqual({});
  });
});

describe("removeTaskMR", () => {
  it("removes only the selected association", () => {
    const store = makeStore();
    store.getState().setTaskMRs("ws-1", {
      "task-1": [makeMR({ id: "remove" }), makeMR({ id: "keep", mr_iid: 2 })],
      "task-2": [makeMR({ id: "other", task_id: "task-2" })],
    });

    store.getState().removeTaskMR("ws-1", "remove");

    expect(workspaceMRs(store)["task-1"]?.map((mr) => mr.id)).toEqual(["keep"]);
    expect(workspaceMRs(store)["task-2"]?.map((mr) => mr.id)).toEqual(["other"]);
  });
});

describe("review watches", () => {
  it("set + add + update + remove round-trip", () => {
    const store = makeStore();
    const watch = {
      id: "w-1",
      workspace_id: "ws-1",
      workflow_id: "",
      workflow_step_id: "",
      projects: [{ path: "group/proj" }],
      agent_profile_id: "",
      executor_profile_id: "",
      prompt: "",
      review_scope: "user",
      custom_query: "",
      enabled: true,
      poll_interval_seconds: 300,
      cleanup_policy: "auto",
      created_at: "",
      updated_at: "",
    };
    store.getState().setGitLabReviewWatches([watch]);
    expect(store.getState().gitlabReviewWatches.items).toHaveLength(1);
    expect(store.getState().gitlabReviewWatches.loaded).toBe(true);

    const watch2 = { ...watch, id: "w-2" };
    store.getState().addGitLabReviewWatch(watch2);
    expect(store.getState().gitlabReviewWatches.items).toHaveLength(2);

    store.getState().updateGitLabReviewWatchInStore({ ...watch, prompt: "updated" });
    expect(store.getState().gitlabReviewWatches.items.find((w) => w.id === "w-1")?.prompt).toBe(
      "updated",
    );

    store.getState().removeGitLabReviewWatch("w-1");
    expect(store.getState().gitlabReviewWatches.items).toHaveLength(1);
    expect(store.getState().gitlabReviewWatches.items[0].id).toBe("w-2");
  });
});

describe("issue watches", () => {
  it("set + add + remove round-trip", () => {
    const store = makeStore();
    const watch = {
      id: "iw-1",
      workspace_id: "ws-1",
      workflow_id: "",
      workflow_step_id: "",
      projects: [],
      agent_profile_id: "",
      executor_profile_id: "",
      prompt: "",
      labels: ["bug"],
      custom_query: "",
      enabled: true,
      poll_interval_seconds: 300,
      cleanup_policy: "auto",
      created_at: "",
      updated_at: "",
    };
    store.getState().setGitLabIssueWatches([watch]);
    expect(store.getState().gitlabIssueWatches.items).toHaveLength(1);

    store.getState().removeGitLabIssueWatch("iw-1");
    expect(store.getState().gitlabIssueWatches.items).toHaveLength(0);
  });
});

describe("action presets + stats", () => {
  it("upserts presets by workspace and stores stats with loadedAt timestamp", () => {
    const store = makeStore();
    const presets = {
      workspace_id: "ws-1",
      mr: [{ id: "x", label: "x", hint: "", icon: "", prompt_template: "" }],
      issue: [],
    };
    store.getState().setGitLabActionPresets("ws-1", presets);
    expect(store.getState().gitlabActionPresets.byWorkspaceId["ws-1"]).toEqual(presets);

    store.getState().setGitLabStats({
      open_mrs: 5,
      mrs_awaiting_my_review: 2,
      open_issues_assigned_me: 3,
    });
    expect(store.getState().gitlabStats.data?.open_mrs).toBe(5);
    expect(store.getState().gitlabStats.loadedAt).not.toBeNull();
  });
});

describe("task MR automation options", () => {
  it("keys options, loading, saving, and errors independently per task", () => {
    const store = makeStore();
    const optionsA = makeOptions({ prompt_on_review_requested: true });

    store.getState().setTaskMRAutomationOptions("task-a", optionsA);
    store.getState().setTaskMRAutomationLoading("task-b", true);
    store.getState().setTaskMRAutomationSaving("task-a", true);
    store.getState().setTaskMRAutomationError("task-b", "boom");

    expect(store.getState().taskMRAutomation.byTaskId["task-a"]).toEqual(optionsA);
    expect(store.getState().taskMRAutomation.byTaskId["task-b"]).toBeUndefined();
    expect(store.getState().taskMRAutomation.loading["task-b"]).toBe(true);
    expect(store.getState().taskMRAutomation.loading["task-a"]).toBeUndefined();
    expect(store.getState().taskMRAutomation.saving["task-a"]).toBe(true);
    expect(store.getState().taskMRAutomation.errors["task-b"]).toBe("boom");
    expect(store.getState().taskMRAutomation.errors["task-a"]).toBeUndefined();
  });

  it("setTaskMRAutomationOptions fully replaces a task's stored options", () => {
    const store = makeStore();
    store
      .getState()
      .setTaskMRAutomationOptions("task-a", makeOptions({ prompt_on_review_requested: true }));
    store.getState().setTaskMRAutomationOptions(
      "task-a",
      makeOptions({
        prompt_on_merged: true,
        prompt_on_closed: true,
        review_reviewer_username: "bob",
        updated_at: "2026-01-02T00:00:00Z",
      }),
    );

    const stored = store.getState().taskMRAutomation.byTaskId["task-a"];
    expect(stored?.prompt_on_review_requested).toBe(false);
    expect(stored?.prompt_on_merged).toBe(true);
    expect(stored?.review_reviewer_username).toBe("bob");
  });

  it("setTaskMRAutomationError clears the error by passing null", () => {
    const store = makeStore();
    store.getState().setTaskMRAutomationError("task-a", "boom");
    expect(store.getState().taskMRAutomation.errors["task-a"]).toBe("boom");
    store.getState().setTaskMRAutomationError("task-a", null);
    expect(store.getState().taskMRAutomation.errors["task-a"]).toBeNull();
  });
});