From fba61f8e2a7cd32d810e4d79f4baf2c5cd491de9 Mon Sep 17 00:00:00 2001
From: yuneng-jiang <yuneng.jiang@gmail.com>
Date: Thu, 15 Jan 2026 21:42:58 -0800
Subject: [PATCH] adding mocks

---
 .../ModelSelect/ModelSelect.test.tsx          | 23 ++++++++-----------
 1 file changed, 10 insertions(+), 13 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/ModelSelect/ModelSelect.test.tsx b/ui/litellm-dashboard/src/components/ModelSelect/ModelSelect.test.tsx
index 9a56d219ba..80acf5d75f 100644
--- a/ui/litellm-dashboard/src/components/ModelSelect/ModelSelect.test.tsx
+++ b/ui/litellm-dashboard/src/components/ModelSelect/ModelSelect.test.tsx
@@ -192,7 +192,12 @@ describe("ModelSelect", () => {
     } as any);
 
     renderWithProviders(
-      <ModelSelect onChange={mockOnChange} context="organization" organizationID="org-1" options={{ includeSpecialOptions: true }} />,
+      <ModelSelect
+        onChange={mockOnChange}
+        context="organization"
+        organizationID="org-1"
+        options={{ includeSpecialOptions: true }}
+      />,
     );
 
     await waitFor(() => {
@@ -392,9 +397,7 @@ describe("ModelSelect", () => {
       isLoading: false,
     } as any);
 
-    renderWithProviders(
-      <ModelSelect onChange={mockOnChange} context="organization" organizationID="org-1" />,
-    );
+    renderWithProviders(<ModelSelect onChange={mockOnChange} context="organization" organizationID="org-1" />);
 
     await waitFor(() => {
       expect(screen.getByText("gpt-4")).toBeInTheDocument();
@@ -426,9 +429,7 @@ describe("ModelSelect", () => {
       isLoading: false,
     } as any);
 
-    renderWithProviders(
-      <ModelSelect onChange={mockOnChange} context="organization" organizationID="org-1" />,
-    );
+    renderWithProviders(<ModelSelect onChange={mockOnChange} context="organization" organizationID="org-1" />);
 
     await waitFor(() => {
       expect(screen.getByText("gpt-4")).toBeInTheDocument();
@@ -510,9 +511,7 @@ describe("ModelSelect", () => {
       isLoading: false,
     } as any);
 
-    renderWithProviders(
-      <ModelSelect onChange={mockOnChange} context="user" options={{ includeUserModels: true }} />,
-    );
+    renderWithProviders(<ModelSelect onChange={mockOnChange} context="user" options={{ includeUserModels: true }} />);
 
     await waitFor(() => {
       expect(screen.getByText("gpt-4")).toBeInTheDocument();
@@ -555,9 +554,7 @@ describe("ModelSelect", () => {
       isLoading: false,
     } as any);
 
-    renderWithProviders(
-      <ModelSelect onChange={mockOnChange} context="team" teamID="team-1" organizationID="org-1" />,
-    );
+    renderWithProviders(<ModelSelect onChange={mockOnChange} context="team" teamID="team-1" organizationID="org-1" />);
 
     await waitFor(() => {
       expect(screen.getByText("gpt-4")).toBeInTheDocument();