Something went wrong. Try again.
Something went wrong. Try again.
2.5 kB · 43 lines
TypeScript
at main
1234567891011121314151617181920212223242526272829303132333435363738394041424344import { describe, expect, it } from "vitest";import { commandModel, splitCommand } from "../../runner/authorModel";
/** The model adapter with real processes: a command that reads the prompt on stdin and answers on stdout. */
describe("splitCommand", () => { it("splits on spaces, keeps quoted parts together, and refuses an unclosed quote or nothing", () => { expect(splitCommand(`claude -p --tools ""`)).toEqual(["claude", "-p", "--tools", ""]); expect(splitCommand(`node -e 'console.log("a b")'`)).toEqual(["node", "-e", 'console.log("a b")']); expect(splitCommand(`a "b c" d\\ e`)).toEqual(["a", "b c", "d e"]); expect(() => splitCommand(`a "b`)).toThrow(/quote/); expect(() => splitCommand(" ")).toThrow(/empty/); });});
describe("commandModel", () => { const node = process.execPath; it("sends the prompt on stdin and returns what the command prints", async () => { const model = commandModel({ command: [node, "-e", "let s='';process.stdin.on('data',d=>s+=d).on('end',()=>process.stdout.write('got:'+s.toUpperCase()))"] }); expect(await model.propose("hello prompt")).toBe("got:HELLO PROMPT"); }); it("does not hand the command the runner's secrets", async () => { process.env.AITE_GITLAB_TOKEN = "glpat-abcdefghij0123456789"; // aite:allow-secret try { const model = commandModel({ command: [node, "-e", "process.stdout.write(Object.keys(process.env).filter(k=>k.startsWith('AITE_')).join(',') + '|' + (process.env.PATH ? 'path' : 'nopath'))"] }); expect(await model.propose("x")).toBe("|path"); } finally { delete process.env.AITE_GITLAB_TOKEN; } }); it("a command that exits with an error is an error carrying its first line of complaint", async () => { const model = commandModel({ command: [node, "-e", "console.error('model is down\\nsecond line');process.exit(3)"] }); await expect(model.propose("x")).rejects.toThrow(/exit 3.*model is down/s); }); it("a command that is not there, and one that answers nothing, are errors", async () => { await expect(commandModel({ command: ["/no/such/command-aite"] }).propose("x")).rejects.toThrow(/cannot run/); await expect(commandModel({ command: [node, "-e", "0"] }).propose("x")).rejects.toThrow(/nothing/); }); it("a command that takes too long is stopped", async () => { const model = commandModel({ command: [node, "-e", "setInterval(()=>{},1000)"], timeoutMs: 700 }); await expect(model.propose("x")).rejects.toThrow(/did not answer within/); });});