wehub-resource-sync
96 lines
2.9 KiB
TypeScript
96 lines
2.9 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from "bun:test";
|
|
import { getBundledModel } from "@oh-my-pi/pi-catalog/models";
|
|
import { runCommitAgentSession } from "@oh-my-pi/pi-coding-agent/commit/agentic/agent";
|
|
import * as toolsModule from "@oh-my-pi/pi-coding-agent/commit/agentic/tools";
|
|
import { Settings } from "@oh-my-pi/pi-coding-agent/config/settings";
|
|
import type { CreateAgentSessionResult } from "@oh-my-pi/pi-coding-agent/sdk";
|
|
import * as sdkModule from "@oh-my-pi/pi-coding-agent/sdk";
|
|
import type { PromptOptions } from "@oh-my-pi/pi-coding-agent/session/agent-session";
|
|
|
|
describe("commit agent prompt attribution", () => {
|
|
afterEach(() => {
|
|
vi.restoreAllMocks();
|
|
});
|
|
|
|
it("marks generated commit prompts and reminders as agent-attributed", async () => {
|
|
const prompts: Array<{ text: string; options?: PromptOptions }> = [];
|
|
const session = {
|
|
prompt: async (text: string, options?: PromptOptions) => {
|
|
prompts.push({ text, options });
|
|
},
|
|
subscribe: () => () => {},
|
|
dispose: async () => {},
|
|
};
|
|
|
|
vi.spyOn(sdkModule, "createAgentSession").mockResolvedValue({ session } as unknown as CreateAgentSessionResult);
|
|
vi.spyOn(toolsModule, "createCommitTools").mockReturnValue([]);
|
|
|
|
const model = getBundledModel("anthropic", "claude-sonnet-4-5");
|
|
if (!model) {
|
|
throw new Error("Expected claude-sonnet-4-5 model to exist");
|
|
}
|
|
|
|
await runCommitAgentSession({
|
|
cwd: "/tmp",
|
|
model,
|
|
settings: Settings.isolated(),
|
|
modelRegistry: {} as never,
|
|
authStorage: {} as never,
|
|
changelogTargets: [],
|
|
requireChangelog: false,
|
|
});
|
|
|
|
expect(prompts).toHaveLength(4);
|
|
for (const prompt of prompts) {
|
|
expect(prompt.options?.attribution).toBe("agent");
|
|
expect(prompt.options?.expandPromptTemplates).toBe(false);
|
|
}
|
|
});
|
|
|
|
it("runs completion before session disposal", async () => {
|
|
const events: string[] = [];
|
|
const session = {
|
|
prompt: async () => {},
|
|
subscribe: () => () => {},
|
|
dispose: async () => {
|
|
events.push("dispose");
|
|
},
|
|
};
|
|
|
|
vi.spyOn(sdkModule, "createAgentSession").mockResolvedValue({ session } as unknown as CreateAgentSessionResult);
|
|
vi.spyOn(toolsModule, "createCommitTools").mockImplementation(options => {
|
|
options.state.proposal = {
|
|
analysis: {
|
|
type: "fix",
|
|
scope: "commit",
|
|
details: [],
|
|
issueRefs: [],
|
|
},
|
|
summary: "create commit before teardown",
|
|
warnings: [],
|
|
};
|
|
return [];
|
|
});
|
|
|
|
const model = getBundledModel("anthropic", "claude-sonnet-4-5");
|
|
if (!model) {
|
|
throw new Error("Expected claude-sonnet-4-5 model to exist");
|
|
}
|
|
|
|
await runCommitAgentSession({
|
|
cwd: "/tmp",
|
|
model,
|
|
settings: Settings.isolated(),
|
|
modelRegistry: {} as never,
|
|
authStorage: {} as never,
|
|
changelogTargets: [],
|
|
requireChangelog: false,
|
|
onComplete: state => {
|
|
events.push(state.proposal?.summary ?? "missing proposal");
|
|
},
|
|
});
|
|
|
|
expect(events).toEqual(["create commit before teardown", "dispose"]);
|
|
});
|
|
});
|