def test_should_copy_all_object_from_latest_to_tag(self):
numeric_version = 0.8
with self.app.app_context():
bit_store = BitStore('test_pub', 'test_package')
s3 = boto3.client('s3')
bucket_name = self.app.config['S3_BUCKET_NAME']
s3.create_bucket(Bucket=bucket_name)
read_me_key = bit_store.build_s3_key('test.md')
data_key = bit_store.build_s3_key('data.csv')
metadata_key = bit_store.build_s3_key('datapackage.json')
s3.put_object(Bucket=bucket_name, Key=read_me_key, Body='readme')
s3.put_object(Bucket=bucket_name, Key=data_key, Body='data')
s3.put_object(Bucket=bucket_name, Key=metadata_key, Body='metedata')
bit_store.copy_to_new_version(numeric_version)
bit_store_numeric = BitStore('test_pub', 'test_package',
numeric_version)
objects_nu = s3.list_objects(Bucket=bucket_name,
Prefix=bit_store_numeric
.build_s3_versioned_prefix())
objects_old = s3.list_objects(Bucket=bucket_name,
Prefix=bit_store
.build_s3_versioned_prefix())
self.assertEqual(len(objects_nu['Contents']),
len(objects_old['Contents']))
def tag_data_package(publisher, package):
"""
DPR metadata put operation.
This API is responsible for tagging data package
---
tags:
- package
parameters:
- in: path
name: publisher
type: string
required: true
description: publisher name
- in: path
name: package
type: string
required: true
description: package name
- in: body
name: version
type: string
required: true
description: version value
- in: header
name: Authorization
type: string
required: true
description: JWT Token
responses:
400:
description: JWT is invalid or req body is not valid
401:
description: Invalid Header for JWT
403:
description: User not allowed for operation
404:
description: User not found
500:
description: Internal Server Error
200:
description: Success Message
schema:
id: put_package_success
properties:
status:
type: string
description: Status of the operation
default: OK
"""
data = request.get_json()
if 'version' not in data:
raise InvalidUsage('version not found', 400)
bitstore = BitStore(publisher, package)
status_db = logic.Package.create_or_update_tag(publisher, package,
data['version'])
try:
status_bitstore = bitstore.copy_to_new_version(data['version'])
except Exception as e:
## TODO roll back changes in db
raise InvalidUsage(e.message, 500)
return jsonify({"status": "OK"}), 200
