refactor(test): share signal tool result test setup

This commit is contained in:
Peter Steinberger
2026-02-14 20:48:54 +00:00
parent 696a358215
commit 20cefd78cb
3 changed files with 148 additions and 168 deletions

View File

@@ -0,0 +1,103 @@
import { beforeEach, vi } from "vitest";
import { resetInboundDedupe } from "../auto-reply/reply/inbound-dedupe.js";
import { resetSystemEventsForTest } from "../infra/system-events.js";
const waitForTransportReadyMock = vi.hoisted(() => vi.fn());
const sendMock = vi.hoisted(() => vi.fn());
const replyMock = vi.hoisted(() => vi.fn());
const updateLastRouteMock = vi.hoisted(() => vi.fn());
const readAllowFromStoreMock = vi.hoisted(() => vi.fn());
const upsertPairingRequestMock = vi.hoisted(() => vi.fn());
const streamMock = vi.hoisted(() => vi.fn());
const signalCheckMock = vi.hoisted(() => vi.fn());
const signalRpcRequestMock = vi.hoisted(() => vi.fn());
export function getSignalToolResultTestMocks() {
return {
waitForTransportReadyMock,
sendMock,
replyMock,
updateLastRouteMock,
readAllowFromStoreMock,
upsertPairingRequestMock,
streamMock,
signalCheckMock,
signalRpcRequestMock,
};
}
export let config: Record<string, unknown> = {};
export function setSignalToolResultTestConfig(next: Record<string, unknown>) {
config = next;
}
export const flush = () => new Promise((resolve) => setTimeout(resolve, 0));
vi.mock("../config/config.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("../config/config.js")>();
return {
...actual,
loadConfig: () => config,
};
});
vi.mock("../auto-reply/reply.js", () => ({
getReplyFromConfig: (...args: unknown[]) => replyMock(...args),
}));
vi.mock("./send.js", () => ({
sendMessageSignal: (...args: unknown[]) => sendMock(...args),
sendTypingSignal: vi.fn().mockResolvedValue(true),
sendReadReceiptSignal: vi.fn().mockResolvedValue(true),
}));
vi.mock("../pairing/pairing-store.js", () => ({
readChannelAllowFromStore: (...args: unknown[]) => readAllowFromStoreMock(...args),
upsertChannelPairingRequest: (...args: unknown[]) => upsertPairingRequestMock(...args),
}));
vi.mock("../config/sessions.js", () => ({
resolveStorePath: vi.fn(() => "/tmp/openclaw-sessions.json"),
updateLastRoute: (...args: unknown[]) => updateLastRouteMock(...args),
readSessionUpdatedAt: vi.fn(() => undefined),
recordSessionMetaFromInbound: vi.fn().mockResolvedValue(undefined),
}));
vi.mock("./client.js", () => ({
streamSignalEvents: (...args: unknown[]) => streamMock(...args),
signalCheck: (...args: unknown[]) => signalCheckMock(...args),
signalRpcRequest: (...args: unknown[]) => signalRpcRequestMock(...args),
}));
vi.mock("./daemon.js", () => ({
spawnSignalDaemon: vi.fn(() => ({ stop: vi.fn() })),
}));
vi.mock("../infra/transport-ready.js", () => ({
waitForTransportReady: (...args: unknown[]) => waitForTransportReadyMock(...args),
}));
export function installSignalToolResultTestHooks() {
beforeEach(() => {
resetInboundDedupe();
config = {
messages: { responsePrefix: "PFX" },
channels: {
signal: { autoStart: false, dmPolicy: "open", allowFrom: ["*"] },
},
};
sendMock.mockReset().mockResolvedValue(undefined);
replyMock.mockReset();
updateLastRouteMock.mockReset();
streamMock.mockReset();
signalCheckMock.mockReset().mockResolvedValue({});
signalRpcRequestMock.mockReset().mockResolvedValue({});
readAllowFromStoreMock.mockReset().mockResolvedValue([]);
upsertPairingRequestMock.mockReset().mockResolvedValue({ code: "PAIRCODE", created: true });
waitForTransportReadyMock.mockReset().mockResolvedValue(undefined);
resetSystemEventsForTest();
});
}