Something went wrong. Try again.
opencode v2 mirror
Something went wrong. Try again.
7.9 kB · 196 lines
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197import { describe, expect, test } from "bun:test"import type { SessionInfo, SessionMessageAssistantTool, ShellInfo } from "@opencode/client/promise"import { createRoot } from "solid-js"import { createStore } from "solid-js/store"import { createSessionBackground } from "@/session/requests/background"
const tool = ( id: string, name: string, metadata: Record<string, string>, input: Record<string, string> = {}, status: "completed" | "running" = "completed",): SessionMessageAssistantTool => ({ id, name, type: "tool", state: status === "running" ? { status, input, metadata } : { status, input, metadata, content: [{ type: "text", text: "backgrounded" }] }, time: { created: 0 },})
const assistant = (id: string, content: SessionMessageAssistantTool[], completed?: number) => ({ id, type: "assistant" as const, agent: "build", model: { id: "model", providerID: "provider" }, content, time: { created: 0, completed },})
const session = (id: string): SessionInfo => ({ id, title: id, parentID: "root", projectID: "project", location: { directory: "/project" }, cost: 0, tokens: { input: 0, output: 0, reasoning: 0, cache: { read: 0, write: 0 } }, time: { created: 0, updated: 0 },})
const shell = (id: string, command: string): ShellInfo => ({ id, command, status: "running", cwd: "/project", shell: "sh", file: "output", metadata: { sessionID: "root" }, time: { started: 0 },})
const notification = (id: string, metadata: Record<string, string>) => ({ id, type: "synthetic" as const, text: "complete", metadata, time: { created: 0 },})
describe("createSessionBackground", () => { test("excludes completed children and shells using either shell or tool-call IDs", () => { createRoot((dispose) => { const background = createSessionBackground({ sessionID: () => "root", messages: () => [ notification("before", { source: "subagent", childID: "before-child" }), assistant("assistant", [ tool("before-part", "subagent", { status: "running", sessionID: "before-child" }), tool("shell-part", "shell", { status: "running", shellID: "process" }), tool("shell-call", "shell", { status: "running", shellID: "shell-id" }), tool("legacy-call", "shell", { status: "running", shellID: "legacy-shell" }), tool("child-part", "subagent", { status: "running", sessionID: "child" }, { agent: "explore" }), ]), notification("shell-done", { source: "shell", jobID: "shell-part" }), notification("shell-id-done", { source: "shell", shellID: "shell-id" }), notification("legacy-done", { source: "shell", jobID: "legacy-shell" }), ], sessions: () => [], status: () => "idle", shells: () => [], }) expect(background.tasks()).toEqual([{ id: "child", type: "subagent", label: "child", agent: "explore" }]) dispose() }) })
test("joins live tasks while idle without rescanning history, then switches sessions", () => { createRoot((dispose) => { const [store, setStore] = createStore({ id: "root" as string | undefined, messages: [ assistant("assistant", [ tool("shell-part", "shell", { status: "running", shellID: "shell" }, { command: "old command" }), tool("child-part", "subagent", { status: "running", sessionID: "child" }), ]), ], sessions: [session("live-child"), session("child")], status: { root: "idle", child: "idle", "live-child": "idle" } as Record<string, "idle" | "running">, shells: [{ ...shell("shell", "command"), status: "exited" as ShellInfo["status"] }], }) let scans = 0 const background = createSessionBackground({ sessionID: () => store.id, messages: (id) => { scans += 1 return id === "root" ? store.messages : [] }, sessions: () => store.sessions, status: (id) => store.status[id], shells: () => store.shells, }) const blocking = background.blocking() const initial = background.tasks() expect(initial.map((task) => task.id)).toEqual(["child", "shell"])
setStore("status", { child: "running", "live-child": "running" }) expect(background.tasks().map((task) => task.id)).toEqual(["child", "live-child", "shell"]) setStore("shells", 0, "status", "running") expect(background.tasks().at(-1)?.label).toBe("command") setStore("sessions", 1, "title", "renamed") expect(background.tasks()[0]?.label).toBe("renamed") setStore("shells", 0, "command", "updated command") const live = background.tasks() expect(live).toEqual([ { id: "child", type: "subagent", label: "renamed" }, { id: "live-child", type: "subagent", label: "live-child" }, { id: "shell", type: "shell", label: "updated command" }, ]) expect(background.blocking()).toBe(blocking) expect(scans).toBe(1)
setStore("id", "other") expect(background.tasks()).toEqual([]) setStore("id", "root") expect(background.tasks()).toEqual(live) setStore("status", { child: "idle", "live-child": "idle" }) setStore("shells", 0, "status", "exited") expect(background.tasks()).toEqual(initial) expect(scans).toBe(3) setStore("id", undefined) expect(background.tasks()).toEqual([]) dispose() }) })
test("tracks blocking, backgrounding, and completion through nested store updates", () => { createRoot((dispose) => { const [store, setStore] = createStore({ messages: [ assistant("earlier", [tool("old-part", "subagent", { sessionID: "old-child" }, {}, "running")]), assistant("current", [ tool("child-part", "subagent", { sessionID: "child" }, {}, "running"), tool("shell-part", "shell", {}, { command: "build" }, "running"), ]), assistant("completed", [], 0), ], notification: notification("notice", { source: "subagent", childID: "other-child" }), status: { child: "running", "old-child": "running" } as Record<string, "idle" | "running">, }) const messages = store.messages const background = createSessionBackground({ sessionID: () => "root", messages: () => [...store.messages, store.notification], sessions: () => [session("child"), session("old-child")], status: (id) => store.status[id], shells: () => [shell("shell", "build")], }) expect(background.blocking()).toEqual([ { type: "subagent", partID: "child-part", id: "child", label: undefined }, { type: "shell", partID: "shell-part", id: undefined, label: "build" }, ]) expect(background.tasks().map((task) => task.id)).toEqual(["old-child"])
setStore("messages", 1, "content", 0, "state", { status: "completed", input: { description: "background child" }, metadata: { status: "running", sessionID: "child" }, content: [{ type: "text", text: "backgrounded" }], }) expect(store.messages).toBe(messages) expect(background.blocking().map((task) => task.partID)).toEqual(["shell-part"]) setStore("status", "child", "idle") expect(background.tasks().map((task) => task.id)).toEqual(["child", "old-child"]) expect(background.tasks()[0]?.label).toBe("background child") setStore("notification", "metadata", "childID", "child") expect(background.tasks().map((task) => task.id)).toEqual(["old-child"]) setStore("messages", [0, 1], "time", "completed", 1) expect(background.blocking()).toEqual([]) expect(background.tasks().map((task) => task.id)).toEqual(["old-child", "shell"]) dispose() }) })})