scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

Adding test

Browse source
This commit is contained in:
yuneng-jiang 2025-12-12 12:23:11 -08:00
parent 5b0bf86ee5
commit 1037dc18b9
1 changed files with 75 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

75
ui/litellm-dashboard/src/components/UsagePage/components/EntityUsage/EntityUsage.test.tsx
View file

@ -269,4 +269,79 @@ describe("EntityUsage", () => {
expect(screen.getByText("Total Spend")).toBeInTheDocument();
expect(screen.getAllByText("0")[0]).toBeInTheDocument();
});
it("should use entityList label when entityList is provided and entity exists", async () => {
const customEntityList = [
{ label: "Custom Tag Label", value: "tag-1" },
{ label: "Tag 2", value: "tag-2" },
];
render(<EntityUsage {...defaultProps} entityList={customEntityList} />);
await waitFor(() => {
expect(mockTagDailyActivityCall).toHaveBeenCalled();
});
await waitFor(() => {
expect(screen.getByText("Custom Tag Label")).toBeInTheDocument();
});
});
it("should fallback to team_alias when entityList is provided but entity does not exist", async () => {
const customEntityList = [{ label: "Tag 2", value: "tag-2" }];
render(<EntityUsage {...defaultProps} entityList={customEntityList} />);
await waitFor(() => {
expect(mockTagDailyActivityCall).toHaveBeenCalled();
});
await waitFor(() => {
expect(screen.getByText("Tag 1")).toBeInTheDocument();
});
});
it("should fallback to team_alias when entityList is null", async () => {
render(<EntityUsage {...defaultProps} entityList={null} />);
await waitFor(() => {
expect(mockTagDailyActivityCall).toHaveBeenCalled();
});
await waitFor(() => {
expect(screen.getByText("Tag 1")).toBeInTheDocument();
});
});
it("should fallback to entity value when no entityList and no team_alias", async () => {
const spendDataWithoutAlias = {
...mockSpendData,
results: [
{
...mockSpendData.results[0],
breakdown: {
...mockSpendData.results[0].breakdown,
entities: {
"tag-1": {
...mockSpendData.results[0].breakdown.entities["tag-1"],
metadata: {},
},
},
},
},
],
};
mockTagDailyActivityCall.mockResolvedValue(spendDataWithoutAlias);
render(<EntityUsage {...defaultProps} entityList={null} />);
await waitFor(() => {
expect(mockTagDailyActivityCall).toHaveBeenCalled();
});
await waitFor(() => {
expect(screen.getByText("tag-1")).toBeInTheDocument();
});
});
});