mirror of
https://github.com/Kilo-Org/kilocode.git
synced 2026-08-28 19:11:03 +08:00
marius-kilocode
90 lines
3.1 KiB
TypeScript
90 lines
3.1 KiB
TypeScript
import { describe, expect, it } from "bun:test"
|
|
import { createProjectStore } from "../../webview-ui/agent-manager/project/store"
|
|
import {
|
|
clearFailedDelete,
|
|
clearMultiVersionBusy,
|
|
markMultiVersionBusy,
|
|
} from "../../webview-ui/agent-manager/project/progress"
|
|
import { createProjectRegistry } from "../../webview-ui/agent-manager/project/registry"
|
|
|
|
const state = (projectId: string) => ({
|
|
type: "agentManager.state" as const,
|
|
projectId,
|
|
worktrees: [
|
|
{
|
|
id: "same",
|
|
branch: `${projectId}-same`,
|
|
path: `/repo/${projectId}/same`,
|
|
parentBranch: "main",
|
|
createdAt: "2026-01-01",
|
|
groupId: "group",
|
|
},
|
|
],
|
|
sessions: [{ id: `${projectId}-session`, worktreeId: "same", createdAt: "2026-01-01" }],
|
|
sections: [],
|
|
})
|
|
|
|
describe("multi-project progress state", () => {
|
|
it.each([undefined, "b"])("clears failed deletion only for the resolved project %s", (projectId) => {
|
|
const registry = createProjectRegistry({ persisted: {}, activeId: () => "a" })
|
|
for (const id of ["a", "b"]) {
|
|
registry.ensure(id).setBusy(
|
|
new Map([
|
|
["same", { reason: "deleting" as const }],
|
|
["other", { reason: "deleting" as const }],
|
|
]),
|
|
)
|
|
}
|
|
|
|
const store = registry.ensure(projectId ?? "a")
|
|
const peer = registry.ensure(projectId ? "a" : "b")
|
|
clearFailedDelete({ type: "error", message: "failed", code: "unrelated", projectId, worktreeId: "same" }, registry)
|
|
expect(store.busy().has("same")).toBe(true)
|
|
clearFailedDelete(
|
|
{ type: "error", message: "failed", code: "agentManager.worktreeDeleteFailed", projectId, worktreeId: "same" },
|
|
registry,
|
|
)
|
|
|
|
expect(store.busy().has("same")).toBe(false)
|
|
expect(peer.busy().has("same")).toBe(true)
|
|
expect(store.busy().has("other")).toBe(true)
|
|
})
|
|
|
|
it("updates only the owning project's grouped worktrees", () => {
|
|
const first = createProjectStore("a")
|
|
const second = createProjectStore("b")
|
|
first.applyState(state("a"))
|
|
second.applyState(state("b"))
|
|
first.setBusy(new Map([["same", { reason: "setting-up" as const }]]))
|
|
second.setBusy(new Map([["same", { reason: "setting-up" as const }]]))
|
|
|
|
clearMultiVersionBusy(second, "group")
|
|
|
|
expect(first.busy().has("same")).toBe(true)
|
|
expect(second.busy().has("same")).toBe(false)
|
|
|
|
second.setBusy(new Map([["same", { reason: "deleting" as const }]]))
|
|
clearMultiVersionBusy(second, "group")
|
|
expect(second.busy().get("same")?.reason).toBe("deleting")
|
|
})
|
|
|
|
it("marks a newly created grouped worktree as busy in its project store", () => {
|
|
const store = createProjectStore("a")
|
|
store.applyState(state("a"))
|
|
|
|
markMultiVersionBusy(store, "a-session")
|
|
|
|
expect(store.busy().get("same")?.reason).toBe("setting-up")
|
|
})
|
|
|
|
it("does not replace deletion progress when marking a grouped worktree", () => {
|
|
const store = createProjectStore("a")
|
|
store.applyState(state("a"))
|
|
store.setBusy(new Map([["same", { reason: "deleting" as const }]]))
|
|
|
|
markMultiVersionBusy(store, "a-session")
|
|
|
|
expect(store.busy().get("same")?.reason).toBe("deleting")
|
|
})
|
|
})
|