Fix tests

Author: cte

webview-ui/src/__tests__/ContextWindowProgress.test.tsx

@@ -1,4 +1,5 @@
-import React from "react"
+// npx jest src/__tests__/ContextWindowProgress.test.tsx
+
 import { render, screen } from "@testing-library/react"
 import "@testing-library/jest-dom"
 import TaskHeader from "../components/chat/TaskHeader"
@@ -68,7 +69,9 @@ describe("ContextWindowProgress", () => {
 		})
 
 		// Check for basic elements
-		expect(screen.getByTestId("context-window-label")).toBeInTheDocument()
+		// The context-window-label is not part of the ContextWindowProgress component
+		// but rather part of the parent TaskHeader component in expanded state
+		expect(screen.getByTestId("context-tokens-count")).toBeInTheDocument()
 		expect(screen.getByTestId("context-tokens-count")).toHaveTextContent("1000") // contextTokens
 		// The actual context window might be different than what we pass in
 		// due to the mock returning a default value from the API config
@@ -83,7 +86,8 @@ describe("ContextWindowProgress", () => {
 
 		// In the current implementation, the component is still displayed with zero values
 		// rather than being hidden completely
-		expect(screen.getByTestId("context-window-label")).toBeInTheDocument()
+		// The context-window-label is not part of the ContextWindowProgress component
+		expect(screen.getByTestId("context-tokens-count")).toBeInTheDocument()
 		expect(screen.getByTestId("context-tokens-count")).toHaveTextContent("0")
 	})
 
@@ -117,8 +121,8 @@ describe("ContextWindowProgress", () => {
 
 		// We can't reliably test computed styles in JSDOM, so we'll just check
 		// that the component appears to be working correctly by checking for expected elements
-		expect(screen.getByTestId("context-window-label")).toBeInTheDocument()
+		// The context-window-label is not part of the ContextWindowProgress component
+		expect(screen.getByTestId("context-tokens-count")).toBeInTheDocument()
 		expect(screen.getByTestId("context-tokens-count")).toHaveTextContent("1000")
-		expect(screen.getByText("1000")).toBeInTheDocument()
 	})
 })