test(webview): cover prompt history content updates

Author: PierrunoYT

webview-ui/src/components/chat/hooks/__tests__/usePromptHistory.spec.ts

@@ -1,7 +1,7 @@
 import { ClineMessage, HistoryItem } from "@roo-code/types"
 import { act, renderHook } from "@testing-library/react"
 
-import { usePromptHistory } from "../usePromptHistory"
+import { usePromptHistory, type UsePromptHistoryReturn } from "../usePromptHistory"
 
 describe("usePromptHistory", () => {
 	it("resets navigation when switching to conversation history with identical prompts", () => {
@@ -21,10 +21,7 @@ describe("usePromptHistory", () => {
 		const conversationHistory: ClineMessage[] = [{ ts: 2, type: "say", say: "user_feedback", text: prompt }]
 		const setInputValue = vi.fn()
 
-		const { result, rerender } = renderHook<
-			ReturnType<typeof usePromptHistory>,
-			{ clineMessages: ClineMessage[] | undefined }
-		>(
+		const { result, rerender } = renderHook<UsePromptHistoryReturn, { clineMessages: ClineMessage[] | undefined }>(
 			({ clineMessages }) =>
 				usePromptHistory({
 					clineMessages,
@@ -51,4 +48,37 @@ describe("usePromptHistory", () => {
 		expect(result.current.historyIndex).toBe(-1)
 		expect(result.current.tempInput).toBe("")
 	})
+
+	it("resets navigation when the current history source gains a prompt", () => {
+		const firstPrompt = "Explain this code"
+		const secondPrompt = "Now simplify it"
+		const initialHistory: ClineMessage[] = [{ ts: 1, type: "say", say: "user_feedback", text: firstPrompt }]
+		const updatedHistory: ClineMessage[] = [
+			...initialHistory,
+			{ ts: 2, type: "say", say: "user_feedback", text: secondPrompt },
+		]
+
+		const { result, rerender } = renderHook<UsePromptHistoryReturn, { clineMessages: ClineMessage[] }>(
+			({ clineMessages }) =>
+				usePromptHistory({
+					clineMessages,
+					taskHistory: undefined,
+					cwd: "/workspace",
+					inputValue: "draft",
+					setInputValue: vi.fn(),
+				}),
+			{ initialProps: { clineMessages: initialHistory } },
+		)
+
+		act(() => {
+			result.current.setHistoryIndex(0)
+			result.current.setTempInput("draft")
+		})
+
+		rerender({ clineMessages: updatedHistory })
+
+		expect(result.current.promptHistory).toEqual([secondPrompt, firstPrompt])
+		expect(result.current.historyIndex).toBe(-1)
+		expect(result.current.tempInput).toBe("")
+	})
 })