From dfd709ad1c5e950098d4a491f1e316ec6fd2dc16 Mon Sep 17 00:00:00 2001
From: Rushil Perera <pererarushil1940@gmail.com>
Date: Sat, 15 Jun 2024 06:24:29 -0400
Subject: [PATCH] test: preload mocked modules

---
 bunfig.toml                                   |  2 +-
 .../episodes/getByAniListId/index.spec.ts     |  2 -
 .../episodes/getEpisodeUrl/index.spec.ts      |  2 -
 src/libs/fcm/verifyFcm.spec.ts                |  1 -
 src/mocks/consumet.ts                         | 60 +++++++++----------
 5 files changed, 30 insertions(+), 37 deletions(-)

diff --git a/bunfig.toml b/bunfig.toml
index 9fd45fb..63675c3 100644
--- a/bunfig.toml
+++ b/bunfig.toml
@@ -1,2 +1,2 @@
 [test]
-preload = ["./testSetup.ts"]
+preload = ["./testSetup.ts", "./src/mocks/consumet.ts", "./src/mocks/gToken.ts"]
diff --git a/src/controllers/episodes/getByAniListId/index.spec.ts b/src/controllers/episodes/getByAniListId/index.spec.ts
index 91146e9..6e50b23 100644
--- a/src/controllers/episodes/getByAniListId/index.spec.ts
+++ b/src/controllers/episodes/getByAniListId/index.spec.ts
@@ -2,10 +2,8 @@ import { describe, expect, it } from "bun:test";
 
 import app from "~/index";
 import { server } from "~/mocks";
-import { mockConsumet } from "~/mocks/consumet";
 
 server.listen();
-mockConsumet();
 
 describe('requests the "/episodes" route', () => {
   it("with list of episodes from Anify", async () => {
diff --git a/src/controllers/episodes/getEpisodeUrl/index.spec.ts b/src/controllers/episodes/getEpisodeUrl/index.spec.ts
index ec67324..423ffa5 100644
--- a/src/controllers/episodes/getEpisodeUrl/index.spec.ts
+++ b/src/controllers/episodes/getEpisodeUrl/index.spec.ts
@@ -2,10 +2,8 @@ import { describe, expect, it } from "bun:test";
 
 import app from "~/index";
 import { server } from "~/mocks";
-import { mockConsumet } from "~/mocks/consumet";
 
 server.listen();
-mockConsumet();
 
 describe('requests the "/episodes/:id/url" route', () => {
   it("with sources from Anify", async () => {
diff --git a/src/libs/fcm/verifyFcm.spec.ts b/src/libs/fcm/verifyFcm.spec.ts
index 374f685..e976360 100644
--- a/src/libs/fcm/verifyFcm.spec.ts
+++ b/src/libs/fcm/verifyFcm.spec.ts
@@ -1,7 +1,6 @@
 import { describe, expect, it } from "bun:test";
 
 import { server } from "~/mocks";
-import "~/mocks/gToken";
 
 import { verifyFcmToken } from "./verifyFcmToken";
 
diff --git a/src/mocks/consumet.ts b/src/mocks/consumet.ts
index c2d1010..a55b3bb 100644
--- a/src/mocks/consumet.ts
+++ b/src/mocks/consumet.ts
@@ -2,36 +2,34 @@ import type { IAnimeEpisode, ISource } from "@consumet/extensions";
 
 import { mock } from "bun:test";
 
-export function mockConsumet() {
-  mock.module("src/consumet", () => ({
-    aniList: {
-      fetchEpisodesListById(
-        id: string,
-        dub?: boolean | undefined,
-        fetchFiller?: boolean | undefined,
-      ): Promise<IAnimeEpisode[]> {
-        if (id === "3") {
-          return Promise.resolve([
-            {
-              id: "consumet-1",
-              number: 1,
-              title: "Consumet 1",
-            },
-          ]);
-        }
+mock.module("src/consumet", () => ({
+  aniList: {
+    fetchEpisodesListById(
+      id: string,
+      dub?: boolean | undefined,
+      fetchFiller?: boolean | undefined,
+    ): Promise<IAnimeEpisode[]> {
+      if (id === "3") {
+        return Promise.resolve([
+          {
+            id: "consumet-1",
+            number: 1,
+            title: "Consumet 1",
+          },
+        ]);
+      }
 
-        return Promise.resolve([]);
-      },
-      fetchEpisodeSources(episodeId: string, ...args: any): Promise<ISource> {
-        if (episodeId === "unknown") {
-          return Promise.resolve({ sources: [] });
-        }
-
-        return Promise.resolve({
-          sources: [{ url: "https://consumet.com" }],
-          subtitles: [],
-        });
-      },
+      return Promise.resolve([]);
     },
-  }));
-}
+    fetchEpisodeSources(episodeId: string, ...args: any): Promise<ISource> {
+      if (episodeId === "unknown") {
+        return Promise.resolve({ sources: [] });
+      }
+
+      return Promise.resolve({
+        sources: [{ url: "https://consumet.com" }],
+        subtitles: [],
+      });
+    },
+  },
+}));