From 1b092c55db1f3d57acab84cd1e2193440727f1bd Mon Sep 17 00:00:00 2001
From: mamadoudicko <mamadoudicko100@gmail.com>
Date: Thu, 6 Jul 2023 18:47:56 +0200
Subject: [PATCH] test(useBrainApi): add unit tests

---
 .../api/brain/__tests__/useBrainApi.test.ts   | 37 +++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 frontend/lib/api/brain/__tests__/useBrainApi.test.ts

diff --git a/frontend/lib/api/brain/__tests__/useBrainApi.test.ts b/frontend/lib/api/brain/__tests__/useBrainApi.test.ts
new file mode 100644
index 000000000..fd4deb881
--- /dev/null
+++ b/frontend/lib/api/brain/__tests__/useBrainApi.test.ts
@@ -0,0 +1,37 @@
+import { renderHook } from "@testing-library/react";
+import { afterEach, describe, expect, it, vi } from "vitest";
+
+import { useBrainApi } from "../useBrainApi";
+
+const axiosGetMock = vi.fn(() => ({
+  data: {
+    documents: [],
+  },
+}));
+
+vi.mock("@/lib/hooks", () => ({
+  useAxios: vi.fn(() => ({
+    axiosInstance: {
+      get: axiosGetMock,
+    },
+  })),
+}));
+
+describe("useBrainApi", () => {
+  afterEach(() => {
+    vi.resetAllMocks();
+  });
+
+  it("should call getBrainDocuments with the correct parameters", async () => {
+    const {
+      result: {
+        current: { getBrainDocuments },
+      },
+    } = renderHook(() => useBrainApi());
+    const brainId = "123";
+    await getBrainDocuments(brainId);
+
+    expect(axiosGetMock).toHaveBeenCalledTimes(1);
+    expect(axiosGetMock).toHaveBeenCalledWith(`/explore/?brain_id=${brainId}`);
+  });
+});