chore: commit pending cli/web test tweaks

src/web/login.test.ts

@@ -2,20 +2,29 @@ import { EventEmitter } from "node:events";
 
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 
-import {
-  baileys,
-  resetBaileysMocks,
-  resetLoadConfigMock,
-} from "./test-helpers.js";
 import { resetLogger, setLoggerOverride } from "../logging.js";
+
+vi.mock("./session.js", () => {
+  const ev = new EventEmitter();
+  const sock = {
+    ev,
+    ws: { close: vi.fn() },
+    sendPresenceUpdate: vi.fn(),
+    sendMessage: vi.fn(),
+  };
+  return {
+    createWaSocket: vi.fn().mockResolvedValue(sock),
+    waitForWaConnection: vi.fn().mockResolvedValue(undefined),
+  };
+});
+
 import { loginWeb } from "./login.js";
 import type { waitForWaConnection } from "./session.js";
+const { createWaSocket } = await import("./session.js");
 
 describe("web login", () => {
   beforeEach(() => {
     vi.clearAllMocks();
-    resetBaileysMocks();
-    resetLoadConfigMock();
   });
 
   afterEach(() => {
@@ -24,19 +33,12 @@ describe("web login", () => {
   });
 
   it("loginWeb waits for connection and closes", async () => {
-    const closeSpy = vi.fn();
-    const ev = new EventEmitter();
-    baileys.makeWASocket.mockImplementation(() => ({
-      ev,
-      ws: { close: closeSpy },
-      sendPresenceUpdate: vi.fn(),
-      sendMessage: vi.fn(),
-    }));
+    const sock = await createWaSocket();
     const waiter: typeof waitForWaConnection = vi
       .fn()
       .mockResolvedValue(undefined);
     await loginWeb(false, waiter);
     await new Promise((resolve) => setTimeout(resolve, 550));
-    expect(closeSpy).toHaveBeenCalled();
+    expect(sock.ws.close).toHaveBeenCalled();
   });
 });