fix: Fix chat history unit test

tjtanjin · Oct 5, 2024 · b5f4890 · b5f4890
1 parent d56d2fd
commit b5f4890
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/__tests__/hooks/useChatHistoryInternal.test.ts b/__tests__/hooks/useChatHistoryInternal.test.ts
@@ -55,7 +55,7 @@ describe("useChatHistoryInternal Hook", () => {
 
 		// mocks loadChatHistory to resolve successfully
 		mockLoadChatHistory.mockImplementation(
-			(settings, styles, chatHistory, setMessages, setTextAreaDisabled) => {
+			(settings, styles, chatHistory, setMessages, prevTextAreaDisabled, setTextAreaDisabled) => {
 				setMessages(chatHistory);
 				setTextAreaDisabled(false);
 				return Promise.resolve();
@@ -90,6 +90,7 @@ describe("useChatHistoryInternal Hook", () => {
 			expect.any(Object),
 			initialChatHistory,
 			expect.any(Function),
+			expect.any(Boolean),
 			expect.any(Function)
 		);