refactor(db): migrate staff and app actions from Prisma facade to Drizzle

Co-authored-by: Cursor <[email protected]>
This commit is contained in:
SimoandCursor committed 2026-07-31 21:35:05 +02:00
1 parent 9854719cfd
commit ed9c23c702
85 files changed
+2612 -1832

No files matched your search

+56 -35
View File
@@ -2,23 +2,41 @@
import { revalidatePath } from "next/cache";
import { beforeEach, describe, expect, it, vi } from "vitest";
import { requirePermissionRateLimited } from "@/lib/admin/guard";
import { prisma } from "@/lib/prisma";
import { logStaffActivity } from "@/lib/services/staff-activity";
import { disbandGuild } from "./admin-guilds";
const { selectLimit, transactionFn, deleteWhere, updateSet } = vi.hoisted(
() => {
const selectLimit = vi.fn();
const transactionFn = vi.fn();
const deleteWhere = vi.fn().mockResolvedValue([{ affectedRows: 1 }]);
const updateSet = vi.fn(() => ({ where: vi.fn().mockResolvedValue([]) }));
return { selectLimit, transactionFn, deleteWhere, updateSet };
},
);
vi.mock("@/lib/admin/guard", () => ({ requirePermissionRateLimited: vi.fn() }));
vi.mock("@/lib/permissions", () => ({ PERMS: { USERS_EDIT: "users.edit" } }));
vi.mock("@/lib/prisma", () => ({
prisma: {
guilds: { findUnique: vi.fn(), delete: vi.fn() },
guildsForumsThreads: { findMany: vi.fn(), deleteMany: vi.fn() },
guildsForumsComments: { deleteMany: vi.fn() },
guildForumViews: { deleteMany: vi.fn() },
guildsMembers: { deleteMany: vi.fn() },
rooms: { updateMany: vi.fn() },
items: { updateMany: vi.fn() },
$transaction: vi.fn(),
vi.mock("@/lib/db", () => ({
db: {
select: vi.fn(() => ({
from: vi.fn(() => ({
where: vi.fn(() => ({
limit: selectLimit,
})),
})),
})),
transaction: transactionFn,
delete: vi.fn(() => ({ where: deleteWhere })),
update: vi.fn(() => ({ set: updateSet })),
},
Guilds: { id: "id", name: "name", userId: "userId" },
GuildsForumsThreads: { id: "id", guildId: "guildId" },
GuildsForumsComments: { threadId: "threadId" },
GuildForumViews: { guildId: "guildId" },
GuildsMembers: { guildId: "guildId" },
Rooms: { guildId: "guildId" },
Items: { guildId: "guildId" },
}));
vi.mock("@/lib/services/staff-activity", () => ({ logStaffActivity: vi.fn() }));
vi.mock("next/cache", () => ({ revalidatePath: vi.fn() }));
@@ -35,29 +53,32 @@ beforeEach(() => {
describe("disbandGuild", () => {
it("disbands guild and cleans related data", async () => {
vi.mocked(prisma.guilds.findUnique).mockResolvedValue({
id: 1,
name: "TestGuild",
userId: 42,
} as never);
vi.mocked(prisma.guildsForumsThreads.findMany).mockResolvedValue([
{ id: 10 },
] as never);
vi.mocked(prisma.$transaction).mockImplementation(async (fn: unknown) => {
const tx = {
guildsForumsComments: { deleteMany: vi.fn().mockResolvedValue({}) },
guildsForumsThreads: {
findMany: vi.fn().mockResolvedValue([{ id: 10 }]),
deleteMany: vi.fn().mockResolvedValue({}),
},
guildForumViews: { deleteMany: vi.fn().mockResolvedValue({}) },
guildsMembers: { deleteMany: vi.fn().mockResolvedValue({}) },
rooms: { updateMany: vi.fn().mockResolvedValue({}) },
items: { updateMany: vi.fn().mockResolvedValue({}) },
guilds: { delete: vi.fn().mockResolvedValue({}) },
} as never;
await (fn as (tx: never) => Promise<void>)(tx);
});
selectLimit.mockResolvedValue([{ id: 1, name: "TestGuild", userId: 42 }]);
transactionFn.mockImplementation(
async (fn: (tx: unknown) => Promise<void>) => {
const txSelectLimit = vi.fn().mockResolvedValue([{ id: 10 }]);
const tx = {
select: vi.fn(() => ({
from: vi.fn(() => ({
where: vi.fn(() => ({
limit: txSelectLimit,
})),
})),
})),
delete: vi.fn(() => ({ where: vi.fn().mockResolvedValue([]) })),
update: vi.fn(() => ({
set: vi.fn(() => ({ where: vi.fn().mockResolvedValue([]) })),
})),
};
// For threads findMany (no limit) — make where resolve to array
tx.select = vi.fn(() => ({
from: vi.fn(() => ({
where: vi.fn().mockResolvedValue([{ id: 10 }]),
})),
}));
await fn(tx);
},
);
await disbandGuild(fakeForm({ id: "1" }) as unknown as FormData);
expect(logStaffActivity).toHaveBeenCalled();
expect(revalidatePath).toHaveBeenCalledWith("/admin/guilds");
@@ -65,6 +86,6 @@ describe("disbandGuild", () => {
it("returns early when id is not positive", async () => {
await disbandGuild(fakeForm({ id: "0" }) as unknown as FormData);
expect(prisma.guilds.findUnique).not.toHaveBeenCalled();
expect(selectLimit).not.toHaveBeenCalled();
});
});