scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

test(bedrock): capture s3 client kwargs from the boto3 mock instead of a mutable dict

Browse source
This commit is contained in:
mateo-berri 2026-08-31 21:26:10 -07:00
parent 76839ca9d8
commit 60b24abd3e
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/llms/bedrock/files/test_bedrock_files_handler.py
View file

@ -243,12 +243,9 @@ async def test_afile_content_assumes_role_with_external_id(monkeypatch):
def get_object(self, Bucket, Key):
return {"Body": FakeS3Body()}
s3_client_kwargs = {}
def fake_boto3_client(service_name, **kwargs):
if service_name == "sts":
return FakeSTSClient()
s3_client_kwargs.update(kwargs)
return FakeS3Client()
optional_params = {
@ -261,7 +258,7 @@ async def test_afile_content_assumes_role_with_external_id(monkeypatch):
"aws_external_id": "external-id-files-download",
}
with patch.object(boto3, "client", side_effect=fake_boto3_client):
with patch.object(boto3, "client", side_effect=fake_boto3_client) as mock_boto3_client:
response = await BedrockFilesHandler().afile_content(
file_content_request={"file_id": "s3://safe-bucket/litellm-bedrock-files-model-id-abc.jsonl"},
optional_params=optional_params,
@ -269,6 +266,7 @@ async def test_afile_content_assumes_role_with_external_id(monkeypatch):
max_retries=None,
)
s3_client_kwargs = next(call.kwargs for call in mock_boto3_client.call_args_list if call.args[0] == "s3")
assert s3_client_kwargs["aws_access_key_id"] == "ASIAFILESDOWNLOADROLE"
assert s3_client_kwargs["aws_session_token"] == "assumed-session-token"
assert response.content == b'{"custom_id": "req-1"}'