github/openclaw
1
1
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-02-19 18:39:20 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore: Fix types.

Browse Source
This commit is contained in:
cpojer
2026-02-17 13:40:17 +09:00
parent d0cb8c19b2
commit be5b28cd6b
1 changed files with 21 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
test/mocks/baileys.ts
View File

@@ -1,6 +1,13 @@
import { EventEmitter } from "node:events";
import { vi } from "vitest";
type BaileysExports = typeof import("@whiskeysockets/baileys");
type FetchLatestBaileysVersionFn = BaileysExports["fetchLatestBaileysVersion"];
type MakeCacheableSignalKeyStoreFn = BaileysExports["makeCacheableSignalKeyStore"];
type MakeWASocketFn = BaileysExports["makeWASocket"];
type UseMultiFileAuthStateFn = BaileysExports["useMultiFileAuthState"];
type DownloadMediaMessageFn = BaileysExports["downloadMediaMessage"];
export type MockBaileysSocket = {
ev: EventEmitter;
ws: { close: ReturnType<typeof vi.fn> };
@@ -12,13 +19,13 @@ export type MockBaileysSocket = {
export type MockBaileysModule = {
DisconnectReason: { loggedOut: number };
fetchLatestBaileysVersion: ReturnType<typeof vi.fn>;
makeCacheableSignalKeyStore: ReturnType<typeof vi.fn>;
makeWASocket: ReturnType<typeof vi.fn>;
useMultiFileAuthState: ReturnType<typeof vi.fn>;
fetchLatestBaileysVersion: ReturnType<typeof vi.fn<FetchLatestBaileysVersionFn>>;
makeCacheableSignalKeyStore: ReturnType<typeof vi.fn<MakeCacheableSignalKeyStoreFn>>;
makeWASocket: ReturnType<typeof vi.fn<MakeWASocketFn>>;
useMultiFileAuthState: ReturnType<typeof vi.fn<UseMultiFileAuthStateFn>>;
jidToE164?: (jid: string) => string | null;
proto?: unknown;
downloadMediaMessage?: ReturnType<typeof vi.fn>;
downloadMediaMessage?: ReturnType<typeof vi.fn<DownloadMediaMessageFn>>;
};
export function createMockBaileys(): {
@@ -26,7 +33,7 @@ export function createMockBaileys(): {
lastSocket: () => MockBaileysSocket;
} {
const sockets: MockBaileysSocket[] = [];
const makeWASocket = vi.fn((_opts: unknown) => {
const makeWASocket = vi.fn<MakeWASocketFn>((_opts) => {
const ev = new EventEmitter();
const sock: MockBaileysSocket = {
ev,
@@ -38,20 +45,22 @@ export function createMockBaileys(): {
};
setImmediate(() => ev.emit("connection.update", { connection: "open" }));
sockets.push(sock);
return sock;
return sock as unknown as ReturnType<MakeWASocketFn>;
});
const mod: MockBaileysModule = {
DisconnectReason: { loggedOut: 401 },
fetchLatestBaileysVersion: vi.fn().mockResolvedValue({ version: [1, 2, 3] }),
makeCacheableSignalKeyStore: vi.fn((keys: unknown) => keys),
fetchLatestBaileysVersion: vi
.fn<FetchLatestBaileysVersionFn>()
.mockResolvedValue({ version: [1, 2, 3], isLatest: true }),
makeCacheableSignalKeyStore: vi.fn<MakeCacheableSignalKeyStoreFn>((keys) => keys),
makeWASocket,
useMultiFileAuthState: vi.fn(async () => ({
state: { creds: {}, keys: {} },
useMultiFileAuthState: vi.fn<UseMultiFileAuthStateFn>(async () => ({
state: { creds: {}, keys: {} } as Awaited<ReturnType<UseMultiFileAuthStateFn>>["state"],
saveCreds: vi.fn(),
})),
jidToE164: (jid: string) => jid.replace(/@.*$/, "").replace(/^/, "+"),
downloadMediaMessage: vi.fn().mockResolvedValue(Buffer.from("img")),
downloadMediaMessage: vi.fn<DownloadMediaMessageFn>().mockResolvedValue(Buffer.from("img")),
};
return {