fix: Add QueryClientProvider to test utils for React Query support

Sannidhya · Sannidhya · commit 1ab1ad9a64ea · 2025-11-30T10:26:49.000+05:30
diff --git a/webview-ui/src/utils/test-utils.tsx b/webview-ui/src/utils/test-utils.tsx
@@ -1,5 +1,6 @@
 import React from "react"
 import { render, RenderOptions } from "@testing-library/react"
+import { QueryClient, QueryClientProvider } from "@tanstack/react-query"
 
 import { TooltipProvider } from "@src/components/ui/tooltip"
 import { STANDARD_TOOLTIP_DELAY } from "@src/components/ui/standard-tooltip"
@@ -9,7 +10,20 @@ interface AllTheProvidersProps {
 }
 
 const AllTheProviders = ({ children }: AllTheProvidersProps) => {
-	return <TooltipProvider delayDuration={STANDARD_TOOLTIP_DELAY}>{children}</TooltipProvider>
+	// Create a new QueryClient for each test to avoid state leakage
+	const queryClient = new QueryClient({
+		defaultOptions: {
+			queries: {
+				retry: false, // Disable retries in tests
+			},
+		},
+	})
+
+	return (
+		<QueryClientProvider client={queryClient}>
+			<TooltipProvider delayDuration={STANDARD_TOOLTIP_DELAY}>{children}</TooltipProvider>
+		</QueryClientProvider>
+	)
 }
 
 const customRender = (ui: React.ReactElement, options?: Omit<RenderOptions, "wrapper">) =>
