// @ts-nocheck import { beforeEach, describe, expect, it, vi } from "vitest"; import { requirePermission } from "@/lib/admin/guard"; import { rcon } from "@/lib/services/rcon"; import { bulkBan, bulkGiveBadge, bulkGiveCurrency, bulkUnban, } from "./bulk-users"; const { deleteWhere, insertValues, updateWhere, selectLimit, selectWhereResolved, onDuplicateKeyUpdate, } = vi.hoisted(() => { const deleteWhere = vi.fn().mockResolvedValue([{ affectedRows: 3 }]); const onDuplicateKeyUpdate = vi.fn().mockResolvedValue([{ affectedRows: 1 }]); const insertValues = vi.fn(() => ({ onDuplicateKeyUpdate, // biome-ignore lint/suspicious/noThenProperty: Drizzle query thenable mock then(resolve, reject) { return Promise.resolve([{ insertId: 1 }]).then(resolve, reject); }, })); const updateWhere = vi.fn().mockResolvedValue([{ affectedRows: 1 }]); const selectLimit = vi.fn().mockResolvedValue([]); /** Rows returned when a select chain is awaited without `.limit()`. */ const selectWhereResolved = vi.fn().mockResolvedValue([]); return { deleteWhere, insertValues, updateWhere, selectLimit, selectWhereResolved, onDuplicateKeyUpdate, }; }); vi.mock("@/lib/admin/guard", () => ({ requirePermission: vi.fn() })); vi.mock("@/lib/permissions", () => ({ PERMS: { USERS_EDIT: "users.edit" } })); vi.mock("@/lib/db", () => ({ db: { delete: vi.fn(() => ({ where: deleteWhere })), insert: vi.fn(() => ({ values: insertValues })), update: vi.fn(() => ({ set: vi.fn(() => ({ where: updateWhere })), })), select: vi.fn(() => ({ from: vi.fn(() => ({ where: vi.fn(() => ({ limit: selectLimit, // biome-ignore lint/suspicious/noThenProperty: Drizzle query thenable mock then(resolve, reject) { return selectWhereResolved().then(resolve, reject); }, })), })), })), transaction: vi.fn(), }, Ban: { userId: "userId", id: "id" }, User: { id: "id", credits: "credits", username: "username", online: "online", }, UsersCurrency: { userId: "userId", type: "type", amount: "amount" }, UsersBadges: { id: "id", userId: "userId", badgeCode: "badgeCode", slotId: "slotId", }, Sanctions: { id: "id", habboId: "habboId" }, UsersSettings: { userId: "userId", canTrade: "canTrade", tradelockAmount: "tradelockAmount", }, })); vi.mock("@/lib/services/rcon", () => ({ rcon: { giveCredits: vi.fn(), giveDuckets: vi.fn(), givePointsGotw: vi.fn(), giveBadge: vi.fn(), }, })); vi.mock("@/lib/services/staff-activity", () => ({ logStaffActivity: vi.fn() })); const staff = { id: 1, rank: 7, username: "admin" }; beforeEach(() => { vi.clearAllMocks(); vi.mocked(requirePermission).mockResolvedValue(staff as never); deleteWhere.mockResolvedValue([{ affectedRows: 3 }]); insertValues.mockImplementation(() => ({ onDuplicateKeyUpdate, // biome-ignore lint/suspicious/noThenProperty: Drizzle query thenable mock then(resolve, reject) { return Promise.resolve([{ insertId: 1 }]).then(resolve, reject); }, })); onDuplicateKeyUpdate.mockResolvedValue([{ affectedRows: 1 }]); updateWhere.mockResolvedValue([{ affectedRows: 1 }]); selectLimit.mockResolvedValue([]); selectWhereResolved.mockResolvedValue([]); }); describe("bulkUnban", () => { it("unbans users", async () => { const r = await bulkUnban({ userIds: [1, 2, 3] }); expect(r.ok).toBe(true); expect(r.data).toEqual({ unbanned: 3, total: 3 }); }); }); describe("bulkBan", () => { it("bans users", async () => { const r = await bulkBan({ userIds: [1, 2], reason: "Spam", duration: 3600, }); expect(r.ok).toBe(true); expect(r.data.banned).toBe(2); expect(insertValues).toHaveBeenCalledTimes(2); }); }); describe("bulkGiveCurrency", () => { it("gives credits", async () => { const r = await bulkGiveCurrency({ userIds: [1], amount: 100, type: "credits", }); expect(r.data.given).toBe(1); expect(rcon.giveCredits).toHaveBeenCalledWith(1, 100); expect(updateWhere).toHaveBeenCalled(); }); it("gives pixels", async () => { const r = await bulkGiveCurrency({ userIds: [2], amount: 50, type: "pixels", }); expect(r.data.given).toBe(1); expect(rcon.giveDuckets).toHaveBeenCalledWith(2, 50); expect(onDuplicateKeyUpdate).toHaveBeenCalled(); }); it("gives points", async () => { const r = await bulkGiveCurrency({ userIds: [3], amount: 25, type: "points", }); expect(r.data.given).toBe(1); expect(rcon.givePointsGotw).toHaveBeenCalledWith(3, 25); expect(onDuplicateKeyUpdate).toHaveBeenCalled(); }); }); describe("bulkGiveBadge", () => { it("gives badge to user", async () => { selectLimit.mockResolvedValueOnce([]); selectWhereResolved.mockResolvedValueOnce([{ maxSlot: 5 }]); const r = await bulkGiveBadge({ userIds: [1], badgeCode: "ADM" }); expect(r.data.given).toBe(1); expect(insertValues).toHaveBeenCalled(); expect(rcon.giveBadge).toHaveBeenCalledWith(1, "ADM"); }); });