|
|
@@ -0,0 +1,637 @@
|
|
|
+/**
|
|
|
+ * Consumer-surface tests for the `pwsh` tool over a FAKE bash executor,
|
|
|
+ * exercised through `ctx.tools.execute()` so nothing bypasses the tool
|
|
|
+ * registry. The fake executor makes every seam outcome scriptable — output
|
|
|
+ * text, truncation, timeout, abort, nonzero exits, background handles — so
|
|
|
+ * these tests verify the schema, argument validation, workdir derivation,
|
|
|
+ * managed `DSH_*` collection, abort translation, canonical result projection,
|
|
|
+ * rendering, background task wiring, and the UI presenters. Real-pwsh behavior
|
|
|
+ * is pinned separately in integration.spec.ts.
|
|
|
+ */
|
|
|
+
|
|
|
+import { describe, expect, it } from 'vitest'
|
|
|
+import { Context } from 'cordis'
|
|
|
+import { mkdtempSync } from 'node:fs'
|
|
|
+import { tmpdir } from 'node:os'
|
|
|
+import { join, resolve as resolvePath } from 'node:path'
|
|
|
+import { CallId } from '@deepseek-ai/dsh-llm'
|
|
|
+import SystemPrompt, { renderPrompt } from '@deepseek-ai/dsh-system-prompt'
|
|
|
+import ToolRegistry, { TOOL_ABORTED, TOOL_ABORTED_BEFORE_DISPATCH } from '@deepseek-ai/dsh-tools'
|
|
|
+import LocalTaskService from '@deepseek-ai/dsh-tasks-local'
|
|
|
+import * as ToolTasks from '@deepseek-ai/dsh-tool-tasks'
|
|
|
+import AgentRegistry from '@deepseek-ai/dsh-agent'
|
|
|
+import type { Agent } from '@deepseek-ai/dsh-agent'
|
|
|
+import { SessionId } from '@deepseek-ai/dsh-session'
|
|
|
+import { BashExecutor } from '@deepseek-ai/dsh-bash'
|
|
|
+import type { BashExecRequest, BashExecSpec, BashProcess, BashRunResult } from '@deepseek-ai/dsh-bash'
|
|
|
+import * as ToolPwsh from '@deepseek-ai/dsh-tool-pwsh'
|
|
|
+import * as BashEnvPlugin from '@deepseek-ai/dsh-bash-env'
|
|
|
+import type { BashProcessRead } from '@deepseek-ai/dsh-bash'
|
|
|
+import { processOutcome } from '../src/background.ts'
|
|
|
+import { renderPwshProcessRead } from '../src/render.ts'
|
|
|
+
|
|
|
+const testToolSignal = new AbortController().signal
|
|
|
+
|
|
|
+/**
|
|
|
+ * A scriptable fake executor: `resolve()` mirrors the real defaulting, `run()`
|
|
|
+ * returns the armed foreground script, `start()` returns the armed background
|
|
|
+ * handle.
|
|
|
+ */
|
|
|
+class FakeBash extends BashExecutor {
|
|
|
+ requests: BashExecRequest[] = []
|
|
|
+ specs: BashExecSpec[] = []
|
|
|
+ startCalls = 0
|
|
|
+ handler: (spec: BashExecSpec) => BashRunResult = () => runResult('')
|
|
|
+ backgroundHandler: (spec: BashExecSpec) => BashProcess = () => fakeProcess('bg-ok\n')
|
|
|
+
|
|
|
+ override resolve(request: BashExecRequest): BashExecSpec {
|
|
|
+ this.requests.push(request)
|
|
|
+ return {
|
|
|
+ command: request.command,
|
|
|
+ workdir: request.workdir ?? process.cwd(),
|
|
|
+ timeoutMs: request.timeoutMs ?? 60_000,
|
|
|
+ stdoutMaxBytes: request.stdoutMaxBytes ?? 64_000,
|
|
|
+ ...request.signal ? { signal: request.signal } : {},
|
|
|
+ ...request.stdin !== undefined ? { stdin: request.stdin } : {},
|
|
|
+ ...request.env !== undefined ? { env: request.env } : {},
|
|
|
+ ...request.dshEnv !== undefined ? { dshEnv: request.dshEnv } : {},
|
|
|
+ sandboxPolicy: request.sandboxPolicy,
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ override async run(spec: BashExecSpec): Promise<BashRunResult> {
|
|
|
+ this.specs.push(spec)
|
|
|
+ return this.handler(spec)
|
|
|
+ }
|
|
|
+
|
|
|
+ override start(spec: BashExecSpec): BashProcess {
|
|
|
+ this.startCalls++
|
|
|
+ this.specs.push(spec)
|
|
|
+ return this.backgroundHandler(spec)
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+/** A successful run result over the given stdout; overrides script the failure shapes. */
|
|
|
+function runResult(stdout: string, overrides?: Partial<BashRunResult>): BashRunResult {
|
|
|
+ return {
|
|
|
+ exitCode: 0,
|
|
|
+ signal: null,
|
|
|
+ timedOut: false,
|
|
|
+ aborted: false,
|
|
|
+ timeoutMs: 60_000,
|
|
|
+ stdout: { text: stdout, truncated: false },
|
|
|
+ stderr: { text: '', truncated: false },
|
|
|
+ ...overrides,
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+/** A settled successful background handle; overrides script failure shapes. */
|
|
|
+function fakeProcess(delta = 'bg-ok\n'): BashProcess {
|
|
|
+ let consumed = false
|
|
|
+ return {
|
|
|
+ status: 'completed',
|
|
|
+ exitCode: 0,
|
|
|
+ signal: null,
|
|
|
+ done: Promise.resolve(),
|
|
|
+ readOutput: () => {
|
|
|
+ if (consumed) return { delta: '', lossy: false }
|
|
|
+ consumed = true
|
|
|
+ return { delta, lossy: false }
|
|
|
+ },
|
|
|
+ kill: () => false,
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+/** A running background handle whose kill() settles it as killed (like a real task_kill). */
|
|
|
+function killableProcess(): BashProcess {
|
|
|
+ let resolveDone: () => void = () => {}
|
|
|
+ const done = new Promise<void>((resolve) => { resolveDone = resolve })
|
|
|
+ const proc: BashProcess = {
|
|
|
+ status: 'running',
|
|
|
+ exitCode: null,
|
|
|
+ signal: null,
|
|
|
+ done,
|
|
|
+ readOutput: () => ({ delta: '', lossy: false }),
|
|
|
+ kill: () => {
|
|
|
+ if (proc.status !== 'running') return false
|
|
|
+ proc.status = 'killed'
|
|
|
+ proc.signal = 'SIGTERM'
|
|
|
+ resolveDone()
|
|
|
+ return true
|
|
|
+ },
|
|
|
+ }
|
|
|
+ return proc
|
|
|
+}
|
|
|
+
|
|
|
+async function setup(toolConfig: Partial<ToolPwsh.Config> = {}, dshHome?: string) {
|
|
|
+ const ctx = new Context()
|
|
|
+ await ctx.plugin(SystemPrompt)
|
|
|
+ await ctx.plugin(ToolRegistry)
|
|
|
+ await ctx.plugin(AgentRegistry)
|
|
|
+ await ctx.plugin(BashEnvPlugin, dshHome === undefined ? {} : { dshHome })
|
|
|
+ await ctx.plugin(FakeBash)
|
|
|
+ await ctx.plugin(ToolPwsh, toolConfig)
|
|
|
+ const bash = ctx.bash as FakeBash
|
|
|
+ return { ctx, bash }
|
|
|
+}
|
|
|
+
|
|
|
+/** Full harness: the generic task runtime + its control surface, then the pwsh tool. */
|
|
|
+async function setupWithTasks(toolConfig: Partial<ToolPwsh.Config> = {}, dshHome?: string) {
|
|
|
+ const ctx = new Context()
|
|
|
+ await ctx.plugin(SystemPrompt)
|
|
|
+ await ctx.plugin(ToolRegistry)
|
|
|
+ await ctx.plugin(AgentRegistry)
|
|
|
+ await ctx.plugin(LocalTaskService)
|
|
|
+ await ctx.plugin(ToolTasks)
|
|
|
+ await ctx.plugin(BashEnvPlugin, dshHome === undefined ? {} : { dshHome })
|
|
|
+ await ctx.plugin(FakeBash)
|
|
|
+ await ctx.plugin(ToolPwsh, toolConfig)
|
|
|
+ const bash = ctx.bash as FakeBash
|
|
|
+ return { ctx, bash }
|
|
|
+}
|
|
|
+
|
|
|
+/**
|
|
|
+ * Build a fake {@link Agent} with the shared agent/session identity, give it a
|
|
|
+ * dedicated lifecycle fiber for `Agent.ctx`, and register it in `ctx.agents`.
|
|
|
+ */
|
|
|
+function registerFakeAgent(ctx: Context, sessionId: string): Agent {
|
|
|
+ const scopeFiber = ctx.plugin(() => {})
|
|
|
+ const id = SessionId(sessionId)
|
|
|
+ const agent = {
|
|
|
+ id,
|
|
|
+ ctx: scopeFiber.ctx,
|
|
|
+ session: { id, header: { version: 0, id, createdAt: 0 } },
|
|
|
+ } as unknown as Agent
|
|
|
+ ctx.agents.register(agent)
|
|
|
+ return agent
|
|
|
+}
|
|
|
+
|
|
|
+let callCounter = 0
|
|
|
+function call(ctx: Context, name: string, args: unknown, agent?: Agent) {
|
|
|
+ return ctx.tools.execute({
|
|
|
+ signal: testToolSignal,
|
|
|
+ callId: CallId(`call-${++callCounter}`),
|
|
|
+ name,
|
|
|
+ arguments: args,
|
|
|
+ ...agent ? { agent } : {},
|
|
|
+ })
|
|
|
+}
|
|
|
+
|
|
|
+function text(result: { content: { type: string; text?: string }[] }): string {
|
|
|
+ return result.content.filter(b => b.type === 'text').map(b => b.text).join('')
|
|
|
+}
|
|
|
+
|
|
|
+async function callUntilText(
|
|
|
+ ctx: Context,
|
|
|
+ name: string,
|
|
|
+ args: unknown,
|
|
|
+ expected: string,
|
|
|
+ timeoutMs = 5_000,
|
|
|
+): Promise<Awaited<ReturnType<typeof call>>> {
|
|
|
+ const deadline = Date.now() + timeoutMs
|
|
|
+ let last: Awaited<ReturnType<typeof call>> | undefined
|
|
|
+ while (Date.now() < deadline) {
|
|
|
+ last = await call(ctx, name, args)
|
|
|
+ if (text(last).includes(expected)) return last
|
|
|
+ await new Promise(resolve => setTimeout(resolve, 20))
|
|
|
+ }
|
|
|
+ throw new Error(`tool output did not include ${JSON.stringify(expected)}; last text ${JSON.stringify(last === undefined ? '' : text(last))}`)
|
|
|
+}
|
|
|
+
|
|
|
+describe('registration', () => {
|
|
|
+ it('registers the pwsh tool with its prompt section and schema', async () => {
|
|
|
+ const { ctx } = await setup()
|
|
|
+ const schema = ctx.tools.schemas().find(s => s.name === 'pwsh')
|
|
|
+ expect(schema).toBeDefined()
|
|
|
+ expect(schema?.description).toContain('PowerShell command')
|
|
|
+ expect(schema?.parameters.properties).toMatchObject({
|
|
|
+ command: { type: 'string' },
|
|
|
+ description: { type: 'string' },
|
|
|
+ timeoutMs: { type: 'number' },
|
|
|
+ workdir: { type: 'string' },
|
|
|
+ run_in_background: { type: 'boolean' },
|
|
|
+ })
|
|
|
+ expect(schema?.parameters.required).toEqual(['command', 'description'])
|
|
|
+ const prompt = renderPrompt(await ctx.systemPrompt.assemble())
|
|
|
+ expect(prompt).toContain('Non-zero exits are reported as `[exit code: N]` markers')
|
|
|
+ expect(prompt).toContain('without a signal marker')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('stays pending until ctx.bash exists (inject)', async () => {
|
|
|
+ const ctx = new Context()
|
|
|
+ await ctx.plugin(SystemPrompt)
|
|
|
+ await ctx.plugin(ToolRegistry)
|
|
|
+ await ctx.plugin(ToolPwsh)
|
|
|
+ expect(ctx.tools.schemas()).toHaveLength(0)
|
|
|
+ })
|
|
|
+
|
|
|
+ it('unregisters everything on fiber disposal (HMR safety)', async () => {
|
|
|
+ const ctx = new Context()
|
|
|
+ await ctx.plugin(SystemPrompt)
|
|
|
+ await ctx.plugin(ToolRegistry)
|
|
|
+ await ctx.plugin(BashEnvPlugin)
|
|
|
+ await ctx.plugin(FakeBash)
|
|
|
+ const fiber = await ctx.plugin(ToolPwsh)
|
|
|
+ expect(ctx.tools.schemas()).toHaveLength(1)
|
|
|
+ await fiber.dispose()
|
|
|
+ expect(ctx.tools.schemas()).toHaveLength(0)
|
|
|
+ })
|
|
|
+})
|
|
|
+
|
|
|
+describe('argument validation', () => {
|
|
|
+ it('rejects a blank command or description and a non-positive timeoutMs', async () => {
|
|
|
+ const { ctx } = await setup()
|
|
|
+ expect(text(await call(ctx, 'pwsh', { command: ' ', description: 'd' }))).toContain('expected a non-empty string')
|
|
|
+ expect(text(await call(ctx, 'pwsh', { command: 'Write-Output hi', description: ' ' }))).toContain('expected a non-empty string')
|
|
|
+ expect(text(await call(ctx, 'pwsh', { command: 'Write-Output hi', description: 'd', timeoutMs: -1 })))
|
|
|
+ .toContain('invalid timeoutMs: expected a positive number')
|
|
|
+ })
|
|
|
+})
|
|
|
+
|
|
|
+describe('execution through the bash seam', () => {
|
|
|
+ it('forwards command, session cwd, timeout, and managed DSH_* environment', async () => {
|
|
|
+ const dshHome = mkdtempSync(join(tmpdir(), 'dsh-tool-pwsh-home-'))
|
|
|
+ const { ctx, bash } = await setup({}, dshHome)
|
|
|
+ bash.handler = () => runResult('hi\n')
|
|
|
+ const agent = registerFakeAgent(ctx, 'session-1')
|
|
|
+ Object.assign(agent.session.header, { cwd: '/sessions/s1' })
|
|
|
+ const result = await call(ctx, 'pwsh', {
|
|
|
+ command: 'Write-Output hi',
|
|
|
+ description: 'say hi',
|
|
|
+ timeoutMs: 1234,
|
|
|
+ }, agent)
|
|
|
+ expect(result.isError).toBe(false)
|
|
|
+ const request = bash.requests[0]
|
|
|
+ expect(request?.command).toBe('Write-Output hi')
|
|
|
+ expect(request?.workdir).toBe('/sessions/s1')
|
|
|
+ expect(request?.timeoutMs).toBe(1234)
|
|
|
+ expect(request?.dshEnv).toEqual({
|
|
|
+ DSH_HOME: dshHome,
|
|
|
+ DSH_SHELL: '1',
|
|
|
+ DSH_SESSION_ID: 'session-1',
|
|
|
+ })
|
|
|
+ expect(bash.specs[0]?.workdir).toBe('/sessions/s1')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('resolves a relative workdir against the session cwd, absolute ones verbatim', async () => {
|
|
|
+ const { ctx, bash } = await setup()
|
|
|
+ bash.handler = () => runResult('ok\n')
|
|
|
+ const agent = registerFakeAgent(ctx, 'session-cwd')
|
|
|
+ Object.assign(agent.session.header, { cwd: '/sessions/s1' })
|
|
|
+ await call(ctx, 'pwsh', { command: 'pwd', description: 'cwd', workdir: 'sub/dir' }, agent)
|
|
|
+ expect(bash.requests[0]?.workdir).toBe(resolvePath('/sessions/s1', 'sub/dir'))
|
|
|
+ await call(ctx, 'pwsh', { command: 'pwd', description: 'cwd', workdir: resolvePath('/abs/path') }, agent)
|
|
|
+ expect(bash.requests[1]?.workdir).toBe(resolvePath('/abs/path'))
|
|
|
+ })
|
|
|
+
|
|
|
+ it('omits workdir and the session id without an agent, so executor defaulting applies', async () => {
|
|
|
+ const { ctx, bash } = await setup()
|
|
|
+ bash.handler = () => runResult('ok\n')
|
|
|
+ await call(ctx, 'pwsh', { command: 'Write-Output ok', description: 'ok' })
|
|
|
+ expect(bash.requests[0]).not.toHaveProperty('workdir')
|
|
|
+ const dshEnv = bash.requests[0]?.dshEnv
|
|
|
+ expect(dshEnv).toBeDefined()
|
|
|
+ expect(dshEnv?.['DSH_SHELL']).toBe('1')
|
|
|
+ expect(dshEnv?.['DSH_HOME']).toEqual(expect.any(String))
|
|
|
+ expect(dshEnv).not.toHaveProperty('DSH_SESSION_ID')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('forwards exec.signal into the resolved request', async () => {
|
|
|
+ const { ctx, bash } = await setup()
|
|
|
+ const controller = new AbortController()
|
|
|
+ bash.handler = () => runResult('ok\n')
|
|
|
+ await ctx.tools.execute({
|
|
|
+ signal: controller.signal,
|
|
|
+ callId: CallId('call-signal'),
|
|
|
+ name: 'pwsh',
|
|
|
+ arguments: { command: 'Write-Output ok', description: 'ok' },
|
|
|
+ })
|
|
|
+ expect(bash.requests[0]?.signal).toBe(controller.signal)
|
|
|
+ })
|
|
|
+
|
|
|
+ it('projects the canonical foreground result with stdout, stderr, and exit facts', async () => {
|
|
|
+ const { ctx, bash } = await setup()
|
|
|
+ bash.handler = () => runResult('out\n', {
|
|
|
+ exitCode: 2,
|
|
|
+ stderr: { text: 'err\n', truncated: false },
|
|
|
+ timeoutMs: 5000,
|
|
|
+ })
|
|
|
+ const result = await call(ctx, 'pwsh', { command: 'failing', description: 'fail' })
|
|
|
+ expect(result.isError).toBe(false)
|
|
|
+ if (result.isError) throw new Error('expected pwsh success')
|
|
|
+ expect(result.value).toEqual({
|
|
|
+ kind: 'foreground',
|
|
|
+ exitCode: 2,
|
|
|
+ signal: null,
|
|
|
+ timedOut: false,
|
|
|
+ aborted: false,
|
|
|
+ timeoutMs: 5000,
|
|
|
+ stdout: { text: 'out\n', truncated: false },
|
|
|
+ stderr: { text: 'err\n', truncated: false },
|
|
|
+ })
|
|
|
+ expect(text(result)).toBe('out\n[stderr]\nerr\n[exit code: 2]')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('renders a clean exit without a marker and an empty body as (no output)', async () => {
|
|
|
+ const { ctx, bash } = await setup()
|
|
|
+ bash.handler = () => runResult('hi\n')
|
|
|
+ const clean = await call(ctx, 'pwsh', { command: 'Write-Output hi', description: 'say hi' })
|
|
|
+ expect(text(clean)).toBe('hi\n')
|
|
|
+
|
|
|
+ bash.handler = () => runResult('')
|
|
|
+ const empty = await call(ctx, 'pwsh', { command: 'Write-Output -NoNewline ""', description: 'nothing' })
|
|
|
+ expect(text(empty)).toBe('(no output)')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('renders stderr-only output without a stdout prefix', async () => {
|
|
|
+ const { ctx, bash } = await setup()
|
|
|
+ bash.handler = () => runResult('', {
|
|
|
+ stderr: { text: 'err\n', truncated: false },
|
|
|
+ exitCode: 1,
|
|
|
+ })
|
|
|
+ const result = await call(ctx, 'pwsh', { command: 'fail', description: 'fail' })
|
|
|
+ expect(text(result)).toBe('[stderr]\nerr\n[exit code: 1]')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('inserts the separating newline before the stderr section when stdout lacks one', async () => {
|
|
|
+ const { ctx, bash } = await setup()
|
|
|
+ bash.handler = () => runResult('out', {
|
|
|
+ stderr: { text: 'err\n', truncated: false },
|
|
|
+ exitCode: 1,
|
|
|
+ })
|
|
|
+ const result = await call(ctx, 'pwsh', { command: 'fail', description: 'fail' })
|
|
|
+ expect(text(result)).toBe('out\n[stderr]\nerr\n[exit code: 1]')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('renders the truncation notice with the spill path, then markers', async () => {
|
|
|
+ const { ctx, bash } = await setup()
|
|
|
+ bash.handler = () => runResult('tail', {
|
|
|
+ stdout: { text: 'tail', truncated: true, spillPath: '/spill/out.log' },
|
|
|
+ stderr: { text: '', truncated: false },
|
|
|
+ })
|
|
|
+ const result = await call(ctx, 'pwsh', { command: 'noisy', description: 'noise' })
|
|
|
+ expect(text(result)).toBe('tail\n[output truncated; full output: /spill/out.log]')
|
|
|
+
|
|
|
+ bash.handler = () => runResult('', { timedOut: true, exitCode: null, signal: 'SIGTERM', timeoutMs: 500 })
|
|
|
+ const timedOut = await call(ctx, 'pwsh', { command: 'slow', description: 'slow' })
|
|
|
+ // A timeout kill carries both facts, mirroring the bash tool's markers.
|
|
|
+ expect(text(timedOut)).toBe('(no output)\n[timed out after 500ms]\n[killed by signal: SIGTERM]')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('renders the truncation notice with (unavailable) when no spill path exists', async () => {
|
|
|
+ const { ctx, bash } = await setup()
|
|
|
+ bash.handler = () => runResult('tail', {
|
|
|
+ stdout: { text: 'tail', truncated: true },
|
|
|
+ stderr: { text: '', truncated: false },
|
|
|
+ })
|
|
|
+ const result = await call(ctx, 'pwsh', { command: 'noisy', description: 'noise' })
|
|
|
+ expect(text(result)).toBe('tail\n[output truncated; full output: (unavailable)]')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('translates an aborted run into the TOOL_ABORTED HarnessError', async () => {
|
|
|
+ const { ctx, bash } = await setup()
|
|
|
+ bash.handler = () => runResult('', { aborted: true, exitCode: null, signal: 'SIGTERM' })
|
|
|
+ const result = await call(ctx, 'pwsh', { command: 'Start-Sleep -Seconds 60', description: 'sleep' })
|
|
|
+ expect(result.isError).toBe(true)
|
|
|
+ expect(result.error).toMatchObject({ info: { name: 'AbortError', code: TOOL_ABORTED } })
|
|
|
+ })
|
|
|
+})
|
|
|
+
|
|
|
+describe('background execution through the task runtime', () => {
|
|
|
+ it('run_in_background acks with the task id, readable through the REAL task_output tool', async () => {
|
|
|
+ const { ctx } = await setupWithTasks()
|
|
|
+ const started = await call(ctx, 'pwsh', { command: 'Write-Output bg-ok', description: 'test command', run_in_background: true })
|
|
|
+ expect(started.isError).toBe(false)
|
|
|
+ if (started.isError) throw new Error('expected background pwsh success')
|
|
|
+ expect(started.value).toEqual({ kind: 'background', taskId: 'pwsh-1' })
|
|
|
+ expect(text(started)).toBe('started background task pwsh-1')
|
|
|
+
|
|
|
+ const read = await callUntilText(ctx, 'task_output', { task_id: 'pwsh-1' }, 'bg-ok')
|
|
|
+ expect(text(read)).toContain('bg-ok')
|
|
|
+ // A later read reports the terminal outcome in the generic status line.
|
|
|
+ const final = await callUntilText(ctx, 'task_output', { task_id: 'pwsh-1' }, '[status: completed, exit code: 0]')
|
|
|
+ expect(final.isError).toBe(false)
|
|
|
+ })
|
|
|
+
|
|
|
+ it('a running background task is killable through the REAL task_kill tool', async () => {
|
|
|
+ const { ctx, bash } = await setupWithTasks()
|
|
|
+ bash.backgroundHandler = () => killableProcess()
|
|
|
+ await call(ctx, 'pwsh', { command: 'Start-Sleep -Seconds 60', description: 'test command', run_in_background: true })
|
|
|
+
|
|
|
+ const killed = await call(ctx, 'task_kill', { task_id: 'pwsh-1' })
|
|
|
+ expect(text(killed)).toBe('requested cancellation of task pwsh-1')
|
|
|
+ // The cancel reached the process handle; the task settles as killed with
|
|
|
+ // the signal detail mapped by processOutcome.
|
|
|
+ const final = await call(ctx, 'task_output', { task_id: 'pwsh-1', wait: true })
|
|
|
+ expect(text(final)).toContain('[status: killed, signal: SIGTERM]')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('a background task started by an agent is registered with that agent as owner', async () => {
|
|
|
+ const { ctx } = await setupWithTasks()
|
|
|
+ const agent = registerFakeAgent(ctx, 'sess-owner')
|
|
|
+ const started = await call(ctx, 'pwsh', { command: 'Start-Sleep -Seconds 60', description: 'test command', run_in_background: true }, agent)
|
|
|
+ expect(text(started)).toBe('started background task pwsh-1')
|
|
|
+
|
|
|
+ const anon = await call(ctx, 'task_output', { task_id: 'pwsh-1' })
|
|
|
+ expect(anon.isError).toBe(true)
|
|
|
+ expect(text(anon)).toMatch(/belongs to another session/)
|
|
|
+
|
|
|
+ const killed = await call(ctx, 'task_kill', { task_id: 'pwsh-1' }, agent)
|
|
|
+ expect(killed.isError).toBe(false)
|
|
|
+ await call(ctx, 'task_output', { task_id: 'pwsh-1', wait: true }, agent) // await settlement — no orphan
|
|
|
+ })
|
|
|
+
|
|
|
+ it('fails loud when the task runtime is not loaded', async () => {
|
|
|
+ const { ctx } = await setup() // no LocalTaskService / ToolTasks
|
|
|
+ const result = await call(ctx, 'pwsh', { command: 'Start-Sleep -Seconds 60', description: 'test command', run_in_background: true })
|
|
|
+ expect(result.isError).toBe(true)
|
|
|
+ expect(text(result)).toContain('background tasks unavailable: load @deepseek-ai/dsh-tasks and @deepseek-ai/dsh-tool-tasks')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('a pre-aborted call is skipped before the process starts', async () => {
|
|
|
+ const { ctx, bash } = await setupWithTasks()
|
|
|
+ const controller = new AbortController()
|
|
|
+ controller.abort()
|
|
|
+ const result = await ctx.tools.execute({
|
|
|
+ callId: CallId('call-pre-aborted'),
|
|
|
+ name: 'pwsh',
|
|
|
+ arguments: { command: 'Start-Sleep -Seconds 60', description: 'test command', run_in_background: true },
|
|
|
+ signal: controller.signal,
|
|
|
+ })
|
|
|
+ expect(result.isError).toBe(true)
|
|
|
+ expect(result.error).toEqual({
|
|
|
+ message: 'tool call aborted before dispatch',
|
|
|
+ info: { name: 'AbortError', code: TOOL_ABORTED_BEFORE_DISPATCH },
|
|
|
+ })
|
|
|
+ expect(bash.startCalls).toBe(0)
|
|
|
+ })
|
|
|
+
|
|
|
+ it('never spawns the process when tasks.start preflight throws (no orphan, by construction)', async () => {
|
|
|
+ // With no control surface, task preflight fails before the executor can spawn.
|
|
|
+ const ctx = new Context()
|
|
|
+ await ctx.plugin(SystemPrompt)
|
|
|
+ await ctx.plugin(ToolRegistry)
|
|
|
+ await ctx.plugin(LocalTaskService)
|
|
|
+ await ctx.plugin(BashEnvPlugin)
|
|
|
+ await ctx.plugin(FakeBash)
|
|
|
+ await ctx.plugin(ToolPwsh)
|
|
|
+ const bash = ctx.bash as FakeBash
|
|
|
+
|
|
|
+ const result = await call(ctx, 'pwsh', { command: 'Start-Sleep -Seconds 60', description: 'test command', run_in_background: true })
|
|
|
+ expect(result.isError).toBe(true)
|
|
|
+ expect(text(result)).toContain('no control surface is attached')
|
|
|
+ // Declare-then-execute: the failed preflight means no process ever ran.
|
|
|
+ expect(bash.startCalls).toBe(0)
|
|
|
+ })
|
|
|
+
|
|
|
+ it('enableRunInBackground: false removes the parameter and flips the description', async () => {
|
|
|
+ const { ctx } = await setup({ enableRunInBackground: false })
|
|
|
+ const schema = ctx.tools.schemas().find(s => s.name === 'pwsh')!
|
|
|
+ expect(Object.keys(schema.parameters.properties as Record<string, unknown>))
|
|
|
+ .toEqual(['command', 'description', 'timeoutMs', 'workdir'])
|
|
|
+ expect(schema.description).toContain('Background execution is not available')
|
|
|
+ expect(schema.description).not.toContain('run_in_background')
|
|
|
+
|
|
|
+ // Schema omission is advertising; execution must also enforce the opt-out.
|
|
|
+ const forced = await call(ctx, 'pwsh', { command: 'Write-Output hi', description: 'test command', run_in_background: true })
|
|
|
+ expect(forced.isError).toBe(true)
|
|
|
+ expect(text(forced)).toContain('run_in_background is disabled for this deployment')
|
|
|
+ const foreground = await call(ctx, 'pwsh', { command: 'Write-Output hi', description: 'test command' })
|
|
|
+ expect(foreground.isError).toBe(false)
|
|
|
+ })
|
|
|
+
|
|
|
+ it('applies the built-in background default when apply() receives a bare config', async () => {
|
|
|
+ // Bypasses the schemastery defaults on purpose: apply() must stand on its
|
|
|
+ // own `?? true` fallback when embedded programmatically without the schema.
|
|
|
+ const ctx = new Context()
|
|
|
+ await ctx.plugin(SystemPrompt)
|
|
|
+ await ctx.plugin(ToolRegistry)
|
|
|
+ await ctx.plugin(BashEnvPlugin)
|
|
|
+ await ctx.plugin(FakeBash)
|
|
|
+ ToolPwsh.apply(ctx, {})
|
|
|
+ const schema = ctx.tools.schemas()[0]!
|
|
|
+ expect(schema.parameters.properties).toHaveProperty('run_in_background')
|
|
|
+ expect(schema.description).toContain('task_output')
|
|
|
+ })
|
|
|
+})
|
|
|
+
|
|
|
+describe('UI presentation', () => {
|
|
|
+ it('a real execute renders the console view through the tool definition presenter', async () => {
|
|
|
+ const { ctx, bash } = await setup()
|
|
|
+ bash.handler = () => runResult('hi\n')
|
|
|
+ const args = { command: 'Write-Output hi', description: 'say hi' }
|
|
|
+ const result = await call(ctx, 'pwsh', args)
|
|
|
+ const view = ctx.tools.get('pwsh')?.presentResult?.(args, result)
|
|
|
+ expect(view).toEqual({
|
|
|
+ card: 'generic',
|
|
|
+ content: [{ type: 'text', text: '```console\nhi\n```' }],
|
|
|
+ })
|
|
|
+ })
|
|
|
+
|
|
|
+ it('the pending call view is a terminal card carrying command, description, and optional cwd', async () => {
|
|
|
+ const { ctx } = await setup()
|
|
|
+ const definition = ctx.tools.get('pwsh')
|
|
|
+ expect(definition?.presentCall?.({ command: 'Get-Process', description: 'List processes' }))
|
|
|
+ .toEqual({ card: 'terminal', title: 'Get-Process', description: 'List processes' })
|
|
|
+ expect(definition?.presentCall?.({ command: 'Get-Process', description: 'List processes', workdir: 'C:\\work' }))
|
|
|
+ .toMatchObject({ cwd: 'C:\\work' })
|
|
|
+ })
|
|
|
+
|
|
|
+ it('a background pending call renders the generic card like the bash tool', async () => {
|
|
|
+ const { ctx } = await setup()
|
|
|
+ const definition = ctx.tools.get('pwsh')
|
|
|
+ expect(definition?.presentCall?.({
|
|
|
+ command: 'Start-Sleep -Seconds 60',
|
|
|
+ description: 'long wait',
|
|
|
+ run_in_background: true,
|
|
|
+ })).toEqual({
|
|
|
+ card: 'generic',
|
|
|
+ title: 'Start-Sleep -Seconds 60',
|
|
|
+ kind: 'execute',
|
|
|
+ rawInput: 'Start-Sleep -Seconds 60',
|
|
|
+ content: [{ type: 'text', text: 'long wait' }],
|
|
|
+ })
|
|
|
+ })
|
|
|
+
|
|
|
+ it('presentResult falls back to undefined for multi-block or non-text content', async () => {
|
|
|
+ const { ctx } = await setup()
|
|
|
+ const definition = ctx.tools.get('pwsh')
|
|
|
+ const args = { command: 'Write-Output hi', description: 'say hi' }
|
|
|
+ const multi = { content: [{ type: 'text' as const, text: 'a' }, { type: 'text' as const, text: 'b' }], isError: false }
|
|
|
+ expect(definition?.presentResult?.(args, multi as never)).toBeUndefined()
|
|
|
+ const image = { content: [{ type: 'image' as const, text: 'a' }], isError: false }
|
|
|
+ expect(definition?.presentResult?.(args, image as never)).toBeUndefined()
|
|
|
+ })
|
|
|
+})
|
|
|
+
|
|
|
+describe('renderPwshProcessRead', () => {
|
|
|
+ const base: BashProcessRead = { delta: 'out\n', lossy: false }
|
|
|
+
|
|
|
+ it('returns the delta verbatim for a lossless read', () => {
|
|
|
+ expect(renderPwshProcessRead(base)).toBe('out\n')
|
|
|
+ expect(renderPwshProcessRead({ delta: '', lossy: false })).toBe('')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('appends the loss notice with the available spill paths', () => {
|
|
|
+ expect(renderPwshProcessRead({ ...base, lossy: true, stdoutSpillPath: 'C:\\spill\\out.log' }))
|
|
|
+ .toBe('out\n[some output was dropped from memory; full output: C:\\spill\\out.log]')
|
|
|
+ expect(renderPwshProcessRead({
|
|
|
+ ...base,
|
|
|
+ lossy: true,
|
|
|
+ stdoutSpillPath: 'C:\\spill\\out.log',
|
|
|
+ stderrSpillPath: 'C:\\spill\\err.log',
|
|
|
+ }))
|
|
|
+ .toBe('out\n[some output was dropped from memory; full output: C:\\spill\\out.log, C:\\spill\\err.log]')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('reports (unavailable) when a lossy read has no safe spill path', () => {
|
|
|
+ expect(renderPwshProcessRead({ ...base, lossy: true }))
|
|
|
+ .toBe('out\n[some output was dropped from memory; full output: (unavailable)]')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('an empty lossy delta is the notice alone', () => {
|
|
|
+ expect(renderPwshProcessRead({ delta: '', lossy: true, stderrSpillPath: 'C:\\spill\\err.log' }))
|
|
|
+ .toBe('[some output was dropped from memory; full output: C:\\spill\\err.log]')
|
|
|
+ })
|
|
|
+
|
|
|
+ it('inserts the separating newline only when the delta lacks one', () => {
|
|
|
+ expect(renderPwshProcessRead({ delta: 'tail', lossy: true }))
|
|
|
+ .toBe('tail\n[some output was dropped from memory; full output: (unavailable)]')
|
|
|
+ expect(renderPwshProcessRead({ delta: 'tail\n', lossy: true }))
|
|
|
+ .toBe('tail\n[some output was dropped from memory; full output: (unavailable)]')
|
|
|
+ })
|
|
|
+})
|
|
|
+
|
|
|
+describe('processOutcome', () => {
|
|
|
+ function settled(over: Partial<BashProcess>): BashProcess {
|
|
|
+ return {
|
|
|
+ status: 'completed',
|
|
|
+ exitCode: 0,
|
|
|
+ signal: null,
|
|
|
+ done: Promise.resolve(),
|
|
|
+ readOutput: () => ({ delta: '', lossy: false }),
|
|
|
+ kill: () => false,
|
|
|
+ ...over,
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ it('maps a signal-killed process to killed with the signal detail', () => {
|
|
|
+ expect(processOutcome(settled({ status: 'killed', signal: 'SIGTERM' })))
|
|
|
+ .toEqual({ status: 'killed', detail: 'signal: SIGTERM' })
|
|
|
+ })
|
|
|
+
|
|
|
+ it('maps a killed process without a recorded signal (kill raced exit / spawn failure)', () => {
|
|
|
+ expect(processOutcome(settled({ status: 'killed', exitCode: null })))
|
|
|
+ .toEqual({ status: 'killed', detail: 'killed before exit' })
|
|
|
+ })
|
|
|
+
|
|
|
+ it('maps a completed process to its exit code', () => {
|
|
|
+ expect(processOutcome(settled({ exitCode: 3 })))
|
|
|
+ .toEqual({ status: 'completed', detail: 'exit code: 3' })
|
|
|
+ })
|
|
|
+
|
|
|
+ it('defensively reads a null exit code as 0 (handle shapes from other executors)', () => {
|
|
|
+ expect(processOutcome(settled({ exitCode: null })))
|
|
|
+ .toEqual({ status: 'completed', detail: 'exit code: 0' })
|
|
|
+ })
|
|
|
+})
|