From 4b951c3ced340ddfd1d6742b9684229be1769e75 Mon Sep 17 00:00:00 2001
From: yuneng-jiang <yuneng.jiang@gmail.com>
Date: Wed, 26 Nov 2025 17:04:12 -0800
Subject: [PATCH] Removing flaky tests

---
 .../organisms/create_key_button.test.tsx      | 123 +-----------------
 1 file changed, 3 insertions(+), 120 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/organisms/create_key_button.test.tsx b/ui/litellm-dashboard/src/components/organisms/create_key_button.test.tsx
index 34eb1c254b0..08c05f5fadf 100644
--- a/ui/litellm-dashboard/src/components/organisms/create_key_button.test.tsx
+++ b/ui/litellm-dashboard/src/components/organisms/create_key_button.test.tsx
@@ -1,5 +1,5 @@
-import { act, fireEvent, render, screen, waitFor } from "@testing-library/react";
-import { describe, expect, it, vi, beforeEach } from "vitest";
+import { render, screen } from "@testing-library/react";
+import { beforeEach, describe, expect, it, vi } from "vitest";
 import CreateKey from "./create_key_button";
 
 const mockKeyCreateCall = vi.fn().mockResolvedValue({
@@ -61,121 +61,4 @@ describe("CreateKey", () => {
     render(<CreateKey {...defaultProps} />);
     expect(screen.getByRole("button", { name: /create new key/i })).toBeInTheDocument();
   });
-
-  it("should keep duration as null when nothing is inputted", async () => {
-    const addKey = vi.fn();
-    render(<CreateKey {...defaultProps} addKey={addKey} />);
-
-    const createButton = screen.getByRole("button", { name: /create new key/i });
-    act(() => {
-      fireEvent.click(createButton);
-    });
-
-    await waitFor(() => {
-      expect(screen.getByText("Key Ownership")).toBeInTheDocument();
-    });
-
-    const keyAliasInput = screen.getByPlaceholderText("");
-    act(() => {
-      fireEvent.change(keyAliasInput, { target: { value: "test-key" } });
-    });
-
-    const modelsSelect = screen.getByPlaceholderText("Select models");
-    act(() => {
-      fireEvent.mouseDown(modelsSelect);
-    });
-
-    await waitFor(() => {
-      const allTeamModelsOption = screen.getByText("All Team Models");
-      act(() => {
-        fireEvent.click(allTeamModelsOption);
-      });
-    });
-
-    const submitButton = screen.getByRole("button", { name: /create key/i });
-
-    let formValues: Record<string, any> = {};
-    mockKeyCreateCall.mockImplementation(async (_token: string, _userId: string, values: Record<string, any>) => {
-      formValues = values;
-      return { key: "test-api-key", soft_budget: null };
-    });
-
-    act(() => {
-      fireEvent.click(submitButton);
-    });
-
-    await waitFor(() => {
-      expect(addKey).toHaveBeenCalled();
-    });
-
-    expect(formValues.duration).toBeNull();
-  }, 10000); // 10 second timeout for complex test
-
-  it("should set duration correctly when a value is provided", async () => {
-    const addKey = vi.fn();
-    render(<CreateKey {...defaultProps} addKey={addKey} />);
-
-    const createButton = screen.getByRole("button", { name: /create new key/i });
-    act(() => {
-      fireEvent.click(createButton);
-    });
-
-    await waitFor(() => {
-      expect(screen.getByText("Key Ownership")).toBeInTheDocument();
-    });
-
-    const keyAliasInput = screen.getByPlaceholderText("");
-    act(() => {
-      fireEvent.change(keyAliasInput, { target: { value: "test-key" } });
-    });
-
-    const modelsSelect = screen.getByPlaceholderText("Select models");
-    act(() => {
-      fireEvent.mouseDown(modelsSelect);
-    });
-
-    await waitFor(() => {
-      const allTeamModelsOption = screen.getByText("All Team Models");
-      act(() => {
-        fireEvent.click(allTeamModelsOption);
-      });
-    });
-
-    const optionalSettingsAccordion = screen.getByText("Optional Settings");
-    act(() => {
-      fireEvent.click(optionalSettingsAccordion);
-    });
-
-    await waitFor(() => {
-      const keyLifecycleAccordion = screen.getByText("Key Lifecycle");
-      act(() => {
-        fireEvent.click(keyLifecycleAccordion);
-      });
-    });
-
-    await waitFor(() => {
-      const durationInput = screen.getByPlaceholderText("e.g., 30d");
-      act(() => {
-        fireEvent.change(durationInput, { target: { value: "30d" } });
-      });
-    });
-
-    const submitButton = screen.getByRole("button", { name: /create key/i });
-
-    let formValues: Record<string, any> = {};
-    mockKeyCreateCall.mockImplementation(async (_token: string, _userId: string, values: Record<string, any>) => {
-      formValues = values;
-      return { key: "test-api-key", soft_budget: null };
-    });
-
-    act(() => {
-      fireEvent.click(submitButton);
-    });
-
-    await waitFor(() => {
-      expect(addKey).toHaveBeenCalled();
-    });
-
-    expect(formValues.duration).toBe("30d");
-  });
-}, 10000); // 10 second timeout for complex test
+});