scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

Fix tests/test_litellm/llms/bedrock/test_base_aws_llm.py

Browse source
This commit is contained in:
Sameer Kankute 2026-01-12 08:54:27 +05:30
parent c6101ef977
commit e3e09db749
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/test_litellm/llms/bedrock/test_base_aws_llm.py
View file

@ -582,7 +582,8 @@ def test_eks_irsa_ambient_credentials_used():
)
# Should create STS client without explicit credentials (using ambient credentials)
mock_boto3_client.assert_called_once_with("sts")
# Note: verify parameter is passed for SSL verification
mock_boto3_client.assert_called_once_with("sts", verify=True)
# Should call assume_role
mock_sts_client.assume_role.assert_called_once_with(
@ -637,11 +638,13 @@ def test_explicit_credentials_used_when_provided():
)
# Should create STS client with explicit credentials
# Note: verify parameter is passed for SSL verification
mock_boto3_client.assert_called_once_with(
"sts",
aws_access_key_id="explicit-access-key",
aws_secret_access_key="explicit-secret-key",
aws_session_token="assumed-session-token",
verify=True,
)
# Should call assume_role
@ -701,6 +704,7 @@ def test_partial_credentials_still_use_ambient():
aws_access_key_id="AKIAEXAMPLE",
aws_secret_access_key=None,
aws_session_token=None,
verify=True,
)
# Should still call assume_role
@ -748,7 +752,7 @@ def test_cross_account_role_assumption():
)
# Should use ambient credentials
mock_boto3_client.assert_called_once_with("sts")
mock_boto3_client.assert_called_once_with("sts", verify=True)
# Should call assume_role with cross-account role
mock_sts_client.assume_role.assert_called_once_with(