def finish(self):
self._write_to_s3(self.buffer)
if self.last_error is not None:
raise self.last_error # pylint: disable=raising-bad-type
request = messages.CompleteMultipartUploadRequest(
self._bucket, self._name, self.upload_id, self.parts)
self._client.complete_multipart_upload(request)
def finish(self):
if len(self.buffer) > 0:
# writes with zero length or mid size files between
# MIN_WRITE_SIZE = 5 * 1024 * 1024
# MAX_WRITE_SIZE = 5 * 1024 * 1024 * 1024
# as we will reach this finish() with len(self.buffer) == 0
# which will fail
self._write_to_s3(self.buffer)
if self.last_error is not None:
raise self.last_error # pylint: disable=raising-bad-type
request = messages.CompleteMultipartUploadRequest(
self._bucket, self._name, self.upload_id, self.parts)
self._client.complete_multipart_upload(request)
def test_complete_multipart_upload_too_many(self):
object = self.test_path + 'upload_part'
request = messages.UploadRequest(self.test_bucket, object, None)
response = self.client.create_multipart_upload(request)
upload_id = response.upload_id
part_number = 1
contents_1 = os.urandom(5 * 1024)
request_1 = messages.UploadPartRequest(self.test_bucket, object,
upload_id, part_number,
contents_1)
response_1 = self.client.upload_part(request_1)
part_number = 2
contents_2 = os.urandom(1024)
request_2 = messages.UploadPartRequest(self.test_bucket, object,
upload_id, part_number,
contents_2)
response_2 = self.client.upload_part(request_2)
parts = [
{
'PartNumber': 1,
'ETag': response_1.etag
},
{
'PartNumber': 2,
'ETag': response_2.etag
},
{
'PartNumber': 3,
'ETag': 'fake-etag'
},
]
complete_request = messages.CompleteMultipartUploadRequest(
self.test_bucket, object, upload_id, parts)
try:
self.client.complete_multipart_upload(complete_request)
except Exception as e:
self.assertIsInstance(e, messages.S3ClientError)
self.assertEqual(e.code, 400)
