test: align remaining action tests with Drizzle mocks

EOF

Co-authored-by: Cursor <[email protected]>

src/actions/auth-precheck.test.ts

@@ -1,15 +1,36 @@
 // @ts-nocheck
 import { beforeEach, describe, expect, it, vi } from "vitest";
 import { checkLogin } from "@/lib/auth/password";
-import { prisma } from "@/lib/prisma";
 import { clientIp, rateLimit } from "@/lib/rate-limit";
 import { captchaConfig, verifyCaptcha } from "@/lib/services/captcha";
 import { siteSettings } from "@/lib/services/site-settings";
 import { precheckLogin } from "./auth-precheck";
 
+const { selectLimit } = vi.hoisted(() => {
+	const selectLimit = vi.fn().mockResolvedValue([]);
+	return { selectLimit };
+});
+
 vi.mock("@/env", () => ({ env: { CONVERT_PASSWORDS: false } }));
 vi.mock("@/lib/auth/password", () => ({ checkLogin: vi.fn() }));
-vi.mock("@/lib/prisma", () => ({ prisma: { user: { findUnique: vi.fn() } } }));
+vi.mock("@/lib/db", () => ({
+	db: {
+		select: vi.fn(() => ({
+			from: vi.fn(() => ({
+				where: vi.fn(() => ({
+					limit: selectLimit,
+				})),
+			})),
+		})),
+	},
+	User: {
+		password: "password",
+		twoFactorConfirmedAt: "twoFactorConfirmedAt",
+		mail: "mail",
+		mailVerified: "mailVerified",
+		username: "username",
+	},
+}));
 vi.mock("@/lib/rate-limit", () => ({ clientIp: vi.fn(), rateLimit: vi.fn() }));
 vi.mock("@/lib/services/captcha", () => ({
 	captchaConfig: vi.fn(),
@@ -25,26 +46,31 @@ beforeEach(() => {
 	vi.mocked(rateLimit).mockResolvedValue({ ok: true });
 	vi.mocked(checkLogin).mockResolvedValue({ valid: true } as never);
 	vi.mocked(captchaConfig).mockResolvedValue({ provider: "none" } as never);
+	selectLimit.mockResolvedValue([]);
 });
 
 describe("precheckLogin", () => {
 	it("returns ok for valid login without 2FA", async () => {
-		vi.mocked(prisma.user.findUnique).mockResolvedValue({
-			password: "hash",
-			twoFactorConfirmedAt: null,
-			mail: null,
-			mailVerified: "0",
-		} as never);
+		selectLimit.mockResolvedValue([
+			{
+				password: "hash",
+				twoFactorConfirmedAt: null,
+				mail: null,
+				mailVerified: "0",
+			},
+		]);
 		expect(await precheckLogin("user", "pass")).toBe("ok");
 	});
 
 	it("returns twofactor when 2FA is set up", async () => {
-		vi.mocked(prisma.user.findUnique).mockResolvedValue({
-			password: "hash",
-			twoFactorConfirmedAt: new Date(),
-			mail: null,
-			mailVerified: "0",
-		} as never);
+		selectLimit.mockResolvedValue([
+			{
+				password: "hash",
+				twoFactorConfirmedAt: new Date(),
+				mail: null,
+				mailVerified: "0",
+			},
+		]);
 		expect(await precheckLogin("user", "pass")).toBe("twofactor");
 	});
 
@@ -57,29 +83,33 @@ describe("precheckLogin", () => {
 			provider: "hcaptcha",
 		} as never);
 		vi.mocked(verifyCaptcha).mockResolvedValue(false);
-		vi.mocked(prisma.user.findUnique).mockResolvedValue({
-			password: "hash",
-			twoFactorConfirmedAt: null,
-			mail: null,
-			mailVerified: "0",
-		} as never);
+		selectLimit.mockResolvedValue([
+			{
+				password: "hash",
+				twoFactorConfirmedAt: null,
+				mail: null,
+				mailVerified: "0",
+			},
+		]);
 		expect(await precheckLogin("user", "pass", "bad-token")).toBe("captcha");
 	});
 
 	it("returns invalid when user not found (dummy hash check)", async () => {
-		vi.mocked(prisma.user.findUnique).mockResolvedValue(null);
+		selectLimit.mockResolvedValue([]);
 		const result = await precheckLogin("nonexistent", "pass");
 		expect(result).toBe("invalid");
 		expect(checkLogin).toHaveBeenCalled();
 	});
 
 	it("returns unverified when email verification required", async () => {
-		vi.mocked(prisma.user.findUnique).mockResolvedValue({
-			password: "hash",
-			twoFactorConfirmedAt: null,
-			mail: "[email protected]",
-			mailVerified: "0",
-		} as never);
+		selectLimit.mockResolvedValue([
+			{
+				password: "hash",
+				twoFactorConfirmedAt: null,
+				mail: "[email protected]",
+				mailVerified: "0",
+			},
+		]);
 		vi.mocked(siteSettings.getBool).mockResolvedValue(true);
 		expect(await precheckLogin("user", "pass")).toBe("unverified");
 	});