From 8a73b16aa74af00fec6615233e6b16ec85bd49ea Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 23 Jul 2026 21:17:30 +0000 Subject: [PATCH] test(ui): fix optimized usage tab assertions Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../cost-optimization/_components/UsageTab.test.tsx | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/ui/litellm-dashboard/src/app/(dashboard)/cost-optimization/_components/UsageTab.test.tsx b/ui/litellm-dashboard/src/app/(dashboard)/cost-optimization/_components/UsageTab.test.tsx index cc1135d34fb..24aaaedf428 100644 --- a/ui/litellm-dashboard/src/app/(dashboard)/cost-optimization/_components/UsageTab.test.tsx +++ b/ui/litellm-dashboard/src/app/(dashboard)/cost-optimization/_components/UsageTab.test.tsx @@ -162,7 +162,7 @@ const renderWithToolSpend = (results: DailyData[], toolSpend: ToolSpendResponse) describe("UsageTab", () => { it("sums compression and caching dollars across days into the summary cards", () => { - const { getByText } = renderWith([ + const { getByText, getAllByText } = renderWith([ day("2026-07-12", { compression_savings_spend: 0.04, prompt_caching_savings_spend: 0.006, @@ -176,7 +176,7 @@ describe("UsageTab", () => { ]); expect(getByText("$0.1560")).toBeInTheDocument(); - expect(getByText("$0.1400")).toBeInTheDocument(); + expect(getAllByText("$0.1400")[0]).toBeInTheDocument(); expect(getByText("$0.0160")).toBeInTheDocument(); expect(getByText("140,000 tokens compressed")).toBeInTheDocument(); }); @@ -223,7 +223,8 @@ describe("UsageTab", () => { const detailQuery = mockUseQuery.mock.calls .map(([args]) => args as { queryKey: string[]; queryFn: () => Promise }) - .find((args) => args.queryKey[0] === "cost-optimization-spend-log"); + .filter((args) => args.queryKey[0] === "cost-optimization-spend-log") + .at(-1); expect(detailQuery).toBeDefined(); await detailQuery?.queryFn(); expect(mockUiSpendLogsCall).toHaveBeenCalledWith(