YeonGyu-Kim
92 lines
2.2 KiB
TypeScript
92 lines
2.2 KiB
TypeScript
const { describe, test, expect, mock } = require("bun:test")
|
|
|
|
describe("executeSync", () => {
|
|
test("passes question=false via tools parameter to block question tool", async () => {
|
|
//#given
|
|
const { executeSync } = require("./sync-executor")
|
|
|
|
const deps = {
|
|
createOrGetSession: mock(async () => ({ sessionID: "ses-test-123", isNew: true })),
|
|
waitForCompletion: mock(async () => {}),
|
|
processMessages: mock(async () => "agent response"),
|
|
}
|
|
|
|
let promptArgs: any
|
|
const promptAsync = mock(async (input: any) => {
|
|
promptArgs = input
|
|
return { data: {} }
|
|
})
|
|
|
|
const args = {
|
|
subagent_type: "explore",
|
|
description: "test task",
|
|
prompt: "find something",
|
|
}
|
|
|
|
const toolContext = {
|
|
sessionID: "parent-session",
|
|
messageID: "msg-1",
|
|
agent: "sisyphus",
|
|
abort: new AbortController().signal,
|
|
metadata: mock(async () => {}),
|
|
}
|
|
|
|
const ctx = {
|
|
client: {
|
|
session: { promptAsync },
|
|
},
|
|
}
|
|
|
|
//#when
|
|
await executeSync(args, toolContext, ctx as any, deps)
|
|
|
|
//#then
|
|
expect(promptAsync).toHaveBeenCalled()
|
|
expect(promptArgs.body.tools.question).toBe(false)
|
|
})
|
|
|
|
test("passes task=false via tools parameter", async () => {
|
|
//#given
|
|
const { executeSync } = require("./sync-executor")
|
|
|
|
const deps = {
|
|
createOrGetSession: mock(async () => ({ sessionID: "ses-test-123", isNew: true })),
|
|
waitForCompletion: mock(async () => {}),
|
|
processMessages: mock(async () => "agent response"),
|
|
}
|
|
|
|
let promptArgs: any
|
|
const promptAsync = mock(async (input: any) => {
|
|
promptArgs = input
|
|
return { data: {} }
|
|
})
|
|
|
|
const args = {
|
|
subagent_type: "librarian",
|
|
description: "search docs",
|
|
prompt: "find docs",
|
|
}
|
|
|
|
const toolContext = {
|
|
sessionID: "parent-session",
|
|
messageID: "msg-2",
|
|
agent: "sisyphus",
|
|
abort: new AbortController().signal,
|
|
metadata: mock(async () => {}),
|
|
}
|
|
|
|
const ctx = {
|
|
client: {
|
|
session: { promptAsync },
|
|
},
|
|
}
|
|
|
|
//#when
|
|
await executeSync(args, toolContext, ctx as any, deps)
|
|
|
|
//#then
|
|
expect(promptAsync).toHaveBeenCalled()
|
|
expect(promptArgs.body.tools.task).toBe(false)
|
|
})
|
|
})
|