All tests fixed

msveshnikov · Mar 24, 2024 · a9bf512 · a9bf512
1 parent b4b2880
commit a9bf512
Showing 1 changed file with 26 additions and 26 deletions.
diff --git a/src/components/__tests__/ChatHistory.test.js b/src/components/__tests__/ChatHistory.test.js
@@ -15,7 +15,7 @@ describe("ChatHistory", () => {
         },
         {
             user: "User",
-            assistant: null,
+            assistant: "Here is the response with an image",
             fileType: "image/png",
             userImageData: "base64data...",
             error: null,
@@ -24,7 +24,7 @@ describe("ChatHistory", () => {
         { user: "User", assistant: null, fileType: null, userImageData: null, error: "An error occurred", image: null },
         {
             user: "User",
-            assistant: "Here is the response with an image",
+            assistant: null,
             fileType: null,
             userImageData: null,
             error: null,
@@ -117,29 +117,29 @@ describe("ChatHistory", () => {
         expect(modelOutputImage).toBeInTheDocument();
     });
 
-    // test("renders file type icon", () => {
-    //     render(
-    //         <ChatHistory
-    //             chatHistory={mockChatHistory}
-    //             isModelResponding={false}
-    //             chatContainerRef={null}
-    //             getFileTypeIcon={mockGetFileTypeIcon}
-    //         />
-    //     );
-    //     const fileTypeIcon = screen.getByText("📷");
-    //     expect(fileTypeIcon).toBeInTheDocument();
-    // });
+    test("renders file type icon", () => {
+        render(
+            <ChatHistory
+                chatHistory={mockChatHistory}
+                isModelResponding={false}
+                chatContainerRef={null}
+                getFileTypeIcon={mockGetFileTypeIcon}
+            />
+        );
+        const fileTypeIcon = screen.getByText("📷");
+        expect(fileTypeIcon).toBeInTheDocument();
+    });
 
-    // test("renders loading spinner when model is responding", () => {
-    //     render(
-    //         <ChatHistory
-    //             chatHistory={mockChatHistory}
-    //             isModelResponding={true}
-    //             chatContainerRef={null}
-    //             getFileTypeIcon={mockGetFileTypeIcon}
-    //         />
-    //     );
-    //     const loadingSpinner = screen.getByRole("progressbar");
-    //     expect(loadingSpinner).toBeInTheDocument();
-    // });
+    test("renders loading spinner when model is responding", () => {
+        render(
+            <ChatHistory
+                chatHistory={mockChatHistory}
+                isModelResponding={true}
+                chatContainerRef={null}
+                getFileTypeIcon={mockGetFileTypeIcon}
+            />
+        );
+        const loadingSpinner = screen.getByRole("progressbar");
+        expect(loadingSpinner).toBeInTheDocument();
+    });
 });