71 lines
2.7 KiB
TypeScript
71 lines
2.7 KiB
TypeScript
|
import { describe, expect, it } from 'vitest'
|
||
|
|
import { Context } from 'cordis'
|
||
|
|
import LocalProcessManager from '@deepseek-ai/dsh-process-local'
|
||
|
|
import type { ProcessSpawnSpec } from '@deepseek-ai/dsh-process'
|
||
|
|
|
||
|
|
function spec(command: string, overrides: Partial<ProcessSpawnSpec> = {}): ProcessSpawnSpec {
|
||
|
|
return {
|
||
|
|
argv: ['bash', '-c', command],
|
||
|
|
cwd: process.cwd(),
|
||
|
|
stdoutMaxBytes: 64_000,
|
||
|
|
stderrMaxBytes: 64_000,
|
||
|
|
maxSpillBytes: 64 * 1024 * 1024,
|
||
|
|
graceMs: 200,
|
||
|
|
...overrides,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
describe('LocalProcessManager', () => {
|
||
|
|
it('registers as ctx.processes and spawns managed handles', async () => {
|
||
|
|
const ctx = new Context()
|
||
|
|
const fiber = await ctx.plugin(LocalProcessManager)
|
||
|
|
const result = await ctx.processes.spawn(spec('echo managed')).done
|
||
|
|
expect(result.exitCode).toBe(0)
|
||
|
|
expect(result.stdout.text).toBe('managed\n')
|
||
|
|
await fiber.dispose()
|
||
|
|
})
|
||
|
|
|
||
|
|
it('disposal kills still-running processes and awaits their exit', async () => {
|
||
|
|
const ctx = new Context()
|
||
|
|
const fiber = await ctx.plugin(LocalProcessManager)
|
||
|
|
const handle = ctx.processes.spawn(spec('sleep 60'))
|
||
|
|
await fiber.dispose()
|
||
|
|
const outcome = await handle.done
|
||
|
|
expect(outcome.signal).toBe('SIGTERM')
|
||
|
|
})
|
||
|
|
|
||
|
|
it('a settled process leaves the live set (disposal does not re-kill it)', async () => {
|
||
|
|
const ctx = new Context()
|
||
|
|
const fiber = await ctx.plugin(LocalProcessManager)
|
||
|
|
const handle = ctx.processes.spawn(spec('true'))
|
||
|
|
const outcome = await handle.done
|
||
|
|
expect(outcome.exitCode).toBe(0)
|
||
|
|
await fiber.dispose()
|
||
|
|
})
|
||
|
|
|
||
|
|
it('disposal tolerates a handle whose spawn already failed', async () => {
|
||
|
|
const ctx = new Context()
|
||
|
|
const fiber = await ctx.plugin(LocalProcessManager)
|
||
|
|
const handle = ctx.processes.spawn(spec('true', { cwd: '/nonexistent-dir-dsh-manager-test' }))
|
||
|
|
await expect(handle.done).rejects.toThrow()
|
||
|
|
await fiber.dispose()
|
||
|
|
})
|
||
|
|
|
||
|
|
it('disposal contains a spawn-failure rejection that races teardown', async () => {
|
||
|
|
const ctx = new Context()
|
||
|
|
const fiber = await ctx.plugin(LocalProcessManager)
|
||
|
|
// Dispose before the rejection continuation removes the handle from the
|
||
|
|
// live set, so teardown itself must swallow the rejected done.
|
||
|
|
const handle = ctx.processes.spawn(spec('true', { cwd: '/nonexistent-dir-dsh-manager-test' }))
|
||
|
|
await fiber.dispose()
|
||
|
|
await expect(handle.done).rejects.toThrow()
|
||
|
|
})
|
||
|
|
|
||
|
|
it('loading a second implementation throws (one processes service per context — cordis standard)', async () => {
|
||
|
|
const ctx = new Context()
|
||
|
|
await ctx.plugin(LocalProcessManager)
|
||
|
|
class SecondManager extends LocalProcessManager {}
|
||
|
|
await expect(ctx.plugin(SecondManager)).rejects.toThrow(/service "processes" has been registered/)
|
||
|
|
})
|
||
|
|
})
|