def test_upload_part_bad_number(self):
object = self.test_path + 'upload_part'
contents = os.urandom(1024)
request = messages.UploadRequest(self.test_bucket, object, None)
response = self.client.create_multipart_upload(request)
upload_id = response.upload_id
part_number = 0.5
request = messages.UploadPartRequest(self.test_bucket,
object,
upload_id,
part_number,
contents)
self.assertRaises(messages.S3ClientError,
self.client.upload_part,
request)
try:
response = self.client.upload_part(request)
except Exception as e:
self.assertIsInstance(e, messages.S3ClientError)
self.assertEqual(e.code, 400)
def _start_upload(self):
# The uploader by default transfers data in chunks of 1024 * 1024 bytes at
# a time, buffering writes until that size is reached.
try:
request = messages.UploadRequest(self._bucket, self._name,
self._mime_type)
response = self._client.create_multipart_upload(request)
self.upload_id = response.upload_id
except Exception as e: # pylint: disable=broad-except
logging.error('Error in _start_upload while inserting file %s: %s',
self._path, traceback.format_exc())
self.last_error = e
raise e
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)