mirror of https://github.com/openclaw/openclaw.git
60 lines
1.7 KiB
TypeScript
60 lines
1.7 KiB
TypeScript
import { describe, expect, it, vi } from "vitest";
|
|
import { createChannelReplyPipeline } from "./channel-reply-pipeline.js";
|
|
|
|
describe("createChannelReplyPipeline", () => {
|
|
it("builds prefix options without forcing typing support", () => {
|
|
const pipeline = createChannelReplyPipeline({
|
|
cfg: {},
|
|
agentId: "main",
|
|
channel: "telegram",
|
|
accountId: "default",
|
|
});
|
|
|
|
expect(typeof pipeline.onModelSelected).toBe("function");
|
|
expect(typeof pipeline.responsePrefixContextProvider).toBe("function");
|
|
expect(pipeline.typingCallbacks).toBeUndefined();
|
|
});
|
|
|
|
it("builds typing callbacks when typing config is provided", async () => {
|
|
const start = vi.fn(async () => {});
|
|
const stop = vi.fn(async () => {});
|
|
const pipeline = createChannelReplyPipeline({
|
|
cfg: {},
|
|
agentId: "main",
|
|
channel: "discord",
|
|
accountId: "default",
|
|
typing: {
|
|
start,
|
|
stop,
|
|
onStartError: () => {},
|
|
},
|
|
});
|
|
|
|
await pipeline.typingCallbacks?.onReplyStart();
|
|
pipeline.typingCallbacks?.onIdle?.();
|
|
|
|
expect(start).toHaveBeenCalled();
|
|
expect(stop).toHaveBeenCalled();
|
|
});
|
|
|
|
it("preserves explicit typing callbacks when a channel needs custom lifecycle hooks", async () => {
|
|
const onReplyStart = vi.fn(async () => {});
|
|
const onIdle = vi.fn(() => {});
|
|
const pipeline = createChannelReplyPipeline({
|
|
cfg: {},
|
|
agentId: "main",
|
|
channel: "bluebubbles",
|
|
typingCallbacks: {
|
|
onReplyStart,
|
|
onIdle,
|
|
},
|
|
});
|
|
|
|
await pipeline.typingCallbacks?.onReplyStart();
|
|
pipeline.typingCallbacks?.onIdle?.();
|
|
|
|
expect(onReplyStart).toHaveBeenCalledTimes(1);
|
|
expect(onIdle).toHaveBeenCalledTimes(1);
|
|
});
|
|
});
|