Add user provided SSE-C arguments to CompleteMultipartUpload call

boto · Sep 22, 2023 · 6969d8e · 6969d8e
1 parent 1eef558
commit 6969d8e
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 2 deletions.
diff --git a/.changes/next-release/bugfix-SSEC-19962.json b/.changes/next-release/bugfix-SSEC-19962.json
@@ -0,0 +1,5 @@
+{
+  "type": "bugfix",
+  "category": "``SSE-C``",
+  "description": "Pass SSECustomer* arguements to CompleteMultipartUpload for upload operations\""
+}
diff --git a/s3transfer/copies.py b/s3transfer/copies.py
@@ -69,7 +69,12 @@ class CopySubmissionTask(SubmissionTask):
         'TaggingDirective',
     ]
 
-    COMPLETE_MULTIPART_ARGS = ['RequestPayer', 'ExpectedBucketOwner']
+    COMPLETE_MULTIPART_ARGS = [
+        'RequestPayer',
+        'ExpectedBucketOwner',
+        'SSECustomerKey',
+        'SSECustomerAlgorithm',
+    ]
 
     def _submit(
         self, client, config, osutil, request_executor, transfer_future

diff --git a/s3transfer/upload.py b/s3transfer/upload.py
@@ -521,7 +521,12 @@ class UploadSubmissionTask(SubmissionTask):
         'ExpectedBucketOwner',
     ]
 
-    COMPLETE_MULTIPART_ARGS = ['RequestPayer', 'ExpectedBucketOwner']
+    COMPLETE_MULTIPART_ARGS = [
+        'RequestPayer',
+        'ExpectedBucketOwner',
+        'SSECustomerKey',
+        'SSECustomerAlgorithm',
+    ]
 
     def _get_upload_input_manager_cls(self, transfer_future):
         """Retrieves a class for managing input for an upload based on file type

diff --git a/tests/unit/test_s3transfer.py b/tests/unit/test_s3transfer.py
@@ -351,6 +351,8 @@ def test_multipart_upload_injects_proper_kwargs(self):
             Bucket='bucket',
             UploadId='upload_id',
             Key='key',
+            SSECustomerKey='fakekey',
+            SSECustomerAlgorithm='AES256',
         )
 
     def test_multipart_upload_is_aborted_on_error(self):