87 lines
2.7 KiB
Python
87 lines
2.7 KiB
Python
from dataclasses import dataclass
|
|
from typing import Optional
|
|
|
|
import allure
|
|
from cluster import Cluster
|
|
from file_helper import generate_file, get_file_hash
|
|
from neofs_testlib.shell import Shell
|
|
from neofs_verbs import put_object, put_object_to_random_node
|
|
from storage_object import StorageObjectInfo
|
|
from wallet import WalletFile
|
|
|
|
|
|
@dataclass
|
|
class StorageContainerInfo:
|
|
id: str
|
|
wallet_file: WalletFile
|
|
|
|
|
|
class StorageContainer:
|
|
def __init__(
|
|
self,
|
|
storage_container_info: StorageContainerInfo,
|
|
shell: Shell,
|
|
cluster: Cluster,
|
|
) -> None:
|
|
self.shell = shell
|
|
self.storage_container_info = storage_container_info
|
|
self.cluster = cluster
|
|
|
|
def get_id(self) -> str:
|
|
return self.storage_container_info.id
|
|
|
|
def get_wallet_path(self) -> str:
|
|
return self.storage_container_info.wallet_file.path
|
|
|
|
def get_wallet_config_path(self) -> str:
|
|
return self.storage_container_info.wallet_file.config_path
|
|
|
|
@allure.step("Generate new object and put in container")
|
|
def generate_object(
|
|
self,
|
|
size: int,
|
|
expire_at: Optional[int] = None,
|
|
bearer_token: Optional[str] = None,
|
|
endpoint: Optional[str] = None,
|
|
) -> StorageObjectInfo:
|
|
with allure.step(f"Generate object with size {size}"):
|
|
file_path = generate_file(size)
|
|
file_hash = get_file_hash(file_path)
|
|
|
|
container_id = self.get_id()
|
|
wallet_path = self.get_wallet_path()
|
|
wallet_config = self.get_wallet_config_path()
|
|
with allure.step(f"Put object with size {size} to container {container_id}"):
|
|
if endpoint:
|
|
object_id = put_object(
|
|
wallet=wallet_path,
|
|
path=file_path,
|
|
cid=container_id,
|
|
expire_at=expire_at,
|
|
shell=self.shell,
|
|
endpoint=endpoint,
|
|
bearer=bearer_token,
|
|
wallet_config=wallet_config,
|
|
)
|
|
else:
|
|
object_id = put_object_to_random_node(
|
|
wallet=wallet_path,
|
|
path=file_path,
|
|
cid=container_id,
|
|
expire_at=expire_at,
|
|
shell=self.shell,
|
|
cluster=self.cluster,
|
|
bearer=bearer_token,
|
|
wallet_config=wallet_config,
|
|
)
|
|
|
|
storage_object = StorageObjectInfo(
|
|
container_id,
|
|
object_id,
|
|
size=size,
|
|
wallet_file_path=wallet_path,
|
|
file_path=file_path,
|
|
file_hash=file_hash,
|
|
)
|
|
|
|
return storage_object
|