WIP: Start of poll command tests

src/commands/poll.ts

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

tests/commands/poll.test.ts

@@ -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', () => {
-    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', () => {
-    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");