From 3c921611af8b8275fe5ef4a2f48087758b0c9461 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Tue, 15 Sep 2026 09:50:23 +0000
Subject: [PATCH] test(ui): settle the TeamSSOSettings save before the test
 returns

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 .../src/components/TeamSSOSettings.test.tsx      | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/TeamSSOSettings.test.tsx b/ui/litellm-dashboard/src/components/TeamSSOSettings.test.tsx
index 4a042c32cda..59508fa92dd 100644
--- a/ui/litellm-dashboard/src/components/TeamSSOSettings.test.tsx
+++ b/ui/litellm-dashboard/src/components/TeamSSOSettings.test.tsx
@@ -564,21 +564,19 @@ describe("TeamSSOSettings", () => {
     });
   });
 
-  it("should disable cancel button while saving", async () => {
+  it("should disable cancel button while saving and re-enable it once the save settles", async () => {
     mockGetDefaultTeamSettings.mockResolvedValue(mockSettingsResponse);
-    mockUpdateDefaultTeamSettings.mockImplementation(
-      () => new Promise((resolve) => setTimeout(() => resolve({ settings: mockSettingsResponse.values }), 100)),
-    );
+    const save = Promise.withResolvers<{ settings: typeof mockSettingsResponse.values }>();
+    mockUpdateDefaultTeamSettings.mockImplementation(() => save.promise);
 
     renderWithProviders(<TeamSSOSettings {...defaultProps} />);
 
-    await waitFor(() => {
-      expect(screen.getByRole("button", { name: /Edit Settings/i })).toBeInTheDocument();
-    });
-
-    await userEvent.click(screen.getByRole("button", { name: /Edit Settings/i }));
+    await userEvent.click(await screen.findByRole("button", { name: /Edit Settings/i }));
     await userEvent.click(screen.getByRole("button", { name: /Save Changes/i }));
 
     expect(screen.getByRole("button", { name: /Cancel/i })).toBeDisabled();
+
+    save.resolve({ settings: mockSettingsResponse.values });
+    expect(await screen.findByRole("button", { name: /Edit Settings/i })).toBeInTheDocument();
   });
 });