def insert_magic_bucket(base_url):
if len(testbench_utils.all_buckets()) == 0:
bucket_name = os.environ.get(
"GOOGLE_CLOUD_CPP_STORAGE_TEST_BUCKET_NAME", "test-bucket"
)
bucket = gcs_bucket.GcsBucket(base_url, bucket_name)
testbench_utils.insert_bucket(bucket_name, bucket)
def insert_magic_bucket(base_url):
if len(testbench_utils.all_buckets()) == 0:
bucket_name = os.environ.get('BUCKET_NAME', 'test-bucket')
bucket = gcs_bucket.GcsBucket(base_url, bucket_name)
# Enable versioning in the Bucket, the integration tests expect this to
# be the case, this brings the metageneration number to 2.
bucket.update_from_metadata({'versioning': {'enabled': True}})
# Perform trivial updates that bring the metageneration to 4, the value
# expected by the integration tests.
bucket.update_from_metadata({})
bucket.update_from_metadata({})
testbench_utils.insert_bucket(bucket_name, bucket)
def insert_magic_bucket(base_url):
if len(testbench_utils.all_buckets()) == 0:
bucket_name = os.environ.get(
"GOOGLE_CLOUD_CPP_STORAGE_TEST_BUCKET_NAME", "test-bucket")
bucket = gcs_bucket.GcsBucket(base_url, bucket_name)
# Enable versioning in the Bucket, the integration tests expect this to
# be the case, this brings the metageneration number to 2.
bucket.update_from_metadata({"versioning": {"enabled": True}})
# Perform trivial updates that bring the metageneration to 4, the value
# expected by the integration tests.
bucket.update_from_metadata({})
bucket.update_from_metadata({})
testbench_utils.insert_bucket(bucket_name, bucket)
def buckets_insert():
"""Implement the 'Buckets: insert' API: create a new Bucket."""
base_url = flask.url_for("gcs_index", _external=True)
insert_magic_bucket(base_url)
payload = json.loads(flask.request.data)
bucket_name = payload.get("name")
if bucket_name is None:
raise error_response.ErrorResponse(
"Missing bucket name in `Buckets: insert`", status_code=412)
if testbench_utils.has_bucket(bucket_name):
raise error_response.ErrorResponse("Bucket %s already exists" %
bucket_name,
status_code=400)
bucket = gcs_bucket.GcsBucket(base_url, bucket_name)
testbench_utils.insert_bucket(bucket_name, bucket)
return testbench_utils.filtered_response(flask.request, bucket.metadata)
def buckets_insert():
"""Implement the 'Buckets: insert' API: create a new Bucket."""
base_url = flask.url_for('gcs_index', _external=True)
insert_magic_bucket(base_url)
payload = json.loads(flask.request.data)
bucket_name = payload.get('name')
if bucket_name is None:
raise error_response.ErrorResponse(
'Missing bucket name in `Buckets: insert`', status_code=412)
if not testbench_utils.validate_bucket_name(bucket_name):
raise error_response.ErrorResponse(
'Invalid bucket name in `Buckets: insert`')
if testbench_utils.has_bucket(bucket_name):
raise error_response.ErrorResponse(
'Bucket %s already exists' % bucket_name, status_code=400)
bucket = gcs_bucket.GcsBucket(base_url, bucket_name)
bucket.update_from_metadata(payload)
testbench_utils.insert_bucket(bucket_name, bucket)
return testbench_utils.filtered_response(flask.request, bucket.metadata)
