github/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-12 23:12:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(test): dedupe telegram draft-stream fixtures

Browse Source
This commit is contained in:
Peter Steinberger
2026-03-03 02:42:14 +00:00
parent 40f2e2b8a6
commit fdb0bf804f
3 changed files with 98 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
src/telegram/lane-delivery.test.ts
View File

@@ -1,42 +1,26 @@
import { describe, expect, it, vi } from "vitest";
import type { ReplyPayload } from "../auto-reply/types.js";
import { createTestDraftStream } from "./draft-stream.test-helpers.js";
import { createLaneTextDeliverer, type DraftLaneState, type LaneName } from "./lane-delivery.js";
type MockStreamState = {
stream: NonNullable<DraftLaneState["stream"]>;
setMessageId: (value: number | undefined) => void;
};
function createMockStream(initialMessageId?: number): MockStreamState {
let messageId = initialMessageId;
const stream = {
update: vi.fn(),
flush: vi.fn().mockResolvedValue(undefined),
messageId: vi.fn().mockImplementation(() => messageId),
clear: vi.fn().mockResolvedValue(undefined),
stop: vi.fn().mockResolvedValue(undefined),
forceNewMessage: vi.fn(),
previewMode: vi.fn().mockReturnValue("message"),
previewRevision: vi.fn().mockReturnValue(0),
} as unknown as NonNullable<DraftLaneState["stream"]>;
return {
stream,
setMessageId: (value) => {
messageId = value;
},
};
}
function createHarness(params?: {
answerMessageId?: number;
draftMaxChars?: number;
answerMessageIdAfterStop?: number;
}) {
const answer = createMockStream(params?.answerMessageId);
const reasoning = createMockStream();
const answer = createTestDraftStream({ messageId: params?.answerMessageId });
const reasoning = createTestDraftStream();
const lanes: Record<LaneName, DraftLaneState> = {
answer: { stream: answer.stream, lastPartialText: "", hasStreamedMessage: false },
reasoning: { stream: reasoning.stream, lastPartialText: "", hasStreamedMessage: false },
answer: {
stream: answer as DraftLaneState["stream"],
lastPartialText: "",
hasStreamedMessage: false,
},
reasoning: {
stream: reasoning as DraftLaneState["stream"],
lastPartialText: "",
hasStreamedMessage: false,
},
};
const sendPayload = vi.fn().mockResolvedValue(true);
const flushDraftLane = vi.fn().mockImplementation(async (lane: DraftLaneState) => {
@@ -73,7 +57,10 @@ function createHarness(params?: {
return {
deliverLaneText,
lanes,
answer,
answer: {
stream: answer,
setMessageId: answer.setMessageId,
},
sendPayload,
flushDraftLane,
stopDraftLane,