From 0587e3b8f8749398d1f83b713baf8944a0505090 Mon Sep 17 00:00:00 2001 From: Yuneng Jiang Date: Fri, 11 Sep 2026 19:07:04 -0700 Subject: [PATCH] test(ui): isolate search provider query state --- .../_components/CreateSearchTools.integration.test.tsx | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ui/litellm-dashboard/src/app/(dashboard)/search-tools/_components/CreateSearchTools.integration.test.tsx b/ui/litellm-dashboard/src/app/(dashboard)/search-tools/_components/CreateSearchTools.integration.test.tsx index 8dbd43225db..dac43885da4 100644 --- a/ui/litellm-dashboard/src/app/(dashboard)/search-tools/_components/CreateSearchTools.integration.test.tsx +++ b/ui/litellm-dashboard/src/app/(dashboard)/search-tools/_components/CreateSearchTools.integration.test.tsx @@ -1,4 +1,4 @@ -import { fireEvent, renderWithProviders, screen, waitFor } from "../../../../../tests/test-utils"; +import { fireEvent, renderWithProviders, screen, testQueryClient, waitFor } from "../../../../../tests/test-utils"; import userEvent from "@testing-library/user-event"; import { beforeEach, describe, expect, it, vi } from "vitest"; import * as networking from "@/components/networking"; @@ -40,6 +40,7 @@ const pickProvider = async (user: ReturnType, label: str describe("CreateSearchTools submit payload", () => { beforeEach(() => { + testQueryClient.clear(); vi.clearAllMocks(); vi.mocked(networking.fetchAvailableSearchProviders).mockResolvedValue({ providers }); vi.mocked(networking.createSearchTool).mockResolvedValue({ search_tool_id: "st-1" }); @@ -147,6 +148,7 @@ describe("CreateSearchTools submit payload", () => { fireEvent.change(await screen.findByLabelText(/Search Tool Name/), { target: { value: "synthetic-search" } }); await pickProvider(user, "Perplexity AI"); await user.click(screen.getByRole("button", { name: "Clear" })); + expect(networking.fetchAvailableSearchProviders).toHaveBeenCalledTimes(1); await user.click(screen.getByRole("button", { name: "Add Search Tool" })); expect(await screen.findByText("Please select a search provider")).toBeInTheDocument(); expect(networking.createSearchTool).not.toHaveBeenCalled();