messageCreate.unit.ts

import { PostgreSqlContainer, StartedPostgreSqlContainer } from "@testcontainers/postgresql";
import chai, { expect } from "chai";
import { Client, Message } from "discord.js";
import sinon, { SinonSandbox } from "sinon";
import sinonChai from "sinon-chai";
import sinonTest from "sinon-test";
import { getConnection } from "typeorm";
import { TournamentFormat, TournamentStatus } from "../../src/database/interface";
import { initializeConnection } from "../../src/database/orm";
import { DeckManager, initializeDeckManager } from "../../src/deck";
import { onDirectMessage } from "../../src/events/messageCreate";
import { ParticipantRoleProvider } from "../../src/role/participant";
import { WebsiteWrapperChallonge } from "../../src/website/challonge";
import { DatabaseWrapperMock } from "../mocks/database";
chai.use(sinonChai);
const test = sinonTest(sinon);

// This is created so we can stub out methods. Most DJS objects also need this as a constructor parameter.
const mockBotClient = new Client({ intents: [] });
// For the purposes of most commands, most fields don't matter. This is the minimum to make the constructor run.
const sampleMessage = Reflect.construct(Message, [
	mockBotClient,
	{
		id: "123456789",
		channel_id: "testChannel",
		author: { id: "testUser", username: "K", discriminator: "1234", avatar: "k.png" },
		mentions: [],
		attachments: [],
		content:
			"ydke://o6lXBaOpVwWjqVcFep21BXqdtQV6nbUF8GFdAvBhXQLwYV0CLdjxAS3Y8QEt2PEBiWdgA4lnYAOJZ2AD0hVTAtIVUwLSFVMC9slUAvbJVAL2yVQCKYF+BSmBfgUpgX4FYW7uA2Fu7gNhbu4DlDaLBJQ2iwSUNosE0GpSAtBqUgLQalICTIHIAEyByABMgcgAXu5QBV7uUAVe7lAFsdjfAQ==!yV+/A8lfvwPJX78D!sdjfAbHY3wE=!",
		timestamp: "1",
		edited_timestamp: "1",
		tts: false,
		mention_everyone: false,
		mention_roles: [],
		mention_channels: [],
		embeds: [],
		pinned: false,
		type: 0
	}
]);
const database = new DatabaseWrapperMock();
let decks: DeckManager;
const challonge = new WebsiteWrapperChallonge("", "");
const participantRole = new ParticipantRoleProvider(mockBotClient);
let container: StartedPostgreSqlContainer;

