Files
openclaw/src/web/inbound/access-control.test-harness.ts
2026-02-14 17:17:30 -05:00

48 lines
1.5 KiB
TypeScript

import { beforeEach, vi } from "vitest";
type AsyncMock<TArgs extends unknown[] = unknown[], TResult = unknown> = {
(...args: TArgs): Promise<TResult>;
mockReset: () => AsyncMock<TArgs, TResult>;
mockResolvedValue: (value: TResult) => AsyncMock<TArgs, TResult>;
mockResolvedValueOnce: (value: TResult) => AsyncMock<TArgs, TResult>;
};
export const sendMessageMock = vi.fn() as AsyncMock;
export const readAllowFromStoreMock = vi.fn() as AsyncMock;
export const upsertPairingRequestMock = vi.fn() as AsyncMock;
let config: Record<string, unknown> = {};
export function setAccessControlTestConfig(next: Record<string, unknown>): void {
config = next;
}
export function setupAccessControlTestHarness(): void {
beforeEach(() => {
config = {
channels: {
whatsapp: {
dmPolicy: "pairing",
allowFrom: [],
},
},
};
sendMessageMock.mockReset().mockResolvedValue(undefined);
readAllowFromStoreMock.mockReset().mockResolvedValue([]);
upsertPairingRequestMock.mockReset().mockResolvedValue({ code: "PAIRCODE", created: true });
});
}
vi.mock("../../config/config.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("../../config/config.js")>();
return {
...actual,
loadConfig: () => config,
};
});
vi.mock("../../pairing/pairing-store.js", () => ({
readChannelAllowFromStore: (...args: unknown[]) => readAllowFromStoreMock(...args),
upsertChannelPairingRequest: (...args: unknown[]) => upsertPairingRequestMock(...args),
}));