def test_establish_sse_c_copy_same_key_id(self):
file_info = None
content_type = 'text/plain'
metadata_directive, new_file_info, new_content_type = CopyManager.establish_sse_c_file_metadata(
MetadataDirectiveMode.COPY,
destination_file_info=file_info,
destination_content_type=content_type,
destination_server_side_encryption=SSE_C_AES,
source_server_side_encryption=SSE_C_AES,
source_file_info=None,
source_content_type=None,
)
self.assertEqual((MetadataDirectiveMode.COPY, None, content_type),
(metadata_directive, new_file_info, new_content_type))
def test_establish_sse_c_copy_sources_unknown(self):
for source_file_info, source_content_type in [
(None, None),
({
'a': 'b'
}, None),
(None, 'text/plain'),
]:
with self.subTest(source_file_info=source_file_info,
source_content_type=source_content_type):
with self.assertRaises(
SSECKeyIdMismatchInCopy,
'attempting to copy file using MetadataDirectiveMode.COPY without providing source_file_info '
'and source_content_type for differing sse_c_key_ids: source="some-id-2", destination="some-id"'
):
CopyManager.establish_sse_c_file_metadata(
MetadataDirectiveMode.COPY,
destination_file_info=None,
destination_content_type=None,
destination_server_side_encryption=SSE_C_AES,
source_server_side_encryption=SSE_C_AES_2,
source_file_info=source_file_info,
source_content_type=source_content_type,
)
def test_establish_sse_c_replace(self):
file_info = {'some_key': 'some_value'}
content_type = 'text/plain'
metadata_directive, new_file_info, new_content_type = CopyManager.establish_sse_c_file_metadata(
MetadataDirectiveMode.REPLACE,
destination_file_info=file_info,
destination_content_type=content_type,
destination_server_side_encryption=SSE_C_AES,
source_server_side_encryption=SSE_C_AES_2,
source_file_info=None,
source_content_type=None,
)
self.assertEqual((MetadataDirectiveMode.REPLACE, {
'some_key': 'some_value',
SSE_C_KEY_ID_FILE_INFO_KEY_NAME: 'some-id'
}, content_type),
(metadata_directive, new_file_info, new_content_type))