bellingcat/auto-archiver
1
0
Fork 0
mirror of https://github.com/bellingcat/auto-archiver.git synced 2026-06-07 19:08:30 +03:00
Code Issues Packages Projects Releases Wiki Activity

fixing s3 storage tests

Browse Source
This commit is contained in:
msramalho
2025-06-04 14:41:00 +01:00
parent 264ba82ea0
commit 5491f3e9e7
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/storages/test_S3_storage.py
View File

@@ -45,6 +45,19 @@ class TestS3Storage:
assert self.storage.get_cdn_url(media) == "https://cdn.example.com/another/path.jpg" assert self.storage.get_cdn_url(media) == "https://cdn.example.com/another/path.jpg"
def test_uploadf_sets_acl_public(self, mocker): def test_uploadf_sets_acl_public(self, mocker):
media = Media("test.png")
mock_file = mocker.MagicMock()
mock_s3_upload = mocker.patch.object(self.storage.s3, "upload_fileobj")
mocker.patch.object(self.storage, "is_upload_needed", return_value=True)
self.storage.uploadf(mock_file, media)
mock_s3_upload.assert_called_once_with(
mock_file,
Bucket="test-bucket",
Key=media.key,
ExtraArgs={"ACL": "public-read", "ContentType": "image/png"},
)
def test_uploadf_detects_charset_for_text_files(self, mocker):
media = Media("test.txt") media = Media("test.txt")
mock_file = mocker.MagicMock() mock_file = mocker.MagicMock()
mock_s3_upload = mocker.patch.object(self.storage.s3, "upload_fileobj") mock_s3_upload = mocker.patch.object(self.storage.s3, "upload_fileobj")
@@ -54,7 +67,7 @@ class TestS3Storage:
mock_file, mock_file,
Bucket="test-bucket", Bucket="test-bucket",
Key=media.key, Key=media.key,
ExtraArgs={"ACL": "public-read", "ContentType": "text/plain"}, ExtraArgs={"ACL": "public-read", "ContentType": "text/plain; charset=utf-8"},
) )
def test_upload_decision_logic(self, mocker): def test_upload_decision_logic(self, mocker):