48 lines
2.9 KiB
TypeScript
48 lines
2.9 KiB
TypeScript
import { spawn } from 'node:child_process'
|
|
import { copyFile, mkdtemp, rm } from 'node:fs/promises'
|
|
import { tmpdir } from 'node:os'
|
|
import { join } from 'node:path'
|
|
import { pathToFileURL } from 'node:url'
|
|
import type { Duplex } from 'node:stream'
|
|
import { expect, it, onTestFinished } from 'vitest'
|
|
import { JsonChannel } from '../src/channel.ts'
|
|
import { decodeCodeJsonWire, encodeCodeJsonWire } from '../src/json-wire.ts'
|
|
|
|
it('boots an unbuilt source closure outside the workspace and exchanges tool replies', async () => {
|
|
const directory = await mkdtemp(join(tmpdir(), 'dsh-node-source-'))
|
|
onTestFinished(async () => { await rm(directory, { recursive: true, force: true }) })
|
|
for (const file of ['process.ts', 'bootstrap.ts', 'channel.ts', 'json-wire.ts', 'output-json.ts', 'protocol.ts']) {
|
|
await copyFile(new URL(`../src/${file}`, import.meta.url), join(directory, file))
|
|
}
|
|
const source = `import {Socket} from 'node:net';import {runNodeMain} from ${JSON.stringify(pathToFileURL(join(directory, 'process.ts')).href)};await runNodeMain(new Socket({fd:7,readable:true,writable:true}),100000,process);`
|
|
const child = spawn(process.execPath, ['--input-type=module', '--eval', source], {
|
|
env: { PLACEHOLDER_SECRET: 'fixture-only' },
|
|
stdio: ['ignore', 'pipe', 'pipe', 'ignore', 'ignore', 'ignore', 'ignore', 'pipe'],
|
|
})
|
|
let stderr = ''
|
|
child.stderr?.on('data', (chunk: Buffer) => { stderr += chunk.toString('utf8') })
|
|
const finished = new Promise<void>((resolve) => { child.once('close', () => { resolve() }) })
|
|
const completed = Promise.withResolvers<unknown>()
|
|
child.once('error', error => completed.reject(error))
|
|
child.once('exit', (code) => { if (code !== 0) completed.reject(new Error(`child exit ${code}: ${stderr}`)) })
|
|
const channel = new JsonChannel(child.stdio[7] as Duplex, 100_000, (raw) => {
|
|
const message = raw as { type: string; id?: number; args?: unknown; value?: unknown; error?: unknown }
|
|
if (message.type === 'ready') {
|
|
void channel.send({ type: 'boot', data: {
|
|
code: 'const answer = await tools.echo({ n: 21 }); return { answer, env: { ...process.env } }',
|
|
namespaces: [{ global: 'tools', names: ['echo'] }],
|
|
maxOutputBytes: 10_000,
|
|
} }).catch(error => completed.reject(error))
|
|
} else if (message.type === 'call') {
|
|
expect(decodeCodeJsonWire(message.args)).toEqual({ n: 21 })
|
|
void channel.send({ type: 'reply', id: message.id, ok: true, value: encodeCodeJsonWire(42) }).catch(error => completed.reject(error))
|
|
} else if (message.type === 'done') {
|
|
if (message.error !== undefined) completed.reject(new Error(JSON.stringify(message.error)))
|
|
else completed.resolve(decodeCodeJsonWire(message.value))
|
|
}
|
|
}, error => completed.reject(error))
|
|
onTestFinished(async () => { channel.close(); child.kill(); await finished })
|
|
expect(await completed.promise).toEqual({ answer: 42, env: {} })
|
|
await finished
|
|
})
|