nyrn
/
openclaw
mirror of https://github.com/openclaw/openclaw.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test: annotate chat abort helper exports (#45346)

This commit is contained in:
Frank Yang 2026-03-14 02:03:14 +08:00 committed by GitHub
parent 1f85c9af68
commit 987c254eea
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 22 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
src/gateway/server-methods/chat.abort.test-helpers.ts
View File

@ -1,5 +1,6 @@
import { vi } from "vitest";
import type { GatewayRequestHandler } from "./types.js";
import type { Mock } from "vitest";
import type { GatewayRequestHandler, RespondFn } from "./types.js";
export function createActiveRun(
sessionKey: string,
@ -20,7 +21,23 @@ export function createActiveRun(
};
}
export function createChatAbortContext(overrides: Record<string, unknown> = {}) {
export type ChatAbortTestContext = Record<string, unknown> & {
chatAbortControllers: Map<string, ReturnType<typeof createActiveRun>>;
chatRunBuffers: Map<string, string>;
chatDeltaSentAt: Map<string, number>;
chatAbortedRuns: Map<string, number>;
removeChatRun: (...args: unknown[]) => { sessionKey: string; clientRunId: string } | undefined;
agentRunSeq: Map<string, number>;
broadcast: (...args: unknown[]) => void;
nodeSendToSession: (...args: unknown[]) => void;
logGateway: { warn: (...args: unknown[]) => void };
};
export type ChatAbortRespondMock = Mock<RespondFn>;
export function createChatAbortContext(
overrides: Record<string, unknown> = {},
): ChatAbortTestContext {
return {
chatAbortControllers: new Map(),
chatRunBuffers: new Map(),
@ -39,7 +56,7 @@ export function createChatAbortContext(overrides: Record<string, unknown> = {})
export async function invokeChatAbortHandler(params: {
handler: GatewayRequestHandler;
context: ReturnType<typeof createChatAbortContext>;
context: ChatAbortTestContext;
request: { sessionKey: string; runId?: string };
client?: {
connId?: string;
@ -48,8 +65,8 @@ export async function invokeChatAbortHandler(params: {
scopes?: string[];
};
} | null;
respond?: ReturnType<typeof vi.fn>;
}) {
respond?: ChatAbortRespondMock;
}): Promise<ChatAbortRespondMock> {
const respond = params.respond ?? vi.fn();
await params.handler({
params: params.request,