WIP: Start of poll command tests
All checks were successful
Test / build (push) Successful in 16s

This commit is contained in:
Ethan Lane 2024-05-13 16:23:51 +01:00
parent 91bc444e71
commit 39231ddc16
2 changed files with 121 additions and 18 deletions

View file

@ -40,15 +40,13 @@ export default class Poll extends Command {
} }
public override async execute(interaction: CommandInteraction) { public override async execute(interaction: CommandInteraction) {
const title = interaction.options.get('title'); const title = interaction.options.get('title', true);
const option1 = interaction.options.get('option1'); const option1 = interaction.options.get('option1', true);
const option2 = interaction.options.get('option2'); const option2 = interaction.options.get('option2', true);
const option3 = interaction.options.get('option3'); const option3 = interaction.options.get('option3');
const option4 = interaction.options.get('option4'); const option4 = interaction.options.get('option4');
const option5 = interaction.options.get('option5'); const option5 = interaction.options.get('option5');
if (!title || !option1 || !option2) return;
const description = [ const description = [
option1.value as string, option1.value as string,
option2.value as string, option2.value as string,
@ -58,15 +56,7 @@ export default class Poll extends Command {
] ]
.filter(x => x != null); .filter(x => x != null);
const arrayOfNumbers = [ const reactionEmojis = ["1⃣", "2⃣", "3⃣", "4⃣", "5⃣"];
':one:',
':two:',
':three:',
':four:',
':five:',
];
const reactionEmojis = ["1⃣", "2⃣", "3⃣", "4⃣", "5⃣", "6⃣", "7⃣", "8⃣", "9⃣"];
description.forEach((value, index) => { description.forEach((value, index) => {
description[index] = `${reactionEmojis[index]} ${description[index]}`; description[index] = `${reactionEmojis[index]} ${description[index]}`;
@ -82,10 +72,11 @@ export default class Poll extends Command {
}); });
const message = await interaction.reply({ embeds: [ embed ]}); const messageResponse = await interaction.reply({ embeds: [ embed ]});
const message = await messageResponse.fetch();
description.forEach(async (value, index) => { description.forEach(async (value, index) => {
await (await message.fetch()).react(reactionEmojis[index]); await message.react(reactionEmojis[index]);
}); });
} }
} }

View file

@ -1,9 +1,121 @@
import { CommandInteraction, EmbedBuilder, InteractionResponse, Message, SlashCommandBuilder, SlashCommandStringOption } from "discord.js";
import Command from "../../src/commands/poll";
import EmbedColours from "../../src/constants/EmbedColours";
describe('Constructor', () => { describe('Constructor', () => {
test.todo('EXPECT properties to be set'); test('EXPECT properties to be set', () => {
const command = new Command();
expect(command.CommandBuilder).toBeDefined();
const commandBuilder = command.CommandBuilder as SlashCommandBuilder;
expect(commandBuilder.name).toBe("poll");
expect(commandBuilder.description).toBe("Run a poll, automatically adding reaction emojis as options");
expect(commandBuilder.options.length).toBe(6);
const commandBuilderTitleOption = commandBuilder.options[0] as SlashCommandStringOption;
expect(commandBuilderTitleOption.name).toBe("title");
expect(commandBuilderTitleOption.description).toBe("Title of the poll");
expect(commandBuilderTitleOption.required).toBe(true);
const commandBuilderOption1Option = commandBuilder.options[1] as SlashCommandStringOption;
expect(commandBuilderOption1Option.name).toBe("option1");
expect(commandBuilderOption1Option.description).toBe("Option 1");
expect(commandBuilderOption1Option.required).toBe(true);
const commandBuilderOption2Option = commandBuilder.options[2] as SlashCommandStringOption;
expect(commandBuilderOption2Option.name).toBe("option2");
expect(commandBuilderOption2Option.description).toBe("Option 2");
expect(commandBuilderOption2Option.required).toBe(true);
const commandBuilderOption3Option = commandBuilder.options[3] as SlashCommandStringOption;
expect(commandBuilderOption3Option.name).toBe("option3");
expect(commandBuilderOption3Option.description).toBe("Option 3");
const commandBuilderOption4Option = commandBuilder.options[4] as SlashCommandStringOption;
expect(commandBuilderOption4Option.name).toBe("option4");
expect(commandBuilderOption4Option.description).toBe("Option 4");
const commandBuilderOption5Option = commandBuilder.options[5] as SlashCommandStringOption;
expect(commandBuilderOption5Option.name).toBe("option5");
expect(commandBuilderOption5Option.description).toBe("Option 5");
});
}); });
describe('Execute', () => { describe('Execute', () => {
test.todo("EXPECT a poll to be created"); test("EXPECT a poll to be created", async () => {
let sentEmbed: EmbedBuilder | undefined;
// Arrange
const message = {
react: jest.fn(),
} as unknown as Message<boolean>;
const response = {
fetch: jest.fn().mockResolvedValue(message),
} as unknown as InteractionResponse<boolean>;
const interaction = {
options: {
get: jest.fn().mockReturnValueOnce({ value: "Title" })
.mockReturnValueOnce({ value: "Option 1" })
.mockReturnValueOnce({ value: "Option 2" })
.mockReturnValueOnce({ value: "Option 3" })
.mockReturnValueOnce({ value: "Option 4" })
.mockReturnValue({ value: "Option 5" }),
},
reply: jest.fn().mockImplementation((options: any) => {
sentEmbed = options.embeds[0];
return response;
}),
user: {
username: "username",
avatarURL: jest.fn().mockReturnValue("https://avatarurl.com/user.png"),
},
} as unknown as CommandInteraction;
// Act
const command = new Command();
await command.execute(interaction);
// Assert
expect(interaction.options.get).toHaveBeenCalledTimes(6);
expect(interaction.options.get).toHaveBeenCalledWith("title", true);
expect(interaction.options.get).toHaveBeenCalledWith("option1", true);
expect(interaction.options.get).toHaveBeenCalledWith("option2", true);
expect(interaction.options.get).toHaveBeenCalledWith("option3");
expect(interaction.options.get).toHaveBeenCalledWith("option4");
expect(interaction.options.get).toHaveBeenCalledWith("option5");
expect(interaction.reply).toHaveBeenCalledTimes(1);
expect(sentEmbed).toBeDefined();
expect(sentEmbed!.data.color).toBe(EmbedColours.Ok);
expect(sentEmbed!.data.title).toBe("Title");
expect(sentEmbed!.data.description).toBe("1⃣ Option 1\n2⃣ Option 2\n3⃣ Option 3\n4⃣ Option 4\n5⃣ Option 5");
expect(sentEmbed!.data.footer).toBeDefined();
expect(sentEmbed!.data.footer!.text).toBe("Poll by username");
expect(sentEmbed!.data.footer!.icon_url).toBe("https://avatarurl.com/user.png");
expect(interaction.user.avatarURL).toHaveBeenCalledTimes(1);
expect(response.fetch).toHaveBeenCalledTimes(1);
expect(message.react).toHaveBeenCalledTimes(5);
expect(message.react).toHaveBeenCalledWith("1⃣");
expect(message.react).toHaveBeenCalledWith("2⃣");
expect(message.react).toHaveBeenCalledWith("3⃣");
expect(message.react).toHaveBeenCalledWith("4⃣");
expect(message.react).toHaveBeenCalledWith("5⃣");
});
test.todo("GIVEN title is not supplied, EXPECT nothing to happen"); test.todo("GIVEN title is not supplied, EXPECT nothing to happen");