lobster/test/tool_mode.test.ts

import test from "node:test";
import assert from "node:assert/strict";
import { parsePipeline } from "../src/parser.js";
import { createDefaultRegistry } from "../src/commands/registry.js";
import { runPipeline } from "../src/runtime.js";

function streamOf(items) {
	return (async function* () {
		for (const item of items) yield item;
	})();
}

test("approve halts pipeline in tool mode", async () => {
	const registry = createDefaultRegistry();
	const pipeline = parsePipeline(
		"exec --json --shell \"node -e 'process.stdout.write(JSON.stringify([{a:1}]))'\" | approve --prompt 'send?' | exec --shell 'exit 1'",
	);

	const output = await runPipeline({
		pipeline,
		registry,
		input: [],
		stdin: process.stdin,
		stdout: process.stdout,
		stderr: process.stderr,
		env: process.env,
		mode: "tool",
	});

	assert.equal(output.halted, true);
	assert.equal(output.items.length, 1);
	assert.equal(output.items[0].type, "approval_request");
	assert.equal(output.items[0].items.length, 1);
	assert.deepEqual(output.items[0].items[0], { a: 1 });
});

test("approve passes through in human interactive mode only (emit required otherwise)", async () => {
	const registry = createDefaultRegistry();
	const cmd = registry.get("approve");

	const result = await cmd.run({
		input: streamOf([{ x: 1 }]),
		args: { _: [], emit: true, prompt: "ok?" },
		ctx: {
			stdin: process.stdin,
			stdout: process.stdout,
			stderr: process.stderr,
			env: process.env,
			registry,
			mode: "human",
			render: { json() {}, lines() {} },
		},
	});

	const items = [];
	for await (const it of result.output) items.push(it);
	assert.equal(result.halt, true);
	assert.equal(items[0].type, "approval_request");
});

test("ask halts pipeline in tool mode with needs_input payload", async () => {
	const registry = createDefaultRegistry();
	const cmd = registry.get("ask");

	const result = await cmd.run({
		input: streamOf([{ text: "draft v1" }]),
		args: {
			_: [],
			prompt: "Review?",
			"subject-from-stdin": true,
			schema:
				'{"type":"object","properties":{"decision":{"type":"string"}},"required":["decision"]}',
		},
		ctx: {
			stdin: process.stdin,
			stdout: process.stdout,
			stderr: process.stderr,
			env: process.env,
			registry,
			mode: "tool",
			render: { json() {}, lines() {} },
		},
	});

	const items = [];
	for await (const it of result.output) items.push(it);
	assert.equal(result.halt, true);
	assert.equal(items.length, 1);
	assert.equal(items[0].type, "input_request");
	assert.equal(items[0].prompt, "Review?");
	assert.deepEqual(items[0].subject, { text: '{"text":"draft v1"}' });
});