def test_store_file_object():
"""Test to validate list_object method."""
s3 = boto3.client('s3')
s3.create_bucket(Bucket=BUCKET)
S3 = S3Helper(aws_access_key_id=AWS_KEY, aws_secret_access_key=AWS_SECRET)
S3.store_file_object('tests/data/dev/weekly/data.json', BUCKET,
'dev/weekly/data.json')
obj = S3.list_objects(BUCKET, 'weekly')
assert len(obj['objects']) > 0
def test_read_json_object():
"""Test to validate read_json method."""
s3 = boto3.resource('s3')
s3.create_bucket(Bucket=BUCKET)
S3 = S3Helper(aws_access_key_id=AWS_KEY, aws_secret_access_key=AWS_SECRET)
s3.meta.client.upload_file('tests/data/data.json', BUCKET, 'data.json')
data = S3.read_json_object(BUCKET, 'data.json')
assert data.get("key1") == "value1"
data = S3.read_json_object('dummy', 'data.json')
assert data is None
def test_list_objects():
"""Test to validate list_object method."""
s3 = boto3.resource('s3')
s3.create_bucket(Bucket=BUCKET)
S3 = S3Helper(aws_access_key_id=AWS_KEY, aws_secret_access_key=AWS_SECRET)
s3.meta.client.upload_file('tests/data/dev/weekly/data.json', BUCKET,
'dev/weekly/data.json')
obj = S3.list_objects(BUCKET, 'weekly')
assert len(obj['objects']) > 0
data = S3.list_objects('dummy', 'weekly')
assert len(data['objects']) == 0
def test_store_json_content():
"""Test to validate store_json method."""
s3 = boto3.resource('s3')
s3.create_bucket(Bucket=BUCKET)
S3 = S3Helper(aws_access_key_id=AWS_KEY, aws_secret_access_key=AWS_SECRET)
S3.store_json_content({"keyA": "valueB"}, BUCKET, 'dummy.json')
def test_s3_client():
"""Test to validate the s3 client method."""
S3 = S3Helper(aws_access_key_id=AWS_KEY, aws_secret_access_key=AWS_SECRET)
s3 = S3.s3_client(BUCKET)
assert s3
def test_s3_helper():
"""Test to validate the s3_helper constructor function."""
assert S3Helper()