77 lines
3.0 KiB
JavaScript
77 lines
3.0 KiB
JavaScript
|
const test = require("node:test");
|
||
|
|
const assert = require("node:assert/strict");
|
||
|
|
const {
|
||
|
|
getDriver,
|
||
|
|
listBackends,
|
||
|
|
hasCodebuddyQueryOnlyOptions,
|
||
|
|
} = require("./index.cjs");
|
||
|
|
const { codebuddySessionManager } = require("./codebuddySessionManager.cjs");
|
||
|
|
|
||
|
|
test("registry exposes SDK backends", () => {
|
||
|
|
assert.deepEqual(listBackends().sort(), ["claude", "codebuddy", "codex", "copilot", "cursor", "grok", "opencode"]);
|
||
|
|
});
|
||
|
|
|
||
|
|
test("getDriver returns a driver with runTurn", () => {
|
||
|
|
for (const key of ["claude", "codebuddy", "codex", "copilot", "cursor", "grok", "opencode"]) {
|
||
|
|
const d = getDriver(key);
|
||
|
|
assert.equal(typeof d.runTurn, "function", `${key} must expose runTurn`);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
test("getDriver throws on unknown backend", () => {
|
||
|
|
assert.throws(() => getDriver("gemini"), /No SDK driver registered for backend: gemini/);
|
||
|
|
});
|
||
|
|
|
||
|
|
test("SDK drivers expose listModels; codex returns [] (no catalog)", async () => {
|
||
|
|
for (const key of ["claude", "codebuddy", "codex", "copilot", "cursor", "grok", "opencode"]) {
|
||
|
|
assert.equal(typeof getDriver(key).listModels, "function", `${key} must expose listModels`);
|
||
|
|
}
|
||
|
|
assert.deepEqual(await getDriver("codex").listModels({}), []);
|
||
|
|
});
|
||
|
|
|
||
|
|
test("CodeBuddy keeps V2 for SessionOptions fields and falls back for query-only fields", () => {
|
||
|
|
assert.equal(hasCodebuddyQueryOnlyOptions({
|
||
|
|
agents: { reviewer: { description: "Reviews changes", prompt: "Review" } },
|
||
|
|
thinking: { type: "adaptive" },
|
||
|
|
effort: "high",
|
||
|
|
}), false);
|
||
|
|
assert.equal(hasCodebuddyQueryOnlyOptions({ maxBudgetUsd: 1 }), true);
|
||
|
|
assert.equal(hasCodebuddyQueryOnlyOptions({ sandbox: { enabled: true } }), true);
|
||
|
|
assert.equal(hasCodebuddyQueryOnlyOptions({ sandbox: { enabled: false } }), false);
|
||
|
|
assert.equal(hasCodebuddyQueryOnlyOptions({ fallbackModel: "fallback" }), true);
|
||
|
|
assert.equal(hasCodebuddyQueryOnlyOptions({ enableFileCheckpointing: false }), false);
|
||
|
|
assert.equal(hasCodebuddyQueryOnlyOptions({ outputFormat: { type: "json_schema" } }), true);
|
||
|
|
});
|
||
|
|
|
||
|
|
test("CodeBuddy forwards the explicit bypass opt-in to V2 sessions", async () => {
|
||
|
|
const originalRunTurn = codebuddySessionManager.runTurn;
|
||
|
|
let capturedSessionOptions;
|
||
|
|
codebuddySessionManager.runTurn = async ({ sessionOptions }) => {
|
||
|
|
capturedSessionOptions = sessionOptions;
|
||
|
|
return { sessionId: "v2-session", usedV2: true };
|
||
|
|
};
|
||
|
|
|
||
|
|
try {
|
||
|
|
const result = await getDriver("codebuddy").runTurn({
|
||
|
|
chatSessionId: "chat-1",
|
||
|
|
prompt: "hello",
|
||
|
|
attachments: [],
|
||
|
|
cwd: "/tmp",
|
||
|
|
env: {},
|
||
|
|
injectedMcpServers: [],
|
||
|
|
permissionMode: "auto",
|
||
|
|
toolIntegrationMode: "mcp",
|
||
|
|
emitter: {},
|
||
|
|
});
|
||
|
|
|
||
|
|
assert.deepEqual(capturedSessionOptions.extraArgs, {
|
||
|
|
"dangerously-skip-permissions": null,
|
||
|
|
});
|
||
|
|
assert.equal(capturedSessionOptions.permissionMode, "bypassPermissions");
|
||
|
|
assert.deepEqual(capturedSessionOptions.settingSources, []);
|
||
|
|
assert.deepEqual(result, { sessionId: "v2-session", usedV2: true });
|
||
|
|
} finally {
|
||
|
|
codebuddySessionManager.runTurn = originalRunTurn;
|
||
|
|
}
|
||
|
|
});
|