diff --git a/ui/litellm-dashboard/src/components/create_user_button.test.tsx b/ui/litellm-dashboard/src/components/create_user_button.test.tsx new file mode 100644 index 00000000000..e40a1e0ac3c --- /dev/null +++ b/ui/litellm-dashboard/src/components/create_user_button.test.tsx @@ -0,0 +1,35 @@ +import React from "react"; +import { render } from "@testing-library/react"; +import { describe, it, expect, vi } from "vitest"; +import { QueryClient, QueryClientProvider } from "@tanstack/react-query"; +import Createuser from "./create_user_button"; + +vi.mock("./networking", () => ({ + userCreateCall: vi.fn(), + modelAvailableCall: vi.fn().mockResolvedValue({ data: [] }), + invitationCreateCall: vi.fn(), + getProxyUISettings: vi.fn().mockResolvedValue({ + PROXY_BASE_URL: null, + PROXY_LOGOUT_URL: null, + DEFAULT_TEAM_DISABLED: false, + SSO_ENABLED: false, + }), + getProxyBaseUrl: vi.fn().mockReturnValue("http://localhost"), +})); + +const createQueryClient = () => + new QueryClient({ + defaultOptions: { queries: { retry: false, gcTime: 0 } }, + }); + +describe("Create User Button", () => { + it("should render the create user button", () => { + const qc = createQueryClient(); + const { getByText } = render( + + + , + ); + expect(getByText("Create User")).toBeInTheDocument(); + }); +}); diff --git a/ui/litellm-dashboard/src/components/create_user_button.tsx b/ui/litellm-dashboard/src/components/create_user_button.tsx index 9973c2e777c..6fb6f80c4b4 100644 --- a/ui/litellm-dashboard/src/components/create_user_button.tsx +++ b/ui/litellm-dashboard/src/components/create_user_button.tsx @@ -26,6 +26,7 @@ import { InfoCircleOutlined } from "@ant-design/icons"; import { getModelDisplayName } from "./key_team_helpers/fetch_available_models_team_key"; import { useQueryClient } from "@tanstack/react-query"; import NotificationsManager from "./molecules/notifications_manager"; +import TeamDropdown from "./common_components/team_dropdown"; // Helper function to generate UUID compatible across all environments const generateUUID = (): string => { @@ -201,19 +202,9 @@ const Createuser: React.FC = ({ ))} - - + @@ -275,24 +266,12 @@ const Createuser: React.FC = ({ - +