describe("Direct message submissions", function () {
	let clock: sinon.SinonFakeTimers;
	before(async () => {
		clock = sinon.useFakeTimers();
		// eslint-disable-next-line @typescript-eslint/no-non-null-assertion
		decks = await initializeDeckManager(process.env.OCTOKIT_TOKEN!);
		container = await new PostgreSqlContainer("postgres:13-alpine").start();
		await initializeConnection(
			`postgresql://${container.getUsername()}:${container.getPassword()}@${container.getHost()}:${container.getPort()}/${container.getDatabase()}`
		);
	});

	after(async () => {
		await getConnection().destroy();
		await container.stop();
		clock.restore();
	});

	// TODO: test attachment version
	it(
		"accepts registrations",
		test(async function (this: SinonSandbox) {
			this.stub(database, "getPendingTournaments").resolves([
				{
					id: "tourn1",
					name: "Tournament 1",
					description: "The first tournament",
					format: TournamentFormat.SWISS,
					status: TournamentStatus.PREPARING,
					hosts: ["host1"],
					players: [],
					limit: 0,
					publicChannels: ["channel1"],
					privateChannels: ["channel2"],
					server: "testServer",
					byes: ["player1"],
					findPlayer: () => {
						throw new Error("unused");
					}
				}
			]);
			const printResult = { embeds: [], files: [] };
			this.stub(decks, "prettyPrint").returns(printResult);
			this.stub(participantRole, "grant").resolves();
			this.stub(challonge, "registerPlayer").resolves();
			const replySpy = this.stub(sampleMessage, "reply").resolves();
			const send = this.spy();
			const fetchStub = this.stub(mockBotClient.channels, "fetch").resolves({
				isTextBased: () => true,
				send
				// eslint-disable-next-line @typescript-eslint/no-explicit-any
			} as any);
			await onDirectMessage(sampleMessage, database, decks, challonge, participantRole, mockBotClient);
			expect(fetchStub).to.have.been.calledWith("channel2");
			expect(send).to.have.been.calledWith(
				"<@testUser> (K#1234) has signed up for **Tournament 1** with the following deck!"
			);
			expect(send).to.have.been.calledWith(printResult);
			expect(replySpy).to.have.been.calledWith(
				"You have successfully signed up for **Tournament 1**! Your deck is below to double-check. You may resubmit at any time before the tournament starts."
			);
			expect(replySpy).to.have.been.calledWith(printResult);
		})
	);
	it(
		"stops ambiguous registrations",
		test(async function (this: SinonSandbox) {
			this.stub(database, "getPendingTournaments").resolves([
				{ name: "Tournament 1" },
				{ name: "Tournament 2" }
			] as never);
			const replySpy = this.stub(sampleMessage, "reply").resolves();
			await onDirectMessage(sampleMessage, database, decks, challonge, participantRole, mockBotClient);
			expect(replySpy).to.have.been.calledWith(
				"You are registering in multiple tournaments. Please register in one at a time by unchecking the reaction on all others.\nTournament 1, Tournament 2"
			);
		})
	);
	it(
		"sends a help message with no tournaments",
		test(async function (this: SinonSandbox) {
			this.stub(database, "getPendingTournaments").resolves([]);
			this.stub(database, "getConfirmedTournaments").resolves([]);
			const replySpy = this.stub(sampleMessage, "reply").resolves();
			await onDirectMessage(sampleMessage, database, decks, challonge, participantRole, mockBotClient);
			expect(replySpy).to.have.been.calledWith(
				"Emcee's documentation can be found at https://github.com/DawnbrandBots/emcee-tournament-bot/blob/master/README.md.\nRevision: **undefined**\nIf you're trying to sign up for a tournament, make sure you've clicked ✅ on a sign-up message and I'll let you know how to proceed."
			);
		})
	);
	it(
		"rejects illegal decks",
		test(async function (this: SinonSandbox) {
			const content = sampleMessage.content;
			sampleMessage.content = "ydke://!!!";
			const replySpy = this.stub(sampleMessage, "reply").resolves();
			await onDirectMessage(sampleMessage, database, decks, challonge, participantRole, mockBotClient);
			sampleMessage.content = content;
			const reply = replySpy.args[0][0];
			expect(reply).to.have.property("embeds");
			if (typeof reply === "object" && "embeds" in reply) {
				expect(reply.embeds?.[0]?.data?.fields?.[1]?.value).to.equal(
					"Main Deck too small! Should be at least 40, is 0!"
				);
			}
		})
	);
	it(
		"stops ambiguous deck updates",
		test(async function (this: SinonSandbox) {
			this.stub(database, "getPendingTournaments").resolves([]);
			this.stub(database, "getConfirmedTournaments").resolves([
				{ name: "Tournament 1" },
				{ name: "Tournament 3" }
			] as never);
			const replySpy = this.stub(sampleMessage, "reply").resolves();
			await onDirectMessage(sampleMessage, database, decks, challonge, participantRole, mockBotClient);
			expect(replySpy).to.have.been.calledWith(
				"You're trying to update your deck for a tournament, but you're in multiple! Please choose one by dropping and registering again.\nTournament 1, Tournament 3"
			);
		})
	);
	it(
		"accepts deck updates",
		test(async function (this: SinonSandbox) {
			this.stub(database, "getPendingTournaments").resolves([]);
			this.stub(database, "getConfirmedTournaments").resolves([
				{
					id: "tourn1",
					name: "Tournament 1",
					description: "The first tournament",
					format: TournamentFormat.SWISS,
					status: TournamentStatus.PREPARING,
					hosts: ["host1"],
					players: [],
					limit: 0,
					publicChannels: ["channel1"],
					privateChannels: ["channel2"],
					server: "testServer",
					byes: ["player1"],
					findPlayer: () => {
						throw new Error("unused");
					}
				}
			]);
			const printResult = { embeds: [], files: [] };
			this.stub(decks, "prettyPrint").returns(printResult);
			this.stub(database, "updateDeck").resolves();
			const replySpy = this.stub(sampleMessage, "reply").resolves();
			const send = this.spy();
			const fetchStub = this.stub(mockBotClient.channels, "fetch").resolves({
				isTextBased: () => true,
				send
				// eslint-disable-next-line @typescript-eslint/no-explicit-any
			} as any);
			await onDirectMessage(sampleMessage, database, decks, challonge, participantRole, mockBotClient);
			expect(fetchStub).to.have.been.calledWith("channel2");
			expect(send).to.have.been.calledWith(
				"<@testUser> (K#1234) has updated their deck for **Tournament 1** to the following!"
			);
			expect(send).to.have.been.calledWith(printResult);
			expect(replySpy).to.have.been.calledWith(
				"You have successfully changed your deck for **Tournament 1**! Your deck is below to double-check. You may resubmit at any time before the tournament starts."
			);
			expect(replySpy).to.have.been.calledWith(printResult);
		})
	);
	it(
		"rejects illegal deck updates",
		test(async function (this: SinonSandbox) {
			this.stub(database, "getPendingTournaments").resolves([]);
			this.stub(database, "getConfirmedTournaments").resolves([
				{
					id: "tourn1",
					name: "Tournament 1",
					description: "The first tournament",
					format: TournamentFormat.SWISS,
					status: TournamentStatus.PREPARING,
					hosts: ["host1"],
					players: [],
					limit: 0,
					publicChannels: ["channel1"],
					privateChannels: ["channel2"],
					server: "testServer",
					byes: ["player1"],
					findPlayer: () => {
						throw new Error("unused");
					}
				}
			]);
			const content = sampleMessage.content;
			sampleMessage.content = "ydke://!!!";
			const replySpy = this.stub(sampleMessage, "reply").resolves();
			await onDirectMessage(sampleMessage, database, decks, challonge, participantRole, mockBotClient);
			sampleMessage.content = content;
			const reply = replySpy.args[0][0];
			expect(reply).to.have.property("embeds");
			if (typeof reply === "object" && "embeds" in reply) {
				expect(reply.embeds?.[0]?.data?.fields?.[1]?.value).to.equal(
					"Main Deck too small! Should be at least 40, is 0!"
				);
			}
		})
	);
});