48 lines
1.3 KiB
TypeScript
48 lines
1.3 KiB
TypeScript
|
import { describe, expect, test } from "bun:test"
|
||
|
|
|
||
|
|
import { resolveOrCreateSessionId } from "./subagent-session-creator"
|
||
|
|
import { _resetForTesting, subagentSessions } from "../../features/claude-code-session-state"
|
||
|
|
|
||
|
|
describe("call-omo-agent resolveOrCreateSessionId", () => {
|
||
|
|
test("tracks newly created child session as subagent session", async () => {
|
||
|
|
// given
|
||
|
|
_resetForTesting()
|
||
|
|
|
||
|
|
const createCalls: Array<unknown> = []
|
||
|
|
const ctx = {
|
||
|
|
directory: "/project",
|
||
|
|
client: {
|
||
|
|
session: {
|
||
|
|
get: async () => ({ data: { directory: "/parent" } }),
|
||
|
|
create: async (args: unknown) => {
|
||
|
|
createCalls.push(args)
|
||
|
|
return { data: { id: "ses_child_sync" } }
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
const args = {
|
||
|
|
description: "sync test",
|
||
|
|
prompt: "hello",
|
||
|
|
subagent_type: "explore",
|
||
|
|
run_in_background: false,
|
||
|
|
}
|
||
|
|
|
||
|
|
const toolContext = {
|
||
|
|
sessionID: "ses_parent",
|
||
|
|
messageID: "msg_parent",
|
||
|
|
agent: "sisyphus",
|
||
|
|
abort: new AbortController().signal,
|
||
|
|
}
|
||
|
|
|
||
|
|
// when
|
||
|
|
const result = await resolveOrCreateSessionId(ctx as any, args as any, toolContext as any)
|
||
|
|
|
||
|
|
// then
|
||
|
|
expect(result).toEqual({ ok: true, sessionID: "ses_child_sync" })
|
||
|
|
expect(createCalls).toHaveLength(1)
|
||
|
|
expect(subagentSessions.has("ses_child_sync")).toBe(true)
|
||
|
|
})
|
||
|
|
})
|