mirror of
https://github.com/TronoSfera/Law.git
synced 2026-05-18 10:03:45 +03:00
commit
71047a46b0
2 changed files with 9 additions and 3 deletions
|
|
@ -77,7 +77,7 @@ class S3Storage:
|
||||||
self.ensure_bucket()
|
self.ensure_bucket()
|
||||||
url = self.client.generate_presigned_url(
|
url = self.client.generate_presigned_url(
|
||||||
"put_object",
|
"put_object",
|
||||||
Params={"Bucket": self.bucket, "Key": key},
|
Params={"Bucket": self.bucket, "Key": key, "ContentType": mime_type},
|
||||||
ExpiresIn=expires_sec,
|
ExpiresIn=expires_sec,
|
||||||
HttpMethod="PUT",
|
HttpMethod="PUT",
|
||||||
)
|
)
|
||||||
|
|
|
||||||
|
|
@ -979,5 +979,11 @@ class UploadsS3Tests(unittest.TestCase):
|
||||||
self.assertTrue(second.startswith("/s3/"))
|
self.assertTrue(second.startswith("/s3/"))
|
||||||
self.assertEqual(fake_client.head_bucket_calls, 1)
|
self.assertEqual(fake_client.head_bucket_calls, 1)
|
||||||
self.assertEqual(fake_client.create_bucket_calls, 0)
|
self.assertEqual(fake_client.create_bucket_calls, 0)
|
||||||
self.assertEqual(fake_client.presign_params[0], {"Bucket": settings.S3_BUCKET, "Key": "avatars/test-user/photo.png"})
|
self.assertEqual(
|
||||||
self.assertEqual(fake_client.presign_params[1], {"Bucket": settings.S3_BUCKET, "Key": "avatars/test-user/photo-2.png"})
|
fake_client.presign_params[0],
|
||||||
|
{"Bucket": settings.S3_BUCKET, "Key": "avatars/test-user/photo.png", "ContentType": "image/png"},
|
||||||
|
)
|
||||||
|
self.assertEqual(
|
||||||
|
fake_client.presign_params[1],
|
||||||
|
{"Bucket": settings.S3_BUCKET, "Key": "avatars/test-user/photo-2.png", "ContentType": "image/png"},
|
||||||
|
)
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue