test_cli_sso_callback_regenerate_existing_key

2025-09-23 15:42:07 -07:00 · 2025-09-23 15:42:07 -07:00 · 7e64df92a5
commit 7e64df92a5
parent 8d42eccc8a
1 changed files with 1 additions and 1 deletions
--- a/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py
+++ b/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py
@ -1275,7 +1275,7 @@ class TestCLIKeyRegenerationFlow:
            )
            
            # Assert
-            mock_regenerate.assert_called_once_with(existing_key, new_key)
+            mock_regenerate.assert_called_once_with(existing_key=existing_key, new_key=new_key, user_id=None)
            assert result.status_code == 200
            assert "Success" in result.body.decode()