import { rmSync } from "node:fs";

import fs from "node:fs/promises";

import type { RuntimeEnv } from "openclaw/plugin-sdk/runtime-env";

import { afterAll, afterEach, beforeEach, describe, expect, it, vi } from "vitest";

import { loginWeb } from "./login.js";

import { renderQrTerminal } from "./qr-terminal.js";

import { createWaSocket, formatError, waitForWaConnection } from "./session.js";

const rmMock = vi.spyOn(fs, "rm");

};

});

vi.mock("./qr-terminal.js", () => ({

renderQrTerminal: vi.fn(async (qr: string) => `terminal:${qr}\n`),

}));

const createWaSocketMock = vi.mocked(createWaSocket);

const waitForWaConnectionMock = vi.mocked(waitForWaConnection);

const formatErrorMock = vi.mocked(formatError);

const renderQrTerminalMock = vi.mocked(renderQrTerminal);

async function flushTasks() {

await Promise.resolve();

.mockRejectedValueOnce({ error: { output: { statusCode: 515 } } })

.mockResolvedValueOnce(undefined);

const runtime = { log: vi.fn(), error: vi.fn() } as never;

const runtime: RuntimeEnv = { log: vi.fn(), error: vi.fn(), exit: vi.fn() };

const pendingLogin = loginWeb(false, waitForWaConnectionMock as never, runtime);

await flushTasks();

expect(secondSock.ws.close).toHaveBeenCalled();

});

it("routes QR output through runtime for initial and restart sockets", async () => {

waitForWaConnectionMock

.mockRejectedValueOnce({ error: { output: { statusCode: 515 } } })

.mockResolvedValueOnce(undefined);

const runtime: RuntimeEnv = { log: vi.fn(), error: vi.fn(), exit: vi.fn() };

await loginWeb(false, waitForWaConnectionMock as never, runtime);

expect(createWaSocketMock).toHaveBeenCalledTimes(2);

expect(createWaSocketMock.mock.calls[0]?.[0]).toBe(false);

const initialOpts = createWaSocketMock.mock.calls[0]?.[2] as

| { onQr?: (qr: string) => void }

| undefined;

const restartOpts = createWaSocketMock.mock.calls[1]?.[2] as

| { onQr?: (qr: string) => void }

| undefined;

expect(initialOpts?.onQr).toBe(restartOpts?.onQr);

initialOpts?.onQr?.("initial-qr");

restartOpts?.onQr?.("restart-qr");

await flushTasks();

expect(runtime.log).toHaveBeenCalledWith("Scan this QR in WhatsApp (Linked Devices):");

expect(runtime.log).toHaveBeenCalledWith("terminal:initial-qr");

expect(runtime.log).toHaveBeenCalledWith("terminal:restart-qr");

expect(renderQrTerminalMock).toHaveBeenCalledWith("initial-qr", { small: true });

expect(renderQrTerminalMock).toHaveBeenCalledWith("restart-qr", { small: true });

});

it("clears creds and throws when logged out", async () => {

waitForWaConnectionMock.mockRejectedValueOnce({

output: { statusCode: 401 },