mirror of
https://gitea.toothfairyai.com/ToothFairyAI/tf_code.git
synced 2026-04-05 00:23:10 +00:00
103 lines
3.1 KiB
TypeScript
103 lines
3.1 KiB
TypeScript
import { describe, expect, test } from "bun:test"
|
|
import type {
|
|
FileDiff,
|
|
Message,
|
|
Part,
|
|
PermissionRequest,
|
|
QuestionRequest,
|
|
SessionStatus,
|
|
Todo,
|
|
} from "@opencode-ai/sdk/v2/client"
|
|
import { dropSessionCaches, pickSessionCacheEvictions } from "./session-cache"
|
|
|
|
const msg = (id: string, sessionID: string) =>
|
|
({
|
|
id,
|
|
sessionID,
|
|
role: "user",
|
|
time: { created: 1 },
|
|
agent: "assistant",
|
|
model: { providerID: "openai", modelID: "gpt" },
|
|
}) as Message
|
|
|
|
const part = (id: string, sessionID: string, messageID: string) =>
|
|
({
|
|
id,
|
|
sessionID,
|
|
messageID,
|
|
type: "text",
|
|
text: id,
|
|
}) as Part
|
|
|
|
describe("app session cache", () => {
|
|
test("dropSessionCaches clears orphaned parts without message rows", () => {
|
|
const store: {
|
|
session_status: Record<string, SessionStatus | undefined>
|
|
session_diff: Record<string, FileDiff[] | undefined>
|
|
todo: Record<string, Todo[] | undefined>
|
|
message: Record<string, Message[] | undefined>
|
|
part: Record<string, Part[] | undefined>
|
|
permission: Record<string, PermissionRequest[] | undefined>
|
|
question: Record<string, QuestionRequest[] | undefined>
|
|
} = {
|
|
session_status: { ses_1: { type: "busy" } as SessionStatus },
|
|
session_diff: { ses_1: [] },
|
|
todo: { ses_1: [] as Todo[] },
|
|
message: {},
|
|
part: { msg_1: [part("prt_1", "ses_1", "msg_1")] },
|
|
permission: { ses_1: [] as PermissionRequest[] },
|
|
question: { ses_1: [] as QuestionRequest[] },
|
|
}
|
|
|
|
dropSessionCaches(store, ["ses_1"])
|
|
|
|
expect(store.message.ses_1).toBeUndefined()
|
|
expect(store.part.msg_1).toBeUndefined()
|
|
expect(store.todo.ses_1).toBeUndefined()
|
|
expect(store.session_diff.ses_1).toBeUndefined()
|
|
expect(store.session_status.ses_1).toBeUndefined()
|
|
expect(store.permission.ses_1).toBeUndefined()
|
|
expect(store.question.ses_1).toBeUndefined()
|
|
})
|
|
|
|
test("dropSessionCaches clears message-backed parts", () => {
|
|
const m = msg("msg_1", "ses_1")
|
|
const store: {
|
|
session_status: Record<string, SessionStatus | undefined>
|
|
session_diff: Record<string, FileDiff[] | undefined>
|
|
todo: Record<string, Todo[] | undefined>
|
|
message: Record<string, Message[] | undefined>
|
|
part: Record<string, Part[] | undefined>
|
|
permission: Record<string, PermissionRequest[] | undefined>
|
|
question: Record<string, QuestionRequest[] | undefined>
|
|
} = {
|
|
session_status: {},
|
|
session_diff: {},
|
|
todo: {},
|
|
message: { ses_1: [m] },
|
|
part: { [m.id]: [part("prt_1", "ses_1", m.id)] },
|
|
permission: {},
|
|
question: {},
|
|
}
|
|
|
|
dropSessionCaches(store, ["ses_1"])
|
|
|
|
expect(store.message.ses_1).toBeUndefined()
|
|
expect(store.part[m.id]).toBeUndefined()
|
|
})
|
|
|
|
test("pickSessionCacheEvictions preserves requested sessions", () => {
|
|
const seen = new Set(["ses_1", "ses_2", "ses_3"])
|
|
|
|
const stale = pickSessionCacheEvictions({
|
|
seen,
|
|
keep: "ses_4",
|
|
limit: 2,
|
|
preserve: ["ses_1"],
|
|
})
|
|
|
|
expect(stale).toEqual(["ses_2", "ses_3"])
|
|
expect([...seen]).toEqual(["ses_1", "ses_4"])
|
|
})
|
|
})
|