228 lines
9.4 KiB
Python
228 lines
9.4 KiB
Python
import logging
|
|
import os
|
|
|
|
import allure
|
|
import pytest
|
|
from frostfs_testlib.resources.common import PUBLIC_ACL
|
|
from pytest import FixtureRequest
|
|
|
|
from pytest_tests.helpers.container import (
|
|
create_container,
|
|
delete_container,
|
|
list_containers,
|
|
wait_for_container_deletion,
|
|
)
|
|
from pytest_tests.helpers.file_helper import generate_file
|
|
from pytest_tests.helpers.frostfs_verbs import delete_object
|
|
from pytest_tests.helpers.http_gate import (
|
|
attr_into_str_header_curl,
|
|
get_object_by_attr_and_verify_hashes,
|
|
try_to_get_object_and_expect_error,
|
|
try_to_get_object_via_passed_request_and_expect_error,
|
|
upload_via_http_gate_curl,
|
|
)
|
|
from pytest_tests.helpers.storage_object_info import StorageObjectInfo
|
|
from pytest_tests.steps.cluster_test_base import ClusterTestBase
|
|
|
|
OBJECT_ALREADY_REMOVED_ERROR = "object already removed"
|
|
logger = logging.getLogger("NeoLogger")
|
|
|
|
|
|
@pytest.mark.sanity
|
|
@pytest.mark.http_gate
|
|
class Test_http_headers(ClusterTestBase):
|
|
PLACEMENT_RULE = "REP 2 IN X CBF 1 SELECT 4 FROM * AS X"
|
|
obj1_keys = ["Writer", "Chapter1", "Chapter2"]
|
|
obj2_keys = ["Writer", "Ch@pter1", "chapter2"]
|
|
values = ["Leo Tolstoy", "peace", "w@r"]
|
|
OBJECT_ATTRIBUTES = [
|
|
{obj1_keys[0]: values[0], obj1_keys[1]: values[1], obj1_keys[2]: values[2]},
|
|
{obj2_keys[0]: values[0], obj2_keys[1]: values[1], obj2_keys[2]: values[2]},
|
|
]
|
|
|
|
@pytest.fixture(scope="class", autouse=True)
|
|
@allure.title("[Class/Autouse]: Prepare wallet and deposit")
|
|
def prepare_wallet(self, default_wallet):
|
|
Test_http_headers.wallet = default_wallet
|
|
|
|
@pytest.fixture(
|
|
params=[
|
|
pytest.lazy_fixture("simple_object_size"),
|
|
pytest.lazy_fixture("complex_object_size"),
|
|
],
|
|
ids=["simple object", "complex object"],
|
|
scope="class",
|
|
)
|
|
def storage_objects_with_attributes(self, request: FixtureRequest) -> list[StorageObjectInfo]:
|
|
storage_objects = []
|
|
wallet = self.wallet
|
|
cid = create_container(
|
|
wallet=self.wallet,
|
|
shell=self.shell,
|
|
endpoint=self.cluster.default_rpc_endpoint,
|
|
rule=self.PLACEMENT_RULE,
|
|
basic_acl=PUBLIC_ACL,
|
|
)
|
|
file_path = generate_file(request.param)
|
|
for attributes in self.OBJECT_ATTRIBUTES:
|
|
storage_object_id = upload_via_http_gate_curl(
|
|
cid=cid,
|
|
filepath=file_path,
|
|
endpoint=self.cluster.default_http_gate_endpoint,
|
|
headers=attr_into_str_header_curl(attributes),
|
|
)
|
|
storage_object = StorageObjectInfo(cid, storage_object_id)
|
|
storage_object.size = os.path.getsize(file_path)
|
|
storage_object.wallet_file_path = wallet
|
|
storage_object.file_path = file_path
|
|
storage_object.attributes = attributes
|
|
|
|
storage_objects.append(storage_object)
|
|
|
|
yield storage_objects
|
|
|
|
@allure.title("Get object1 by attribute")
|
|
def test_object1_can_be_get_by_attr(
|
|
self, storage_objects_with_attributes: list[StorageObjectInfo]
|
|
):
|
|
"""
|
|
Test to get object#1 by attribute and comapre hashes
|
|
|
|
Steps:
|
|
1. Download object#1 with attributes [Chapter2=w@r] and compare hashes
|
|
"""
|
|
|
|
storage_object_1 = storage_objects_with_attributes[0]
|
|
|
|
with allure.step(
|
|
f'Download object#1 via wget with attributes Chapter2: {storage_object_1.attributes["Chapter2"]} and compare hashes'
|
|
):
|
|
get_object_by_attr_and_verify_hashes(
|
|
oid=storage_object_1.oid,
|
|
file_name=storage_object_1.file_path,
|
|
cid=storage_object_1.cid,
|
|
attrs={"Chapter2": storage_object_1.attributes["Chapter2"]},
|
|
endpoint=self.cluster.default_http_gate_endpoint,
|
|
)
|
|
|
|
@allure.title("Test get object2 with different attributes, then delete object2 and get object1")
|
|
def test_object2_can_be_get_by_attr(
|
|
self, storage_objects_with_attributes: list[StorageObjectInfo]
|
|
):
|
|
"""
|
|
Test to get object2 with different attributes, then delete object2 and get object1 using 1st attribute. Note: obj1 and obj2 have the same attribute#1,
|
|
and when obj2 is deleted you can get obj1 by 1st attribute
|
|
|
|
Steps:
|
|
1. Download object#2 with attributes [chapter2=w@r] and compare hashes
|
|
2. Download object#2 with attributes [Ch@pter1=peace] and compare hashes
|
|
3. Delete object#2
|
|
4. Download object#1 with attributes [Writer=Leo Tolstoy] and compare hashes
|
|
"""
|
|
storage_object_1 = storage_objects_with_attributes[0]
|
|
storage_object_2 = storage_objects_with_attributes[1]
|
|
|
|
with allure.step(
|
|
f'Download object#2 via wget with attributes [chapter2={storage_object_2.attributes["chapter2"]}] / [Ch@pter1={storage_object_2.attributes["Ch@pter1"]}] and compare hashes'
|
|
):
|
|
selected_attributes_object2 = [
|
|
{"chapter2": storage_object_2.attributes["chapter2"]},
|
|
{"Ch@pter1": storage_object_2.attributes["Ch@pter1"]},
|
|
]
|
|
for attributes in selected_attributes_object2:
|
|
get_object_by_attr_and_verify_hashes(
|
|
oid=storage_object_2.oid,
|
|
file_name=storage_object_2.file_path,
|
|
cid=storage_object_2.cid,
|
|
attrs=attributes,
|
|
endpoint=self.cluster.default_http_gate_endpoint,
|
|
)
|
|
with allure.step("Delete object#2 and verify is the container deleted"):
|
|
delete_object(
|
|
wallet=self.wallet,
|
|
cid=storage_object_2.cid,
|
|
oid=storage_object_2.oid,
|
|
shell=self.shell,
|
|
endpoint=self.cluster.default_rpc_endpoint,
|
|
)
|
|
try_to_get_object_and_expect_error(
|
|
cid=storage_object_2.cid,
|
|
oid=storage_object_2.oid,
|
|
error_pattern=OBJECT_ALREADY_REMOVED_ERROR,
|
|
endpoint=self.cluster.default_http_gate_endpoint,
|
|
)
|
|
storage_objects_with_attributes.remove(storage_object_2)
|
|
|
|
with allure.step(
|
|
f'Download object#1 with attributes [Writer={storage_object_1.attributes["Writer"]}] and compare hashes'
|
|
):
|
|
key_value_pair = {"Writer": storage_object_1.attributes["Writer"]}
|
|
get_object_by_attr_and_verify_hashes(
|
|
oid=storage_object_1.oid,
|
|
file_name=storage_object_1.file_path,
|
|
cid=storage_object_1.cid,
|
|
attrs=key_value_pair,
|
|
endpoint=self.cluster.default_http_gate_endpoint,
|
|
)
|
|
|
|
@allure.title("[Negative] Try to put object and get right after container is deleted")
|
|
def test_negative_put_and_get_object3(
|
|
self, storage_objects_with_attributes: list[StorageObjectInfo]
|
|
):
|
|
"""
|
|
Test to attempt to put object and try to download it right after the container has been deleted
|
|
|
|
Steps:
|
|
1. [Negative] Allocate and attempt to put object#3 via http with attributes: [Writer=Leo Tolstoy, Writer=peace, peace=peace]
|
|
Expected: "Error duplication of attributes detected"
|
|
2. Delete container
|
|
3. [Negative] Try to download object with attributes [peace=peace]
|
|
Expected: "HTTP request sent, awaiting response... 404 Not Found"
|
|
"""
|
|
storage_object_1 = storage_objects_with_attributes[0]
|
|
|
|
with allure.step(
|
|
"[Negative] Allocate and attemt to put object#3 via http with attributes: [Writer=Leo Tolstoy, Writer=peace, peace=peace]"
|
|
):
|
|
file_path_3 = generate_file(storage_object_1.size)
|
|
attrs_obj3 = {"Writer": "Leo Tolstoy", "peace": "peace"}
|
|
headers = attr_into_str_header_curl(attrs_obj3)
|
|
headers.append(" ".join(attr_into_str_header_curl({"Writer": "peace"})))
|
|
error_pattern = f"key duplication error: X-Attribute-Writer"
|
|
upload_via_http_gate_curl(
|
|
cid=storage_object_1.cid,
|
|
filepath=file_path_3,
|
|
endpoint=self.cluster.default_http_gate_endpoint,
|
|
headers=headers,
|
|
error_pattern=error_pattern,
|
|
)
|
|
with allure.step("Delete container and verify container deletion"):
|
|
delete_container(
|
|
wallet=self.wallet,
|
|
cid=storage_object_1.cid,
|
|
shell=self.shell,
|
|
endpoint=self.cluster.default_rpc_endpoint,
|
|
)
|
|
self.tick_epoch()
|
|
wait_for_container_deletion(
|
|
self.wallet,
|
|
storage_object_1.cid,
|
|
shell=self.shell,
|
|
endpoint=self.cluster.default_rpc_endpoint,
|
|
)
|
|
assert storage_object_1.cid not in list_containers(
|
|
self.wallet, shell=self.shell, endpoint=self.cluster.default_rpc_endpoint
|
|
)
|
|
with allure.step(
|
|
"[Negative] Try to download (wget) object via wget with attributes [peace=peace]"
|
|
):
|
|
request = f"/get/{storage_object_1.cid}/peace/peace"
|
|
error_pattern = "404 Not Found"
|
|
try_to_get_object_via_passed_request_and_expect_error(
|
|
cid=storage_object_1.cid,
|
|
oid="",
|
|
error_pattern=error_pattern,
|
|
attrs=attrs_obj3,
|
|
http_request_path=request,
|
|
endpoint=self.cluster.default_http_gate_endpoint,
|
|
)
|