def add_files(authorized_transport, bucket):
blob_names = []
for info in ALL_FILES:
to_upload = get_contents_for_upload(info)
blob_name = get_blob_name(info)
blob_names.append(blob_name)
if u"metadata" in info:
upload = resumable_requests.MultipartUpload(utils.MULTIPART_UPLOAD)
metadata = copy.deepcopy(info[u"metadata"])
metadata[u"name"] = blob_name
response = upload.transmit(authorized_transport, to_upload,
metadata, info[u"content_type"])
else:
upload_url = utils.SIMPLE_UPLOAD_TEMPLATE.format(
blob_name=blob_name)
upload = resumable_requests.SimpleUpload(upload_url)
response = upload.transmit(authorized_transport, to_upload,
info[u"content_type"])
assert response.status_code == http_client.OK
yield
# Clean-up the blobs we created.
for blob_name in blob_names:
delete_blob(authorized_transport, blob_name)
def simple_file(authorized_transport, bucket):
blob_name = u"basic-file.txt"
upload_url = utils.SIMPLE_UPLOAD_TEMPLATE.format(blob_name=blob_name)
upload = resumable_requests.SimpleUpload(upload_url)
data = b"Simple contents"
response = upload.transmit(authorized_transport, data, PLAIN_TEXT)
assert response.status_code == http_client.OK
yield blob_name, data
delete_blob(authorized_transport, blob_name)
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(authorized_transport, bucket, cleanup):
with open(ICO_FILE, u"rb") as file_obj:
actual_contents = file_obj.read()
blob_name = os.path.basename(ICO_FILE)
# 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)
upload = resumable_requests.SimpleUpload(upload_url)
# Transmit the resource.
response = upload.transmit(authorized_transport, actual_contents, ICO_CONTENT_TYPE)
check_response(response, blob_name, actual_contents=actual_contents)
# Download the content to make sure it's "working as expected".
check_content(blob_name, actual_contents, authorized_transport)
# Make sure the upload is tombstoned.
check_tombstoned(upload, authorized_transport, actual_contents, ICO_CONTENT_TYPE)
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)
