tests/mocks/client.mock.ts

import { ClientUser } from "discord.js";
import { vi } from "vitest";
import Client from "../../src/core/Client";
import { randomSnowflake } from "./snowflakes";

vi.mock("@prisma/client", () => {
    return {
        __esModule: true,
        PrismaClient: vi.fn().mockImplementation(() => {
            return {
                user: {
                    findUnique: vi.fn(),
                    create: vi.fn(),
                    findMany: vi.fn(),
                    update: vi.fn(),
                    delete: vi.fn()
                },
                troll: {
                    findUnique: vi.fn(),
                    create: vi.fn(),
                    findMany: vi.fn(),
                    update: vi.fn(),
                    delete: vi.fn()
                }
            };
        })
    };
});

export function createClient() {
    const client = new Client({
        intents: []
    });

    client.user = {
        id: randomSnowflake(),
        username: "SudoBot",
        discriminator: "0000",
        tag: "SudoBot#0000",
        avatar: "avatar",
        bot: true,
        system: false,
        client,
        toString() {
            return `<@${this.id}>`;
        }
    } as unknown as ClientUser;

    return client as Client<true>;
}
1	rakinar2	577	import { ClientUser } from "discord.js";
2			import { vi } from "vitest";
3			import Client from "../../src/core/Client";
4			import { randomSnowflake } from "./snowflakes";
5
6			vi.mock("@prisma/client", () => {
7			return {
8			__esModule: true,
9			PrismaClient: vi.fn().mockImplementation(() => {
10			return {
11			user: {
12			findUnique: vi.fn(),
13			create: vi.fn(),
14			findMany: vi.fn(),
15			update: vi.fn(),
16			delete: vi.fn()
17			},
18			troll: {
19			findUnique: vi.fn(),
20			create: vi.fn(),
21			findMany: vi.fn(),
22			update: vi.fn(),
23			delete: vi.fn()
24			}
25			};
26			})
27			};
28			});
29
30			export function createClient() {
31			const client = new Client({
32			intents: []
33			});
34
35			client.user = {
36			id: randomSnowflake(),
37			username: "SudoBot",
38			discriminator: "0000",
39			tag: "SudoBot#0000",
40			avatar: "avatar",
41			bot: true,
42			system: false,
43			client,
44			toString() {
45			return `<@${this.id}>`;
46			}
47			} as unknown as ClientUser;
48
49			return client as Client<true>;
50			}