526 lines
28 KiB
Python
526 lines
28 KiB
Python
|
import logging
|
||
|
import os
|
||
|
from random import choice, choices
|
||
|
|
||
|
import allure
|
||
|
import pytest
|
||
|
from common import ASSETS_DIR, COMPLEX_OBJ_SIZE, SIMPLE_OBJ_SIZE
|
||
|
from contract_keywords import tick_epoch
|
||
|
from python_keywords import s3_gate_bucket, s3_gate_object
|
||
|
from python_keywords.aws_cli_client import AwsCliClient
|
||
|
from python_keywords.container import list_containers
|
||
|
from python_keywords.utility_keywords import (generate_file_and_file_hash,
|
||
|
get_file_hash)
|
||
|
from utility import create_file_with_content, get_file_content, split_file
|
||
|
|
||
|
logger = logging.getLogger('NeoLogger')
|
||
|
|
||
|
|
||
|
def pytest_generate_tests(metafunc):
|
||
|
if "s3_client" in metafunc.fixturenames:
|
||
|
metafunc.parametrize("s3_client", ['aws cli', 'boto3'], indirect=True)
|
||
|
|
||
|
|
||
|
@allure.link('https://github.com/nspcc-dev/neofs-s3-gw#neofs-s3-gateway', name='neofs-s3-gateway')
|
||
|
@pytest.mark.s3_gate
|
||
|
class TestS3Gate:
|
||
|
s3_client = None
|
||
|
|
||
|
@pytest.fixture(scope='class', autouse=True)
|
||
|
@allure.title('[Class/Autouse]: Create S3 client')
|
||
|
def s3_client(self, prepare_wallet_and_deposit, request):
|
||
|
wallet, wif = prepare_wallet_and_deposit
|
||
|
s3_bearer_rules_file = f"{os.getcwd()}/robot/resources/files/s3_bearer_rules.json"
|
||
|
|
||
|
cid, bucket, access_key_id, secret_access_key, owner_private_key = \
|
||
|
s3_gate_bucket.init_s3_credentials(wallet, s3_bearer_rules_file=s3_bearer_rules_file)
|
||
|
containers_list = list_containers(wallet)
|
||
|
assert cid in containers_list, f'Expected cid {cid} in {containers_list}'
|
||
|
|
||
|
if request.param == 'aws cli':
|
||
|
try:
|
||
|
client = AwsCliClient(access_key_id, secret_access_key)
|
||
|
except Exception as err:
|
||
|
if 'command was not found or was not executable' in str(err):
|
||
|
pytest.skip('AWS CLI was not found')
|
||
|
else:
|
||
|
raise RuntimeError('Error on creating instance for AwsCliClient') from err
|
||
|
else:
|
||
|
client = s3_gate_bucket.config_s3_client(access_key_id, secret_access_key)
|
||
|
TestS3Gate.s3_client = client
|
||
|
|
||
|
@pytest.fixture
|
||
|
@allure.title('Create two buckets')
|
||
|
def create_buckets(self):
|
||
|
bucket_1 = s3_gate_bucket.create_bucket_s3(self.s3_client)
|
||
|
bucket_2 = s3_gate_bucket.create_bucket_s3(self.s3_client)
|
||
|
return bucket_1, bucket_2
|
||
|
|
||
|
@pytest.fixture
|
||
|
@allure.title('Create/delete bucket')
|
||
|
def bucket(self):
|
||
|
bucket = s3_gate_bucket.create_bucket_s3(self.s3_client)
|
||
|
yield bucket
|
||
|
objects = s3_gate_object.list_objects_s3(self.s3_client, bucket)
|
||
|
if objects:
|
||
|
s3_gate_object.delete_objects_s3(self.s3_client, bucket, objects)
|
||
|
s3_gate_bucket.delete_bucket_s3(self.s3_client, bucket)
|
||
|
|
||
|
@allure.title('Test S3 Bucket API')
|
||
|
def test_s3_buckets(self, generate_files):
|
||
|
"""
|
||
|
Test base S3 Bucket API (Create/List/Head/Delete).
|
||
|
"""
|
||
|
|
||
|
file_name_simple, file_name_large = generate_files
|
||
|
file_name = self.object_key_from_file_path(file_name_simple)
|
||
|
|
||
|
with allure.step('Create buckets'):
|
||
|
bucket_1 = s3_gate_bucket.create_bucket_s3(self.s3_client)
|
||
|
bucket_2 = s3_gate_bucket.create_bucket_s3(self.s3_client)
|
||
|
|
||
|
with allure.step('Check buckets are presented in the system'):
|
||
|
buckets = s3_gate_bucket.list_buckets_s3(self.s3_client)
|
||
|
assert bucket_1 in buckets, f'Expected bucket {bucket_1} is in the list'
|
||
|
assert bucket_2 in buckets, f'Expected bucket {bucket_2} is in the list'
|
||
|
|
||
|
with allure.step('Bucket must be empty'):
|
||
|
for bucket in (bucket_1, bucket_2):
|
||
|
objects_list = s3_gate_object.list_objects_s3(self.s3_client, bucket)
|
||
|
assert not objects_list, f'Expected empty bucket, got {objects_list}'
|
||
|
|
||
|
with allure.step('Check buckets are visible with S3 head command'):
|
||
|
s3_gate_bucket.head_bucket(self.s3_client, bucket_1)
|
||
|
s3_gate_bucket.head_bucket(self.s3_client, bucket_2)
|
||
|
|
||
|
with allure.step('Check we can put/list object with S3 commands'):
|
||
|
s3_gate_object.put_object_s3(self.s3_client, bucket_1, file_name_simple)
|
||
|
s3_gate_object.head_object_s3(self.s3_client, bucket_1, file_name)
|
||
|
|
||
|
bucket_objects = s3_gate_object.list_objects_s3(self.s3_client, bucket_1)
|
||
|
assert file_name in bucket_objects, \
|
||
|
f'Expected file {file_name} in objects list {bucket_objects}'
|
||
|
|
||
|
with allure.step('Try to delete not empty bucket and get error'):
|
||
|
with pytest.raises(Exception, match=r'.*The bucket you tried to delete is not empty.*'):
|
||
|
s3_gate_bucket.delete_bucket_s3(self.s3_client, bucket_1)
|
||
|
|
||
|
s3_gate_bucket.head_bucket(self.s3_client, bucket_1)
|
||
|
|
||
|
with allure.step(f'Delete empty bucket {bucket_2}'):
|
||
|
s3_gate_bucket.delete_bucket_s3(self.s3_client, bucket_2)
|
||
|
tick_epoch()
|
||
|
|
||
|
with allure.step(f'Check bucket {bucket_2} deleted'):
|
||
|
with pytest.raises(Exception, match=r'.*Not Found.*'):
|
||
|
s3_gate_bucket.head_bucket(self.s3_client, bucket_2)
|
||
|
|
||
|
buckets = s3_gate_bucket.list_buckets_s3(self.s3_client)
|
||
|
assert bucket_1 in buckets, f'Expected bucket {bucket_1} is in the list'
|
||
|
assert bucket_2 not in buckets, f'Expected bucket {bucket_2} is not in the list'
|
||
|
|
||
|
@allure.title('Test S3 Object API')
|
||
|
@pytest.mark.sanity
|
||
|
@pytest.mark.parametrize('file_type', ['simple', 'large'], ids=['Simple object', 'Large object'])
|
||
|
def test_s3_api_object(self, generate_files, file_type):
|
||
|
"""
|
||
|
Test base S3 Object API (Put/Head/List) for simple and large objects.
|
||
|
"""
|
||
|
file_name_simple, file_name_large = generate_files
|
||
|
file_name_path = file_name_simple if file_type == 'simple' else file_name_large
|
||
|
file_name = self.object_key_from_file_path(file_name_path)
|
||
|
|
||
|
bucket_1 = s3_gate_bucket.create_bucket_s3(self.s3_client)
|
||
|
bucket_2 = s3_gate_bucket.create_bucket_s3(self.s3_client)
|
||
|
|
||
|
for bucket in (bucket_1, bucket_2):
|
||
|
with allure.step('Bucket must be empty'):
|
||
|
objects_list = s3_gate_object.list_objects_s3(self.s3_client, bucket)
|
||
|
assert not objects_list, f'Expected empty bucket, got {objects_list}'
|
||
|
|
||
|
s3_gate_object.put_object_s3(self.s3_client, bucket, file_name_path)
|
||
|
s3_gate_object.put_object_s3(self.s3_client, bucket, file_name_large)
|
||
|
s3_gate_object.head_object_s3(self.s3_client, bucket, file_name)
|
||
|
|
||
|
bucket_objects = s3_gate_object.list_objects_s3(self.s3_client, bucket)
|
||
|
assert file_name in bucket_objects, \
|
||
|
f'Expected file {file_name} in objects list {bucket_objects}'
|
||
|
|
||
|
with allure.step("Check object's attributes"):
|
||
|
for attrs in (['ETag'], ['ObjectSize', 'StorageClass']):
|
||
|
s3_gate_object.get_object_attributes(self.s3_client, bucket, file_name, *attrs)
|
||
|
|
||
|
@allure.title('Test S3 Sync directory')
|
||
|
def test_s3_sync_dir(self, bucket):
|
||
|
"""
|
||
|
Test checks sync directory with AWS CLI utility.
|
||
|
"""
|
||
|
file_path_1 = f"{os.getcwd()}/{ASSETS_DIR}/test_sync/test_file_1"
|
||
|
file_path_2 = f"{os.getcwd()}/{ASSETS_DIR}/test_sync/test_file_2"
|
||
|
key_to_path = {'test_file_1': file_path_1, 'test_file_2': file_path_2}
|
||
|
|
||
|
if not isinstance(self.s3_client, AwsCliClient):
|
||
|
pytest.skip('This test is not supported with boto3 client')
|
||
|
|
||
|
create_file_with_content(file_path=file_path_1)
|
||
|
create_file_with_content(file_path=file_path_2)
|
||
|
|
||
|
self.s3_client.sync(bucket_name=bucket, dir_path=os.path.dirname(file_path_1))
|
||
|
|
||
|
with allure.step('Check objects are synced'):
|
||
|
objects = s3_gate_object.list_objects_s3(self.s3_client, bucket)
|
||
|
|
||
|
with allure.step('Check these are the same objects'):
|
||
|
assert set(key_to_path.keys()) == set(objects), f'Expected all abjects saved. Got {objects}'
|
||
|
for obj_key in objects:
|
||
|
got_object = s3_gate_object.get_object_s3(self.s3_client, bucket, obj_key)
|
||
|
assert get_file_hash(got_object) == get_file_hash(key_to_path.get(obj_key)), \
|
||
|
'Expected hashes are the same'
|
||
|
|
||
|
@allure.title('Test S3 Object versioning')
|
||
|
def test_s3_api_versioning(self, bucket):
|
||
|
"""
|
||
|
Test checks basic versioning functionality for S3 bucket.
|
||
|
"""
|
||
|
version_1_content = 'Version 1'
|
||
|
version_2_content = 'Version 2'
|
||
|
file_name_simple = create_file_with_content(content=version_1_content)
|
||
|
obj_key = os.path.basename(file_name_simple)
|
||
|
|
||
|
with allure.step('Set versioning enable for bucket'):
|
||
|
s3_gate_bucket.get_bucket_versioning_status(self.s3_client, bucket)
|
||
|
|
||
|
s3_gate_bucket.set_bucket_versioning(self.s3_client, bucket, status=s3_gate_bucket.VersioningStatus.ENABLED)
|
||
|
status = s3_gate_bucket.get_bucket_versioning_status(self.s3_client, bucket)
|
||
|
assert status == s3_gate_bucket.VersioningStatus.ENABLED.value, f'Expected enabled status. Got {status}'
|
||
|
|
||
|
with allure.step('Put several versions of object into bucket'):
|
||
|
version_id_1 = s3_gate_object.put_object_s3(self.s3_client, bucket, file_name_simple)
|
||
|
create_file_with_content(file_path=file_name_simple, content=version_2_content)
|
||
|
version_id_2 = s3_gate_object.put_object_s3(self.s3_client, bucket, file_name_simple)
|
||
|
|
||
|
with allure.step('Check bucket shows all versions'):
|
||
|
versions = s3_gate_object.list_objects_versions_s3(self.s3_client, bucket)
|
||
|
obj_versions = {version.get('VersionId') for version in versions if version.get('Key') == obj_key}
|
||
|
assert obj_versions == {version_id_1, version_id_2}, \
|
||
|
f'Expected object has versions: {version_id_1, version_id_2}'
|
||
|
|
||
|
with allure.step('Show information about particular version'):
|
||
|
for version_id in (version_id_1, version_id_2):
|
||
|
response = s3_gate_object.head_object_s3(self.s3_client, bucket, obj_key, version_id=version_id)
|
||
|
assert 'LastModified' in response, 'Expected LastModified field'
|
||
|
assert 'ETag' in response, 'Expected ETag field'
|
||
|
assert response.get('VersionId') == version_id, f'Expected VersionId is {version_id}'
|
||
|
assert response.get('ContentLength') != 0, 'Expected ContentLength is not zero'
|
||
|
|
||
|
with allure.step("Check object's attributes"):
|
||
|
for version_id in (version_id_1, version_id_2):
|
||
|
got_attrs = s3_gate_object.get_object_attributes(self.s3_client, bucket, obj_key, 'ETag',
|
||
|
version_id=version_id)
|
||
|
if got_attrs:
|
||
|
assert got_attrs.get('VersionId') == version_id, f'Expected VersionId is {version_id}'
|
||
|
|
||
|
with allure.step('Delete object and check it was deleted'):
|
||
|
response = s3_gate_object.delete_object_s3(self.s3_client, bucket, obj_key)
|
||
|
version_id_delete = response.get('VersionId')
|
||
|
|
||
|
with pytest.raises(Exception, match=r'.*Not Found.*'):
|
||
|
s3_gate_object.head_object_s3(self.s3_client, bucket, obj_key)
|
||
|
|
||
|
with allure.step('Get content for all versions and check it is correct'):
|
||
|
for version, content in ((version_id_2, version_2_content), (version_id_1, version_1_content)):
|
||
|
file_name = s3_gate_object.get_object_s3(self.s3_client, bucket, obj_key, version_id=version)
|
||
|
got_content = get_file_content(file_name)
|
||
|
assert got_content == content, f'Expected object content is\n{content}\nGot\n{got_content}'
|
||
|
|
||
|
with allure.step('Restore previous object version'):
|
||
|
s3_gate_object.delete_object_s3(self.s3_client, bucket, obj_key, version_id=version_id_delete)
|
||
|
|
||
|
file_name = s3_gate_object.get_object_s3(self.s3_client, bucket, obj_key)
|
||
|
got_content = get_file_content(file_name)
|
||
|
assert got_content == version_2_content, \
|
||
|
f'Expected object content is\n{version_2_content}\nGot\n{got_content}'
|
||
|
|
||
|
@allure.title('Test S3 Object Multipart API')
|
||
|
def test_s3_api_multipart(self, bucket):
|
||
|
"""
|
||
|
Test checks S3 Multipart API (Create multipart upload/Abort multipart upload/List multipart upload/
|
||
|
Upload part/List parts/Complete multipart upload).
|
||
|
"""
|
||
|
parts_count = 3
|
||
|
file_name_large, _ = generate_file_and_file_hash(SIMPLE_OBJ_SIZE * 1024 * 6 * parts_count) # 5Mb - min part
|
||
|
# file_name_large, _ = generate_file_and_file_hash(SIMPLE_OBJ_SIZE * 1024 * 30 * parts_count) # 5Mb - min part
|
||
|
object_key = self.object_key_from_file_path(file_name_large)
|
||
|
part_files = split_file(file_name_large, parts_count)
|
||
|
parts = []
|
||
|
|
||
|
uploads = s3_gate_object.list_multipart_uploads_s3(self.s3_client, bucket)
|
||
|
assert not uploads, f'Expected there is no uploads in bucket {bucket}'
|
||
|
|
||
|
with allure.step('Create and abort multipart upload'):
|
||
|
upload_id = s3_gate_object.create_multipart_upload_s3(self.s3_client, bucket, object_key)
|
||
|
uploads = s3_gate_object.list_multipart_uploads_s3(self.s3_client, bucket)
|
||
|
assert uploads, f'Expected there one upload in bucket {bucket}'
|
||
|
assert uploads[0].get('Key') == object_key, f'Expected correct key {object_key} in upload {uploads}'
|
||
|
assert uploads[0].get('UploadId') == upload_id, f'Expected correct UploadId {upload_id} in upload {uploads}'
|
||
|
|
||
|
s3_gate_object.abort_multipart_uploads_s3(self.s3_client, bucket, object_key, upload_id)
|
||
|
uploads = s3_gate_object.list_multipart_uploads_s3(self.s3_client, bucket)
|
||
|
assert not uploads, f'Expected there is no uploads in bucket {bucket}'
|
||
|
|
||
|
with allure.step('Create new multipart upload and upload several parts'):
|
||
|
upload_id = s3_gate_object.create_multipart_upload_s3(self.s3_client, bucket, object_key)
|
||
|
for part_id, file_path in enumerate(part_files, start=1):
|
||
|
etag = s3_gate_object.upload_part_s3(self.s3_client, bucket, object_key, upload_id, part_id, file_path)
|
||
|
parts.append((part_id, etag))
|
||
|
|
||
|
with allure.step('Check all parts are visible in bucket'):
|
||
|
got_parts = s3_gate_object.list_parts_s3(self.s3_client, bucket, object_key, upload_id)
|
||
|
assert len(got_parts) == len(part_files), f'Expected {parts_count} parts, got\n{got_parts}'
|
||
|
|
||
|
s3_gate_object.complete_multipart_upload_s3(self.s3_client, bucket, object_key, upload_id, parts)
|
||
|
|
||
|
uploads = s3_gate_object.list_multipart_uploads_s3(self.s3_client, bucket)
|
||
|
assert not uploads, f'Expected there is no uploads in bucket {bucket}'
|
||
|
|
||
|
with allure.step('Check we can get whole object from bucket'):
|
||
|
got_object = s3_gate_object.get_object_s3(self.s3_client, bucket, object_key)
|
||
|
assert get_file_hash(got_object) == get_file_hash(file_name_large)
|
||
|
|
||
|
self.check_object_attributes(bucket, object_key, parts_count)
|
||
|
|
||
|
@allure.title('Test S3 Bucket tagging API')
|
||
|
def test_s3_api_bucket_tagging(self, bucket):
|
||
|
"""
|
||
|
Test checks S3 Bucket tagging API (Put tag/Get tag).
|
||
|
"""
|
||
|
key_value_pair = [('some-key', 'some-value'), ('some-key-2', 'some-value-2')]
|
||
|
|
||
|
s3_gate_bucket.put_bucket_tagging(self.s3_client, bucket, key_value_pair)
|
||
|
got_tags = s3_gate_bucket.get_bucket_tagging(self.s3_client, bucket)
|
||
|
|
||
|
with allure.step('Check all tags are presented'):
|
||
|
assert got_tags, f'Expected tags, got {got_tags}'
|
||
|
expected_tags = [{'Key': key, 'Value': value} for key, value in key_value_pair]
|
||
|
for tag in expected_tags:
|
||
|
assert tag in got_tags
|
||
|
|
||
|
s3_gate_bucket.delete_bucket_tagging(self.s3_client, bucket)
|
||
|
tags = s3_gate_bucket.get_bucket_tagging(self.s3_client, bucket)
|
||
|
assert not tags, f'Expected there is no tags for bucket {bucket}, got {tags}'
|
||
|
|
||
|
@allure.title('Test S3 Object tagging API')
|
||
|
def test_s3_api_object_tagging(self, bucket):
|
||
|
"""
|
||
|
Test checks S3 Object tagging API (Put tag/Get tag/Update tag).
|
||
|
"""
|
||
|
key_value_pair_bucket = [('some-key', 'some-value'), ('some-key-2', 'some-value-2')]
|
||
|
key_value_pair_obj = [('some-key-obj', 'some-value-obj'), ('some-key--obj2', 'some-value--obj2')]
|
||
|
key_value_pair_obj_new = [('some-key-obj-new', 'some-value-obj-new')]
|
||
|
file_name_simple, _ = generate_file_and_file_hash(SIMPLE_OBJ_SIZE)
|
||
|
obj_key = self.object_key_from_file_path(file_name_simple)
|
||
|
|
||
|
s3_gate_bucket.put_bucket_tagging(self.s3_client, bucket, key_value_pair_bucket)
|
||
|
|
||
|
s3_gate_object.put_object_s3(self.s3_client, bucket, file_name_simple)
|
||
|
|
||
|
for tags in (key_value_pair_obj, key_value_pair_obj_new):
|
||
|
s3_gate_object.put_object_tagging(self.s3_client, bucket, obj_key, tags)
|
||
|
|
||
|
got_tags = s3_gate_object.get_object_tagging(self.s3_client, bucket, obj_key)
|
||
|
assert got_tags, f'Expected tags, got {got_tags}'
|
||
|
expected_tags = [{'Key': key, 'Value': value} for key, value in tags]
|
||
|
for tag in expected_tags:
|
||
|
assert tag in got_tags
|
||
|
|
||
|
s3_gate_object.delete_object_tagging(self.s3_client, bucket, obj_key)
|
||
|
got_tags = s3_gate_object.get_object_tagging(self.s3_client, bucket, obj_key)
|
||
|
assert not got_tags, f'Expected there is no tags for bucket {bucket}, got {got_tags}'
|
||
|
|
||
|
@allure.title('Test S3: Delete object & delete objects S3 API')
|
||
|
def test_s3_api_delete(self, create_buckets):
|
||
|
"""
|
||
|
Check delete_object and delete_objects S3 API operation. From first bucket some objects deleted one by one.
|
||
|
From second bucket some objects deleted all at once.
|
||
|
"""
|
||
|
max_obj_count = 20
|
||
|
max_delete_objects = 17
|
||
|
put_objects = []
|
||
|
file_paths = []
|
||
|
obj_sizes = [SIMPLE_OBJ_SIZE, COMPLEX_OBJ_SIZE]
|
||
|
|
||
|
bucket_1, bucket_2 = create_buckets
|
||
|
|
||
|
with allure.step(f'Generate {max_obj_count} files'):
|
||
|
for _ in range(max_obj_count):
|
||
|
file_paths.append(generate_file_and_file_hash(choice(obj_sizes))[0])
|
||
|
|
||
|
for bucket in (bucket_1, bucket_2):
|
||
|
with allure.step(f'Bucket {bucket} must be empty as it just created'):
|
||
|
objects_list = s3_gate_object.list_objects_s3_v2(self.s3_client, bucket)
|
||
|
assert not objects_list, f'Expected empty bucket, got {objects_list}'
|
||
|
|
||
|
for file_path in file_paths:
|
||
|
s3_gate_object.put_object_s3(self.s3_client, bucket, file_path)
|
||
|
put_objects.append(self.object_key_from_file_path(file_path))
|
||
|
|
||
|
with allure.step(f'Check all objects put in bucket {bucket} successfully'):
|
||
|
bucket_objects = s3_gate_object.list_objects_s3_v2(self.s3_client, bucket)
|
||
|
assert set(put_objects) == set(bucket_objects), \
|
||
|
f'Expected all objects {put_objects} in objects list {bucket_objects}'
|
||
|
|
||
|
with allure.step('Delete some objects from bucket_1 one by one'):
|
||
|
objects_to_delete_b1 = choices(put_objects, k=max_delete_objects)
|
||
|
for obj in objects_to_delete_b1:
|
||
|
s3_gate_object.delete_object_s3(self.s3_client, bucket_1, obj)
|
||
|
|
||
|
with allure.step('Check deleted objects are not visible in bucket bucket_1'):
|
||
|
bucket_objects = s3_gate_object.list_objects_s3_v2(self.s3_client, bucket_1)
|
||
|
assert set(put_objects).difference(set(objects_to_delete_b1)) == set(bucket_objects), \
|
||
|
f'Expected all objects {put_objects} in objects list {bucket_objects}'
|
||
|
self.try_to_get_object_and_got_error(bucket_1, objects_to_delete_b1)
|
||
|
|
||
|
with allure.step('Delete some objects from bucket_2 at once'):
|
||
|
objects_to_delete_b2 = choices(put_objects, k=max_delete_objects)
|
||
|
s3_gate_object.delete_objects_s3(self.s3_client, bucket_2, objects_to_delete_b2)
|
||
|
|
||
|
with allure.step('Check deleted objects are not visible in bucket bucket_2'):
|
||
|
objects_list = s3_gate_object.list_objects_s3_v2(self.s3_client, bucket_2)
|
||
|
assert set(put_objects).difference(set(objects_to_delete_b2)) == set(objects_list), \
|
||
|
f'Expected all objects {put_objects} in objects list {bucket_objects}'
|
||
|
self.try_to_get_object_and_got_error(bucket_2, objects_to_delete_b2)
|
||
|
|
||
|
@allure.title('Test S3: Copy object to the same bucket')
|
||
|
def test_s3_copy_same_bucket(self, generate_files):
|
||
|
"""
|
||
|
Test object can be copied to the same bucket.
|
||
|
"""
|
||
|
file_simple, file_large = generate_files
|
||
|
file_name_simple = self.object_key_from_file_path(file_simple)
|
||
|
file_name_large = self.object_key_from_file_path(file_large)
|
||
|
bucket_objects = [file_name_simple, file_name_large]
|
||
|
|
||
|
bucket = s3_gate_bucket.create_bucket_s3(self.s3_client)
|
||
|
|
||
|
with allure.step('Bucket must be empty'):
|
||
|
objects_list = s3_gate_object.list_objects_s3(self.s3_client, bucket)
|
||
|
assert not objects_list, f'Expected empty bucket, got {objects_list}'
|
||
|
|
||
|
with allure.step('Put objects into bucket'):
|
||
|
for obj in (file_simple, file_large):
|
||
|
s3_gate_object.put_object_s3(self.s3_client, bucket, obj)
|
||
|
|
||
|
with allure.step('Copy one object into the same bucket'):
|
||
|
copy_obj_path = s3_gate_object.copy_object_s3(self.s3_client, bucket, file_name_simple)
|
||
|
bucket_objects.append(copy_obj_path)
|
||
|
|
||
|
self.check_objects_in_bucket(bucket, bucket_objects)
|
||
|
|
||
|
with allure.step('Check copied object has the same content'):
|
||
|
got_copied_file = s3_gate_object.get_object_s3(self.s3_client, bucket, copy_obj_path)
|
||
|
assert get_file_hash(file_simple) == get_file_hash(got_copied_file), 'Hashes must be the same'
|
||
|
|
||
|
with allure.step('Delete one object from bucket'):
|
||
|
s3_gate_object.delete_object_s3(self.s3_client, bucket, file_name_simple)
|
||
|
bucket_objects.remove(file_name_simple)
|
||
|
|
||
|
self.check_objects_in_bucket(bucket, expected_objects=bucket_objects, unexpected_objects=[file_name_simple])
|
||
|
|
||
|
@allure.title('Test S3: Copy object to another bucket')
|
||
|
def test_s3_copy_to_another_bucket(self, generate_files):
|
||
|
"""
|
||
|
Test object can be copied to another bucket.
|
||
|
"""
|
||
|
file_simple, file_large = generate_files
|
||
|
file_name_simple = self.object_key_from_file_path(file_simple)
|
||
|
file_name_large = self.object_key_from_file_path(file_large)
|
||
|
bucket_1_objects = [file_name_simple, file_name_large]
|
||
|
|
||
|
bucket_1 = s3_gate_bucket.create_bucket_s3(self.s3_client)
|
||
|
bucket_2 = s3_gate_bucket.create_bucket_s3(self.s3_client)
|
||
|
|
||
|
with allure.step('Buckets must be empty'):
|
||
|
for bucket in (bucket_1, bucket_2):
|
||
|
objects_list = s3_gate_object.list_objects_s3(self.s3_client, bucket)
|
||
|
assert not objects_list, f'Expected empty bucket, got {objects_list}'
|
||
|
|
||
|
with allure.step('Put objects into one bucket'):
|
||
|
for obj in (file_simple, file_large):
|
||
|
s3_gate_object.put_object_s3(self.s3_client, bucket_1, obj)
|
||
|
|
||
|
with allure.step('Copy object from first bucket into second'):
|
||
|
copy_obj_path_b2 = s3_gate_object.copy_object_s3(self.s3_client, bucket_1, file_name_large,
|
||
|
bucket_dst=bucket_2)
|
||
|
self.check_objects_in_bucket(bucket_1, expected_objects=bucket_1_objects)
|
||
|
self.check_objects_in_bucket(bucket_2, expected_objects=[copy_obj_path_b2])
|
||
|
|
||
|
with allure.step('Check copied object has the same content'):
|
||
|
got_copied_file_b2 = s3_gate_object.get_object_s3(self.s3_client, bucket_2, copy_obj_path_b2)
|
||
|
assert get_file_hash(file_large) == get_file_hash(got_copied_file_b2), 'Hashes must be the same'
|
||
|
|
||
|
with allure.step('Delete one object from first bucket'):
|
||
|
s3_gate_object.delete_object_s3(self.s3_client, bucket_1, file_name_simple)
|
||
|
bucket_1_objects.remove(file_name_simple)
|
||
|
|
||
|
self.check_objects_in_bucket(bucket_1, expected_objects=bucket_1_objects)
|
||
|
self.check_objects_in_bucket(bucket_2, expected_objects=[copy_obj_path_b2])
|
||
|
|
||
|
with allure.step('Delete one object from second bucket and check it is empty'):
|
||
|
s3_gate_object.delete_object_s3(self.s3_client, bucket_2, copy_obj_path_b2)
|
||
|
self.check_objects_in_bucket(bucket_2, expected_objects=[])
|
||
|
|
||
|
def check_object_attributes(self, bucket: str, object_key: str, parts_count: int):
|
||
|
if not isinstance(self.s3_client, AwsCliClient):
|
||
|
logger.warning('Attributes check is not supported for boto3 implementation')
|
||
|
return
|
||
|
|
||
|
with allure.step("Check object's attributes"):
|
||
|
obj_parts = s3_gate_object.get_object_attributes(self.s3_client, bucket, object_key, 'ObjectParts',
|
||
|
get_full_resp=False)
|
||
|
assert obj_parts.get('TotalPartsCount') == parts_count, f'Expected TotalPartsCount is {parts_count}'
|
||
|
assert len(obj_parts.get('Parts')) == parts_count, f'Expected Parts cunt is {parts_count}'
|
||
|
|
||
|
with allure.step("Check object's attribute max-parts"):
|
||
|
max_parts = 2
|
||
|
obj_parts = s3_gate_object.get_object_attributes(self.s3_client, bucket, object_key, 'ObjectParts',
|
||
|
max_parts=max_parts, get_full_resp=False)
|
||
|
assert obj_parts.get('TotalPartsCount') == parts_count, f'Expected TotalPartsCount is {parts_count}'
|
||
|
assert obj_parts.get('MaxParts') == max_parts, f'Expected MaxParts is {parts_count}'
|
||
|
assert len(obj_parts.get('Parts')) == max_parts, f'Expected Parts count is {parts_count}'
|
||
|
|
||
|
with allure.step("Check object's attribute part-number-marker"):
|
||
|
part_number_marker = 3
|
||
|
obj_parts = s3_gate_object.get_object_attributes(self.s3_client, bucket, object_key, 'ObjectParts',
|
||
|
part_number=part_number_marker, get_full_resp=False)
|
||
|
assert obj_parts.get('TotalPartsCount') == parts_count, f'Expected TotalPartsCount is {parts_count}'
|
||
|
assert obj_parts.get(
|
||
|
'PartNumberMarker') == part_number_marker, f'Expected PartNumberMarker is {part_number_marker}'
|
||
|
assert len(obj_parts.get('Parts')) == 1, f'Expected Parts count is {parts_count}'
|
||
|
|
||
|
@allure.step('Expected all objects are presented in the bucket')
|
||
|
def check_objects_in_bucket(self, bucket, expected_objects: list, unexpected_objects: list = None):
|
||
|
unexpected_objects = unexpected_objects or []
|
||
|
bucket_objects = s3_gate_object.list_objects_s3(self.s3_client, bucket)
|
||
|
assert len(bucket_objects) == len(expected_objects), f'Expected {len(expected_objects)} objects in the bucket'
|
||
|
|
||
|
for bucket_object in expected_objects:
|
||
|
assert bucket_object in bucket_objects, \
|
||
|
f'Expected object {bucket_object} in objects list {bucket_objects}'
|
||
|
|
||
|
for bucket_object in unexpected_objects:
|
||
|
assert bucket_object not in bucket_objects, \
|
||
|
f'Expected object {bucket_object} not in objects list {bucket_objects}'
|
||
|
|
||
|
@allure.step('Try to get object and got error')
|
||
|
def try_to_get_object_and_got_error(self, bucket: str, unexpected_objects: list):
|
||
|
for obj in unexpected_objects:
|
||
|
try:
|
||
|
s3_gate_object.get_object_s3(self.s3_client, bucket, obj)
|
||
|
raise AssertionError(f'Object {obj} found in bucket {bucket}')
|
||
|
except Exception as err:
|
||
|
assert 'The specified key does not exist' in str(err), f'Expected error in exception {err}'
|
||
|
|
||
|
@staticmethod
|
||
|
def object_key_from_file_path(full_path: str) -> str:
|
||
|
return os.path.basename(full_path)
|