scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 133

Merge pull request #17161 from BerriAI/litellm_fix_flaky_ui_test_teaminfo

Browse source 
[Fix] UI - Fix Flaky Test
This commit is contained in:
yuneng-jiang 2025-11-26 14:02:38 -08:00 • committed by GitHub
commit d8e4aaf301
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
ui/litellm-dashboard/src/components/team/team_info.test.tsx
View file

@ -58,7 +58,7 @@ describe("TeamInfoView", () => {
team_memberships: [],
});
vi.mocked(networking.getGuardrailsList).mockResolvedValue([]);
vi.mocked(networking.getGuardrailsList).mockResolvedValue({ guardrails: [] });
vi.mocked(networking.fetchMCPAccessGroups).mockResolvedValue([]);
render(
@ -74,9 +74,13 @@ describe("TeamInfoView", () => {
premiumUser={false}
/>,
);
await waitFor(() => {
expect(screen.queryByText("User ID")).not.toBeNull();
});
await waitFor(
() => {
expect(screen.queryByText("User ID")).not.toBeNull();
},
// This is a workaround to fix the flaky test issue. TODO: Remove this once we have a better solution.
{ timeout: 10000 },
);
});
it("should not show all-proxy-models option when user has no access to it", async () => {
@ -116,7 +120,7 @@ describe("TeamInfoView", () => {
team_memberships: [],
});
vi.mocked(networking.getGuardrailsList).mockResolvedValue([]);
vi.mocked(networking.getGuardrailsList).mockResolvedValue({ guardrails: [] });
vi.mocked(networking.fetchMCPAccessGroups).mockResolvedValue([]);
render(
@ -157,5 +161,6 @@ describe("TeamInfoView", () => {
const allProxyModelsOption = screen.queryByText("All Proxy Models");
expect(allProxyModelsOption).not.toBeInTheDocument();
});
}, // This is a workaround to fix the flaky test issue. TODO: Remove this once we have a better solution.
10000);
});