refactor(db): migrate app pages and APIs from Prisma facade to Drizzle (5)

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

src/lib/auth/jwt-version-cache.test.ts

@@ -1,14 +1,16 @@
 import { beforeEach, describe, expect, it, vi } from "vitest";
 
-const mockFindUnique = vi.hoisted(() => vi.fn());
+const mockLimit = vi.hoisted(() => vi.fn());
+const mockWhere = vi.hoisted(() => vi.fn(() => ({ limit: mockLimit })));
+const mockFrom = vi.hoisted(() => vi.fn(() => ({ where: mockWhere })));
+const mockSelect = vi.hoisted(() => vi.fn(() => ({ from: mockFrom })));
 const mockRedisGet = vi.hoisted(() => vi.fn());
 const mockRedisSetex = vi.hoisted(() => vi.fn());
 const mockRedisDel = vi.hoisted(() => vi.fn());
 
-vi.mock("@/lib/prisma", () => ({
-	prisma: {
-		user: { findUnique: mockFindUnique },
-	},
+vi.mock("@/lib/db", () => ({
+	db: { select: mockSelect },
+	User: { id: "id", websiteJwtVersion: "websiteJwtVersion" },
 }));
 
 vi.mock("@/lib/redis", () => ({
@@ -23,29 +25,32 @@ describe("jwt-version-cache", () => {
 	beforeEach(() => {
 		vi.resetModules();
 		vi.clearAllMocks();
+		mockWhere.mockReturnValue({ limit: mockLimit });
+		mockFrom.mockReturnValue({ where: mockWhere });
+		mockSelect.mockReturnValue({ from: mockFrom });
 		mockRedisGet.mockResolvedValue(null);
 		mockRedisSetex.mockResolvedValue("OK");
 		mockRedisDel.mockResolvedValue(1);
 	});
 
 	it("returns DB version and caches it", async () => {
-		mockFindUnique.mockResolvedValue({ websiteJwtVersion: 3 });
+		mockLimit.mockResolvedValue([{ websiteJwtVersion: 3 }]);
 		const { getCachedJwtVersion } = await import("./jwt-version-cache");
 		await expect(getCachedJwtVersion(42)).resolves.toBe(3);
-		expect(mockFindUnique).toHaveBeenCalledTimes(1);
+		expect(mockSelect).toHaveBeenCalledTimes(1);
 		await expect(getCachedJwtVersion(42)).resolves.toBe(3);
-		expect(mockFindUnique).toHaveBeenCalledTimes(1);
+		expect(mockSelect).toHaveBeenCalledTimes(1);
 	});
 
 	it("invalidates memory and redis entries", async () => {
-		mockFindUnique.mockResolvedValue({ websiteJwtVersion: 1 });
+		mockLimit.mockResolvedValue([{ websiteJwtVersion: 1 }]);
 		const { getCachedJwtVersion, invalidateJwtVersionCache } = await import(
 			"./jwt-version-cache"
 		);
 		await getCachedJwtVersion(7);
 		await invalidateJwtVersionCache(7);
 		expect(mockRedisDel).toHaveBeenCalled();
-		mockFindUnique.mockResolvedValue({ websiteJwtVersion: 2 });
+		mockLimit.mockResolvedValue([{ websiteJwtVersion: 2 }]);
 		await expect(getCachedJwtVersion(7)).resolves.toBe(2);
 	});
 });