daf036a4f6
Merged via /review-pr -> /prepare-pr -> /merge-pr. Prepared head SHA: 29fa20c7d773b2aac62dea912e00e438ce8ba9f6 Co-authored-by: hydro13 <6640526+hydro13@users.noreply.github.com> Co-authored-by: obviyus <22031114+obviyus@users.noreply.github.com> Reviewed-by: @obviyus
77 lines
3.1 KiB
TypeScript
77 lines
3.1 KiB
TypeScript
import { vi } from "vitest";
|
|
|
|
const mocks = vi.hoisted(() => ({
|
|
dispatchMock: vi.fn(),
|
|
readAllowFromStoreMock: vi.fn(),
|
|
upsertPairingRequestMock: vi.fn(),
|
|
resolveAgentRouteMock: vi.fn(),
|
|
finalizeInboundContextMock: vi.fn(),
|
|
resolveConversationLabelMock: vi.fn(),
|
|
createReplyPrefixOptionsMock: vi.fn(),
|
|
recordSessionMetaFromInboundMock: vi.fn(),
|
|
resolveStorePathMock: vi.fn(),
|
|
}));
|
|
|
|
vi.mock("../../auto-reply/reply/provider-dispatcher.js", () => ({
|
|
dispatchReplyWithDispatcher: (...args: unknown[]) => mocks.dispatchMock(...args),
|
|
}));
|
|
|
|
vi.mock("../../pairing/pairing-store.js", () => ({
|
|
readChannelAllowFromStore: (...args: unknown[]) => mocks.readAllowFromStoreMock(...args),
|
|
upsertChannelPairingRequest: (...args: unknown[]) => mocks.upsertPairingRequestMock(...args),
|
|
}));
|
|
|
|
vi.mock("../../routing/resolve-route.js", () => ({
|
|
resolveAgentRoute: (...args: unknown[]) => mocks.resolveAgentRouteMock(...args),
|
|
}));
|
|
|
|
vi.mock("../../auto-reply/reply/inbound-context.js", () => ({
|
|
finalizeInboundContext: (...args: unknown[]) => mocks.finalizeInboundContextMock(...args),
|
|
}));
|
|
|
|
vi.mock("../../channels/conversation-label.js", () => ({
|
|
resolveConversationLabel: (...args: unknown[]) => mocks.resolveConversationLabelMock(...args),
|
|
}));
|
|
|
|
vi.mock("../../channels/reply-prefix.js", () => ({
|
|
createReplyPrefixOptions: (...args: unknown[]) => mocks.createReplyPrefixOptionsMock(...args),
|
|
}));
|
|
|
|
vi.mock("../../config/sessions.js", () => ({
|
|
recordSessionMetaFromInbound: (...args: unknown[]) =>
|
|
mocks.recordSessionMetaFromInboundMock(...args),
|
|
resolveStorePath: (...args: unknown[]) => mocks.resolveStorePathMock(...args),
|
|
}));
|
|
|
|
type SlashHarnessMocks = {
|
|
dispatchMock: ReturnType<typeof vi.fn>;
|
|
readAllowFromStoreMock: ReturnType<typeof vi.fn>;
|
|
upsertPairingRequestMock: ReturnType<typeof vi.fn>;
|
|
resolveAgentRouteMock: ReturnType<typeof vi.fn>;
|
|
finalizeInboundContextMock: ReturnType<typeof vi.fn>;
|
|
resolveConversationLabelMock: ReturnType<typeof vi.fn>;
|
|
createReplyPrefixOptionsMock: ReturnType<typeof vi.fn>;
|
|
recordSessionMetaFromInboundMock: ReturnType<typeof vi.fn>;
|
|
resolveStorePathMock: ReturnType<typeof vi.fn>;
|
|
};
|
|
|
|
export function getSlackSlashMocks(): SlashHarnessMocks {
|
|
return mocks;
|
|
}
|
|
|
|
export function resetSlackSlashMocks() {
|
|
mocks.dispatchMock.mockReset().mockResolvedValue({ counts: { final: 1, tool: 0, block: 0 } });
|
|
mocks.readAllowFromStoreMock.mockReset().mockResolvedValue([]);
|
|
mocks.upsertPairingRequestMock.mockReset().mockResolvedValue({ code: "PAIRCODE", created: true });
|
|
mocks.resolveAgentRouteMock.mockReset().mockReturnValue({
|
|
agentId: "main",
|
|
sessionKey: "session:1",
|
|
accountId: "acct",
|
|
});
|
|
mocks.finalizeInboundContextMock.mockReset().mockImplementation((ctx: unknown) => ctx);
|
|
mocks.resolveConversationLabelMock.mockReset().mockReturnValue(undefined);
|
|
mocks.createReplyPrefixOptionsMock.mockReset().mockReturnValue({ onModelSelected: () => {} });
|
|
mocks.recordSessionMetaFromInboundMock.mockReset().mockResolvedValue(undefined);
|
|
mocks.resolveStorePathMock.mockReset().mockReturnValue("/tmp/openclaw-sessions.json");
|
|
}
|