mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-25 01:02:15 +00:00
test(auth): pass throttle to authenticate_user in reset-required tests
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
07c9355f4e
commit
6bfbc7dba4
1 changed files with 8 additions and 1 deletions
|
|
@ -733,7 +733,12 @@ async def _db_login(throttle, username: str, password: str, *, correct: bool):
|
|||
),
|
||||
):
|
||||
return await authenticate_user(
|
||||
username=username, password=password, master_key="sk-master", prisma_client=MagicMock(), throttle=throttle
|
||||
username=username,
|
||||
password=password,
|
||||
master_key="sk-master",
|
||||
prisma_client=MagicMock(),
|
||||
throttle=throttle,
|
||||
general_settings=_POLICY_NO_BREACH_CHECK,
|
||||
)
|
||||
|
||||
|
||||
|
|
@ -2122,6 +2127,7 @@ class TestPasswordResetRequiredSessionMinting:
|
|||
password="Str0ng!Passw0rd",
|
||||
master_key="sk-1234",
|
||||
prisma_client=mock_prisma_client,
|
||||
throttle=_unlimited_throttle(),
|
||||
general_settings=_POLICY_NO_BREACH_CHECK,
|
||||
)
|
||||
return result, mock_generate_key.call_args.kwargs
|
||||
|
|
@ -2163,6 +2169,7 @@ class TestPasswordResetRequiredSessionMinting:
|
|||
password="Str0ng!Passw0rd",
|
||||
master_key="sk-1234",
|
||||
prisma_client=mock_prisma_client,
|
||||
throttle=_unlimited_throttle(),
|
||||
general_settings=_POLICY_NO_BREACH_CHECK,
|
||||
)
|
||||
return result, mock_generate_key.call_args.kwargs, mock_screen.call_args.kwargs
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue