scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-12 08:42:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 207

test(ui): decouple usage table test from antd

Browse source
This commit is contained in:
Yuneng Jiang 2026-08-12 15:44:21 -07:00
parent 5bc523a09f
commit 8d396ac408
No known key found for this signature in database
1 changed files with 14 additions and 39 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
ui/litellm-dashboard/src/app/(dashboard)/usage/_components/components/EndpointUsage/components/EndpointUsageTable.test.tsx
View file

@ -1,36 +1,7 @@
import { render, screen } from "@testing-library/react";
import { describe, expect, it, vi } from "vitest";
import { renderWithProviders, screen } from "@/../tests/test-utils";
import { describe, expect, it } from "vitest";
import EndpointUsageTable from "./EndpointUsageTable";
vi.mock("antd", async () => {
const React = await import("react");
function Table({ columns, dataSource }: any) {
return React.createElement(
"div",
{ "data-testid": "antd-table" },
columns?.map((col: any) =>
React.createElement("div", { key: col.key, "data-testid": `column-${col.key}` }, col.title),
),
dataSource?.map((row: any) =>
React.createElement(
"div",
{ key: row.key, "data-testid": `row-${row.key}` },
React.createElement("div", null, row.endpoint),
),
),
);
}
(Table as any).displayName = "Table";
function Progress({ percent }: any) {
return React.createElement("div", { "data-testid": "antd-progress", "data-percent": percent });
}
(Progress as any).displayName = "Progress";
return { Table, Progress };
});
describe("EndpointUsageTable", () => {
it("should render", () => {
const mockEndpointData = {
@ -51,14 +22,18 @@ describe("EndpointUsageTable", () => {
},
};
render(<EndpointUsageTable endpointData={mockEndpointData} />);
renderWithProviders(<EndpointUsageTable endpointData={mockEndpointData} />);
expect(screen.getByTestId("antd-table")).toBeInTheDocument();
expect(screen.getByTestId("column-endpoint")).toBeInTheDocument();
expect(screen.getByTestId("column-requests")).toBeInTheDocument();
expect(screen.getByTestId("column-api_requests")).toBeInTheDocument();
expect(screen.getByTestId("column-successRate")).toBeInTheDocument();
expect(screen.getByTestId("column-total_tokens")).toBeInTheDocument();
expect(screen.getByTestId("column-spend")).toBeInTheDocument();
expect(screen.getAllByRole("columnheader").map((header) => header.textContent)).toEqual([
"Endpoint",
"Successful / Failed",
"Total Request",
"Success Rate",
"Total Tokens",
"Spend",
]);
expect(screen.getByText("endpoint-1")).toBeInTheDocument();
expect(screen.getByText("95.00%")).toBeInTheDocument();
expect(screen.getByText("$100.50")).toBeInTheDocument();
});
});