def test_resumable_upload_with_headers(authorized_transport, img_stream,
bucket, cleanup):
headers = utils.get_encryption_headers()
_resumable_upload_helper(authorized_transport,
img_stream,
cleanup,
headers=headers)
async def test_multipart_upload_with_headers(authorized_transport, bucket,
cleanup):
blob_name = u"some-multipart-stuff.bin"
# Make sure to clean up the uploaded blob when we are done.
await cleanup(blob_name, authorized_transport)
await check_does_not_exist(authorized_transport, blob_name)
# Create the actual upload object.
upload_url = utils.MULTIPART_UPLOAD
headers = utils.get_encryption_headers()
upload = resumable_requests.MultipartUpload(upload_url, headers=headers)
# Transmit the resource.
metadata = {u"name": blob_name}
data = b"Other binary contents\x03\x04\x05."
response = await upload.transmit(authorized_transport, data, metadata,
BYTES_CONTENT_TYPE)
await check_response(response,
blob_name,
actual_contents=data,
content_type=BYTES_CONTENT_TYPE)
# Download the content to make sure it's "working as expected".
await check_content(blob_name, data, authorized_transport, headers=headers)
# Make sure the upload is tombstoned.
await check_tombstoned(upload, authorized_transport, data, metadata,
BYTES_CONTENT_TYPE)
def secret_file(authorized_transport, bucket):
blob_name = u"super-seekrit.txt"
data = b"Please do not tell anyone my encrypted seekrit."
upload_url = utils.SIMPLE_UPLOAD_TEMPLATE.format(blob_name=blob_name)
headers = utils.get_encryption_headers()
upload = resumable_requests.SimpleUpload(upload_url, headers=headers)
response = upload.transmit(authorized_transport, data, PLAIN_TEXT)
assert response.status_code == http_client.OK
yield blob_name, data, headers
delete_blob(authorized_transport, blob_name)
def test_simple_upload_with_headers(authorized_transport, bucket, cleanup):
blob_name = u"some-stuff.bin"
# Make sure to clean up the uploaded blob when we are done.
cleanup(blob_name, authorized_transport)
check_does_not_exist(authorized_transport, blob_name)
# Create the actual upload object.
upload_url = utils.SIMPLE_UPLOAD_TEMPLATE.format(blob_name=blob_name)
headers = utils.get_encryption_headers()
upload = resumable_requests.SimpleUpload(upload_url, headers=headers)
# Transmit the resource.
data = b"Binary contents\x00\x01\x02."
response = upload.transmit(authorized_transport, data, BYTES_CONTENT_TYPE)
check_response(
response, blob_name, actual_contents=data, content_type=BYTES_CONTENT_TYPE
)
# Download the content to make sure it's "working as expected".
check_content(blob_name, data, authorized_transport, headers=headers)
# Make sure the upload is tombstoned.
check_tombstoned(upload, authorized_transport, data, BYTES_CONTENT_TYPE)
async def test_resumable_upload_recover_with_headers(authorized_transport,
bucket, cleanup):
headers = utils.get_encryption_headers()
await _resumable_upload_recover_helper(authorized_transport,
cleanup,
headers=headers)
