From be5b28cd6b473df2ab672bdf09a81ee5d224ec8d Mon Sep 17 00:00:00 2001
From: cpojer <christoph.pojer@gmail.com>
Date: Tue, 17 Feb 2026 13:40:17 +0900
Subject: [PATCH] chore: Fix types.

---
 test/mocks/baileys.ts | 33 +++++++++++++++++++++------------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/test/mocks/baileys.ts b/test/mocks/baileys.ts
index e04ef1a2d3..f3cf089488 100644
--- a/test/mocks/baileys.ts
+++ b/test/mocks/baileys.ts
@@ -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 {