LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-05-22 14:17:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: guard label generator mock calls

Browse Source
This commit is contained in:
Peter Steinberger
2026-05-12 03:56:31 +01:00
parent 64e7cfa668
commit 21a80ced0e
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/auto-reply/reply/conversation-label-generator.test.ts
View File

@@ -39,6 +39,14 @@ vi.mock("../../plugins/runtime/runtime-model-auth.runtime.js", () => ({
import { generateConversationLabel } from "./conversation-label-generator.js";
function requireFirstMockCall<T>(mock: { mock: { calls: T[][] } }, label: string): T[] {
const call = mock.mock.calls.at(0);
if (!call) {
throw new Error(`expected ${label} call`);
}
return call;
}
describe("generateConversationLabel", () => {
beforeEach(() => {
completeSimple.mockReset();
@@ -108,10 +116,7 @@ describe("generateConversationLabel", () => {
});
expect(completeSimple).toHaveBeenCalledOnce();
const call = completeSimple.mock.calls[0];
if (!call) {
throw new Error("expected simple completion call");
}
const call = requireFirstMockCall(completeSimple, "simple completion");
expect(call[0]).toStrictEqual({ provider: "openai" });
expect(call[1]).toStrictEqual({
systemPrompt: "Generate a label",
@@ -144,7 +149,7 @@ describe("generateConversationLabel", () => {
});
expect(completeSimple).toHaveBeenCalledOnce();
const options = completeSimple.mock.calls[0]?.[2];
const options = requireFirstMockCall(completeSimple, "simple completion")[2];
if (!options) {
throw new Error("expected simple completion options");
}