feat: first version of bot (#2)

### Explanation This should set up everything we need for our initial launch. Test coverage is at 100% to ensure nothing breaks. ### Issue _No response_ ### Attestations - [x] I have read and agree to the [Code of Conduct](https://docs.nhcarrigan.com/community/coc/) - [x] I have read and agree to the [Community Guidelines](https://docs.nhcarrigan.com/community/guide/). - [x] My contribution complies with the [Contributor Covenant](https://docs.nhcarrigan.com/dev/covenant/). ### Dependencies - [x] I have pinned the dependencies to a specific patch version. ### Style - [x] I have run the linter and resolved any errors. - [x] My pull request uses an appropriate title, matching the conventional commit standards. - [x] My scope of feat/fix/chore/etc. correctly matches the nature of changes in my pull request. ### Tests - [x] My contribution adds new code, and I have added tests to cover it. - [ ] My contribution modifies existing code, and I have updated the tests to reflect these changes. - [x] All new and existing tests pass locally with my changes. - [x] Code coverage remains at or above the configured threshold. ### Documentation Coming soon - I'm working on the infra for docs next ### Versioning Major - My pull request introduces a breaking change. Reviewed-on: https://codeberg.org/nhcarrigan/rig-task-bot/pulls/2 Co-authored-by: Naomi Carrigan <commits@nhcarrigan.com> Co-committed-by: Naomi Carrigan <commits@nhcarrigan.com>
2024-09-30 01:41:25 +00:00
parent da6fbfd45e
commit 296a50fedd
49 changed files with 4816 additions and 3 deletions
@@ -3,3 +3,132 @@
 * @license Naomi's Public License
 * @author Naomi Carrigan
 */
+
+import { PrismaClient } from "@prisma/client";
+import { Client, Events, WebhookClient } from "discord.js";
+import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
+import { intents } from "../src/config/intents.js";
+import { onInteractionCreate } from "../src/events/onInteractionCreate.js";
+import { onReady } from "../src/events/onReady.js";
+import { boot } from "../src/index.js";
+import { validateEnvironmentVariables }
+  from "../src/utils/validateEnvironmentVariables.js";
+
+vi.mock("@prisma/client");
+vi.mock("discord.js");
+vi.mock("../src/utils/sendDebugLog.js");
+vi.mock("../src/utils/validateEnvironmentVariables.js");
+vi.mock("../src/events/onReady.ts", () => {
+  return {
+    onReady: vi.fn(),
+  };
+});
+vi.mock("../src/events/onInteractionCreate.ts", () => {
+  return {
+    onInteractionCreate: vi.fn(),
+  };
+});
+
+const mockBot = {
+  database: {
+    $connect: vi.fn().mockResolvedValue(undefined),
+  } as never as PrismaClient,
+  discord: {
+    login: vi.fn().mockResolvedValue("Logged in"),
+    on:    vi.fn(),
+  } as never as Client,
+  env: {
+    discordToken: "mock-token",
+    // Add other necessary environment variables
+  },
+};
+
+describe("boot function", () => {
+  beforeEach(() => {
+    vi.resetAllMocks();
+
+    vi.mocked(Client).mockImplementation(() => {
+      return mockBot.discord;
+    });
+    vi.mocked(PrismaClient).mockImplementation(() => {
+      return mockBot.database;
+    });
+    vi.mocked(validateEnvironmentVariables).mockReturnValue(
+      mockBot.env as never,
+    );
+  });
+
+  afterEach(() => {
+    vi.restoreAllMocks();
+  });
+
+  it("should initialize the bot and connect to all platforms", async() => {
+    expect.assertions(9);
+    await boot();
+
+    expect(
+      validateEnvironmentVariables,
+      "should validate env",
+    ).toHaveBeenCalledWith();
+    expect(Client, "should construct discord bot").toHaveBeenCalledWith({
+      intents,
+    });
+    expect(PrismaClient, "should construct prisma client").
+      toHaveBeenCalledTimes(1);
+    expect(
+      mockBot.database.$connect,
+      "should connect to database",
+    ).toHaveBeenCalledTimes(1);
+    expect(mockBot.discord.on, "did not mount interaction create").
+      toHaveBeenCalledWith(
+        Events.InteractionCreate,
+        expect.any(Function),
+      );
+    const interactionCallback = mockBot.discord.on.mock.calls.find((call) => {
+      return call[0] === Events.InteractionCreate;
+    })?.[1];
+    interactionCallback({} as never);
+    expect(onInteractionCreate, "should call oninteractioncreate").
+      toHaveBeenCalledWith(mockBot, {});
+    expect(mockBot.discord.on, "did not mount ready").toHaveBeenCalledWith(
+      Events.ClientReady,
+      expect.any(Function),
+    );
+
+    const readyCallback = mockBot.discord.on.mock.calls.find((call) => {
+      return call[0] === Events.ClientReady;
+    })?.[1];
+    readyCallback();
+    expect(onReady, "should call onready").toHaveBeenCalledWith(mockBot);
+    expect(
+      mockBot.discord.login,
+      "should login to Discord",
+    ).toHaveBeenCalledWith(mockBot.env.discordToken);
+  });
+
+  it("should handle errors and send them to the debug webhook", async() => {
+    expect.assertions(1);
+    const mockError = new Error("Test error");
+    vi.mocked(validateEnvironmentVariables).mockImplementation(() => {
+      throw mockError;
+    });
+
+    const mockWebhookSend = vi.fn().mockResolvedValue(undefined);
+    vi.mocked(WebhookClient).mockImplementation(() => {
+      return {
+        send: mockWebhookSend,
+      } as unknown as WebhookClient;
+    });
+
+    process.env.DISCORD_DEBUG_WEBHOOK = "https://mock-webhook-url";
+
+    await boot();
+
+    expect(
+      mockWebhookSend,
+      "should send directly to webhook",
+    ).toHaveBeenCalledWith({
+      content: `Error: ${JSON.stringify(mockError, null, 2)}`,
+    });
+  });
+});