From efdf05cab011bbb08d7ece0983c7df845b9d0510 Mon Sep 17 00:00:00 2001
From: yuneng-jiang <yuneng.jiang@gmail.com>
Date: Mon, 9 Feb 2026 18:01:52 -0800
Subject: [PATCH] e2e tests, key delete, update tpm rpm, and regenerate

---
 ui/litellm-dashboard/e2e_tests/constants.ts   |  5 ++++
 .../e2e_tests/tests/keys/deleteKey.spec.ts    | 25 +++++++++++++++++
 .../tests/keys/regenerateKey.spec.ts          | 21 +++++++++++++++
 .../tests/keys/updateKeyLimits.spec.ts        | 27 +++++++++++++++++++
 4 files changed, 78 insertions(+)
 create mode 100644 ui/litellm-dashboard/e2e_tests/tests/keys/deleteKey.spec.ts
 create mode 100644 ui/litellm-dashboard/e2e_tests/tests/keys/regenerateKey.spec.ts
 create mode 100644 ui/litellm-dashboard/e2e_tests/tests/keys/updateKeyLimits.spec.ts

diff --git a/ui/litellm-dashboard/e2e_tests/constants.ts b/ui/litellm-dashboard/e2e_tests/constants.ts
index b07bd68fcf..58b56af0a2 100644
--- a/ui/litellm-dashboard/e2e_tests/constants.ts
+++ b/ui/litellm-dashboard/e2e_tests/constants.ts
@@ -1 +1,6 @@
 export const ADMIN_STORAGE_PATH = "admin.storageState.json";
+
+export const E2E_UPDATE_LIMITS_KEY_ID_PREFIX = "102c";
+export const E2E_DELETE_KEY_ID_PREFIX = "94a5";
+export const E2E_DELETE_KEY_NAME = "e2eDeleteKey";
+export const E2E_REGENERATE_KEY_ID_PREFIX = "593a";
diff --git a/ui/litellm-dashboard/e2e_tests/tests/keys/deleteKey.spec.ts b/ui/litellm-dashboard/e2e_tests/tests/keys/deleteKey.spec.ts
new file mode 100644
index 0000000000..a584131625
--- /dev/null
+++ b/ui/litellm-dashboard/e2e_tests/tests/keys/deleteKey.spec.ts
@@ -0,0 +1,25 @@
+import { test, expect } from "@playwright/test";
+import { ADMIN_STORAGE_PATH, E2E_DELETE_KEY_ID_PREFIX, E2E_DELETE_KEY_NAME } from "../../constants";
+import { Page } from "../../fixtures/pages";
+import { navigateToPage } from "../../helpers/navigation";
+
+test.describe("Delete Key", () => {
+  test.use({ storageState: ADMIN_STORAGE_PATH });
+
+  test("Able to delete a key", async ({ page }) => {
+    await navigateToPage(page, Page.ApiKeys);
+    await expect(page.getByRole("button", { name: "Next" })).toBeVisible();
+    await page
+      .locator("button", {
+        hasText: E2E_DELETE_KEY_ID_PREFIX,
+      })
+      .click();
+    await page.getByRole("button", { name: "Delete Key" }).click();
+    await page.getByRole("textbox", { name: E2E_DELETE_KEY_NAME }).click();
+    await page.getByRole("textbox", { name: E2E_DELETE_KEY_NAME }).fill(E2E_DELETE_KEY_NAME);
+    const deleteButton = page.getByRole("button", { name: "Delete", exact: true });
+    await expect(deleteButton).toBeEnabled();
+    await deleteButton.click();
+    await expect(page.getByText("Key deleted successfully")).toBeVisible();
+  });
+});
diff --git a/ui/litellm-dashboard/e2e_tests/tests/keys/regenerateKey.spec.ts b/ui/litellm-dashboard/e2e_tests/tests/keys/regenerateKey.spec.ts
new file mode 100644
index 0000000000..0188a4f81c
--- /dev/null
+++ b/ui/litellm-dashboard/e2e_tests/tests/keys/regenerateKey.spec.ts
@@ -0,0 +1,21 @@
+import { test, expect } from "@playwright/test";
+import { ADMIN_STORAGE_PATH, E2E_REGENERATE_KEY_ID_PREFIX } from "../../constants";
+import { Page } from "../../fixtures/pages";
+import { navigateToPage } from "../../helpers/navigation";
+
+test.describe("Regenerate Key", () => {
+  test.use({ storageState: ADMIN_STORAGE_PATH });
+
+  test("Able to regenerate a key", async ({ page }) => {
+    await navigateToPage(page, Page.ApiKeys);
+    await expect(page.getByRole("button", { name: "Next" })).toBeVisible();
+    await page
+      .locator("button", {
+        hasText: E2E_REGENERATE_KEY_ID_PREFIX,
+      })
+      .click();
+    await page.getByRole("button", { name: "Regenerate Key" }).click();
+    await page.getByRole("button", { name: "Regenerate", exact: true }).click();
+    await expect(page.getByText("Virtual Key regenerated")).toBeVisible();
+  });
+});
diff --git a/ui/litellm-dashboard/e2e_tests/tests/keys/updateKeyLimits.spec.ts b/ui/litellm-dashboard/e2e_tests/tests/keys/updateKeyLimits.spec.ts
new file mode 100644
index 0000000000..6cae36272a
--- /dev/null
+++ b/ui/litellm-dashboard/e2e_tests/tests/keys/updateKeyLimits.spec.ts
@@ -0,0 +1,27 @@
+import { test, expect } from "@playwright/test";
+import { ADMIN_STORAGE_PATH, E2E_UPDATE_LIMITS_KEY_ID_PREFIX } from "../../constants";
+import { Page } from "../../fixtures/pages";
+import { navigateToPage } from "../../helpers/navigation";
+
+test.describe("Update Key TPM and RPM Limits", () => {
+  test.use({ storageState: ADMIN_STORAGE_PATH });
+
+  test("Able to update a key's TPM and RPM limits", async ({ page }) => {
+    await navigateToPage(page, Page.ApiKeys);
+    await expect(page.getByRole("button", { name: "Next" })).toBeVisible();
+    await page
+      .locator("button", {
+        hasText: E2E_UPDATE_LIMITS_KEY_ID_PREFIX,
+      })
+      .click();
+    await page.getByRole("tab", { name: "Settings" }).click();
+    await page.getByRole("button", { name: "Edit Settings" }).click();
+    await page.getByRole("spinbutton", { name: "TPM Limit" }).click();
+    await page.getByRole("spinbutton", { name: "TPM Limit" }).fill("123");
+    await page.getByRole("spinbutton", { name: "RPM Limit" }).click();
+    await page.getByRole("spinbutton", { name: "RPM Limit" }).fill("456");
+    await page.getByRole("button", { name: "Save Changes" }).click();
+    await expect(page.getByRole("paragraph").filter({ hasText: "TPM: 123" })).toBeVisible();
+    await expect(page.getByRole("paragraph").filter({ hasText: "RPM: 456" })).toBeVisible();
+  });
+});