forked from TrueCloudLab/frostfs-testlib
758 lines
24 KiB
Python
758 lines
24 KiB
Python
import json
|
|
import logging
|
|
import os
|
|
import re
|
|
import uuid
|
|
from typing import Any, Optional
|
|
|
|
from frostfs_testlib import reporter
|
|
from frostfs_testlib.cli import FrostfsCli
|
|
from frostfs_testlib.cli.neogo import NeoGo
|
|
from frostfs_testlib.resources.cli import CLI_DEFAULT_TIMEOUT, FROSTFS_CLI_EXEC, NEOGO_EXECUTABLE
|
|
from frostfs_testlib.resources.common import ASSETS_DIR
|
|
from frostfs_testlib.shell import Shell
|
|
from frostfs_testlib.storage.cluster import Cluster, ClusterNode
|
|
from frostfs_testlib.storage.dataclasses.wallet import WalletInfo
|
|
from frostfs_testlib.testing import wait_for_success
|
|
from frostfs_testlib.utils import json_utils
|
|
from frostfs_testlib.utils.cli_utils import parse_netmap_output
|
|
from frostfs_testlib.utils.file_utils import TestFile
|
|
|
|
logger = logging.getLogger("NeoLogger")
|
|
|
|
|
|
@reporter.step("Get object from random node")
|
|
def get_object_from_random_node(
|
|
wallet: WalletInfo,
|
|
cid: str,
|
|
oid: str,
|
|
shell: Shell,
|
|
cluster: Cluster,
|
|
bearer: Optional[str] = None,
|
|
write_object: Optional[str] = None,
|
|
xhdr: Optional[dict] = None,
|
|
no_progress: bool = True,
|
|
session: Optional[str] = None,
|
|
timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
|
|
) -> str:
|
|
"""
|
|
GET from FrostFS random storage node
|
|
|
|
Args:
|
|
wallet: wallet on whose behalf GET is done
|
|
cid: ID of Container where we get the Object from
|
|
oid: Object ID
|
|
shell: executor for cli command
|
|
cluster: cluster object
|
|
bearer (optional, str): path to Bearer Token file, appends to `--bearer` key
|
|
write_object (optional, str): path to downloaded file, appends to `--file` key
|
|
no_progress(optional, bool): do not show progress bar
|
|
xhdr (optional, dict): Request X-Headers in form of Key=Value
|
|
session (optional, dict): path to a JSON-encoded container session token
|
|
timeout: Timeout for the operation.
|
|
Returns:
|
|
(str): path to downloaded file
|
|
"""
|
|
endpoint = cluster.get_random_storage_rpc_endpoint()
|
|
return get_object(
|
|
wallet,
|
|
cid,
|
|
oid,
|
|
shell,
|
|
endpoint,
|
|
bearer,
|
|
write_object,
|
|
xhdr,
|
|
no_progress,
|
|
session,
|
|
timeout,
|
|
)
|
|
|
|
|
|
@reporter.step("Get object from {endpoint}")
|
|
def get_object(
|
|
wallet: WalletInfo,
|
|
cid: str,
|
|
oid: str,
|
|
shell: Shell,
|
|
endpoint: str,
|
|
bearer: Optional[str] = None,
|
|
write_object: Optional[str] = None,
|
|
xhdr: Optional[dict] = None,
|
|
no_progress: bool = True,
|
|
session: Optional[str] = None,
|
|
timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
|
|
) -> TestFile:
|
|
"""
|
|
GET from FrostFS.
|
|
|
|
Args:
|
|
wallet (WalletInfo): wallet on whose behalf GET is done
|
|
cid (str): ID of Container where we get the Object from
|
|
oid (str): Object ID
|
|
shell: executor for cli command
|
|
bearer: path to Bearer Token file, appends to `--bearer` key
|
|
write_object: path to downloaded file, appends to `--file` key
|
|
endpoint: FrostFS endpoint to send request to, appends to `--rpc-endpoint` key
|
|
no_progress(optional, bool): do not show progress bar
|
|
xhdr (optional, dict): Request X-Headers in form of Key=Value
|
|
session (optional, dict): path to a JSON-encoded container session token
|
|
timeout: Timeout for the operation.
|
|
Returns:
|
|
(str): path to downloaded file
|
|
"""
|
|
|
|
if not write_object:
|
|
write_object = str(uuid.uuid4())
|
|
test_file = TestFile(os.path.join(ASSETS_DIR, write_object))
|
|
|
|
cli = FrostfsCli(shell, FROSTFS_CLI_EXEC, wallet.config_path)
|
|
cli.object.get(
|
|
rpc_endpoint=endpoint,
|
|
cid=cid,
|
|
oid=oid,
|
|
file=test_file,
|
|
bearer=bearer,
|
|
no_progress=no_progress,
|
|
xhdr=xhdr,
|
|
session=session,
|
|
timeout=timeout,
|
|
)
|
|
|
|
return test_file
|
|
|
|
|
|
@reporter.step("Get Range Hash from {endpoint}")
|
|
def get_range_hash(
|
|
wallet: WalletInfo,
|
|
cid: str,
|
|
oid: str,
|
|
range_cut: str,
|
|
shell: Shell,
|
|
endpoint: str,
|
|
bearer: Optional[str] = None,
|
|
xhdr: Optional[dict] = None,
|
|
session: Optional[str] = None,
|
|
timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
|
|
):
|
|
"""
|
|
GETRANGEHASH of given Object.
|
|
|
|
Args:
|
|
wallet: wallet on whose behalf GETRANGEHASH is done
|
|
cid: ID of Container where we get the Object from
|
|
oid: Object ID
|
|
shell: executor for cli command
|
|
bearer: path to Bearer Token file, appends to `--bearer` key
|
|
range_cut: Range to take hash from in the form offset1:length1,...,
|
|
value to pass to the `--range` parameter
|
|
endpoint: FrostFS endpoint to send request to, appends to `--rpc-endpoint` key
|
|
xhdr: Request X-Headers in form of Key=Values
|
|
session: Filepath to a JSON- or binary-encoded token of the object RANGEHASH session.
|
|
timeout: Timeout for the operation.
|
|
Returns:
|
|
None
|
|
"""
|
|
cli = FrostfsCli(shell, FROSTFS_CLI_EXEC, wallet.config_path)
|
|
result = cli.object.hash(
|
|
rpc_endpoint=endpoint,
|
|
cid=cid,
|
|
oid=oid,
|
|
range=range_cut,
|
|
bearer=bearer,
|
|
xhdr=xhdr,
|
|
session=session,
|
|
timeout=timeout,
|
|
)
|
|
|
|
# cutting off output about range offset and length
|
|
return result.stdout.split(":")[1].strip()
|
|
|
|
|
|
@reporter.step("Put object to random node")
|
|
def put_object_to_random_node(
|
|
wallet: WalletInfo,
|
|
path: str,
|
|
cid: str,
|
|
shell: Shell,
|
|
cluster: Cluster,
|
|
bearer: Optional[str] = None,
|
|
copies_number: Optional[int] = None,
|
|
attributes: Optional[dict] = None,
|
|
xhdr: Optional[dict] = None,
|
|
expire_at: Optional[int] = None,
|
|
no_progress: bool = True,
|
|
session: Optional[str] = None,
|
|
timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
|
|
):
|
|
"""
|
|
PUT of given file to a random storage node.
|
|
|
|
Args:
|
|
wallet: wallet on whose behalf PUT is done
|
|
path: path to file to be PUT
|
|
cid: ID of Container where we get the Object from
|
|
shell: executor for cli command
|
|
cluster: cluster under test
|
|
bearer: path to Bearer Token file, appends to `--bearer` key
|
|
copies_number: Number of copies of the object to store within the RPC call
|
|
attributes: User attributes in form of Key1=Value1,Key2=Value2
|
|
cluster: cluster under test
|
|
no_progress: do not show progress bar
|
|
expire_at: Last epoch in the life of the object
|
|
xhdr: Request X-Headers in form of Key=Value
|
|
session: path to a JSON-encoded container session token
|
|
timeout: Timeout for the operation.
|
|
Returns:
|
|
ID of uploaded Object
|
|
"""
|
|
|
|
endpoint = cluster.get_random_storage_rpc_endpoint()
|
|
return put_object(
|
|
wallet,
|
|
path,
|
|
cid,
|
|
shell,
|
|
endpoint,
|
|
bearer,
|
|
copies_number,
|
|
attributes,
|
|
xhdr,
|
|
expire_at,
|
|
no_progress,
|
|
session,
|
|
timeout=timeout,
|
|
)
|
|
|
|
|
|
@reporter.step("Put object at {endpoint} in container {cid}")
|
|
def put_object(
|
|
wallet: WalletInfo,
|
|
path: str,
|
|
cid: str,
|
|
shell: Shell,
|
|
endpoint: str,
|
|
bearer: Optional[str] = None,
|
|
copies_number: Optional[int] = None,
|
|
attributes: Optional[dict] = None,
|
|
xhdr: Optional[dict] = None,
|
|
expire_at: Optional[int] = None,
|
|
no_progress: bool = True,
|
|
session: Optional[str] = None,
|
|
timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
|
|
):
|
|
"""
|
|
PUT of given file.
|
|
|
|
Args:
|
|
wallet: wallet on whose behalf PUT is done
|
|
path: path to file to be PUT
|
|
cid: ID of Container where we get the Object from
|
|
shell: executor for cli command
|
|
bearer: path to Bearer Token file, appends to `--bearer` key
|
|
copies_number: Number of copies of the object to store within the RPC call
|
|
attributes: User attributes in form of Key1=Value1,Key2=Value2
|
|
endpoint: FrostFS endpoint to send request to, appends to `--rpc-endpoint` key
|
|
no_progress: do not show progress bar
|
|
expire_at: Last epoch in the life of the object
|
|
xhdr: Request X-Headers in form of Key=Value
|
|
session: path to a JSON-encoded container session token
|
|
timeout: Timeout for the operation.
|
|
Returns:
|
|
(str): ID of uploaded Object
|
|
"""
|
|
|
|
cli = FrostfsCli(shell, FROSTFS_CLI_EXEC, wallet.config_path)
|
|
result = cli.object.put(
|
|
rpc_endpoint=endpoint,
|
|
file=path,
|
|
cid=cid,
|
|
attributes=attributes,
|
|
bearer=bearer,
|
|
copies_number=copies_number,
|
|
expire_at=expire_at,
|
|
no_progress=no_progress,
|
|
xhdr=xhdr,
|
|
session=session,
|
|
timeout=timeout,
|
|
)
|
|
|
|
# Splitting CLI output to separate lines and taking the penultimate line
|
|
id_str = result.stdout.strip().split("\n")[-2]
|
|
oid = id_str.split(":")[1]
|
|
return oid.strip()
|
|
|
|
|
|
@reporter.step("Delete object {cid}/{oid} from {endpoint}")
|
|
def delete_object(
|
|
wallet: WalletInfo,
|
|
cid: str,
|
|
oid: str,
|
|
shell: Shell,
|
|
endpoint: str,
|
|
bearer: str = "",
|
|
xhdr: Optional[dict] = None,
|
|
session: Optional[str] = None,
|
|
timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
|
|
):
|
|
"""
|
|
DELETE an Object.
|
|
|
|
Args:
|
|
wallet: wallet on whose behalf DELETE is done
|
|
cid: ID of Container where we get the Object from
|
|
oid: ID of Object we are going to delete
|
|
shell: executor for cli command
|
|
bearer: path to Bearer Token file, appends to `--bearer` key
|
|
endpoint: FrostFS endpoint to send request to, appends to `--rpc-endpoint` key
|
|
xhdr: Request X-Headers in form of Key=Value
|
|
session: path to a JSON-encoded container session token
|
|
timeout: Timeout for the operation.
|
|
Returns:
|
|
(str): Tombstone ID
|
|
"""
|
|
|
|
cli = FrostfsCli(shell, FROSTFS_CLI_EXEC, wallet.config_path)
|
|
result = cli.object.delete(
|
|
rpc_endpoint=endpoint,
|
|
cid=cid,
|
|
oid=oid,
|
|
bearer=bearer,
|
|
xhdr=xhdr,
|
|
session=session,
|
|
timeout=timeout,
|
|
)
|
|
|
|
id_str = result.stdout.split("\n")[1]
|
|
tombstone = id_str.split(":")[1]
|
|
return tombstone.strip()
|
|
|
|
|
|
@reporter.step("Get Range")
|
|
def get_range(
|
|
wallet: WalletInfo,
|
|
cid: str,
|
|
oid: str,
|
|
range_cut: str,
|
|
shell: Shell,
|
|
endpoint: str,
|
|
bearer: str = "",
|
|
xhdr: Optional[dict] = None,
|
|
session: Optional[str] = None,
|
|
timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
|
|
):
|
|
"""
|
|
GETRANGE an Object.
|
|
|
|
Args:
|
|
wallet: wallet on whose behalf GETRANGE is done
|
|
cid: ID of Container where we get the Object from
|
|
oid: ID of Object we are going to request
|
|
range_cut: range to take data from in the form offset:length
|
|
shell: executor for cli command
|
|
endpoint: FrostFS endpoint to send request to, appends to `--rpc-endpoint` key
|
|
bearer: path to Bearer Token file, appends to `--bearer` key
|
|
xhdr: Request X-Headers in form of Key=Value
|
|
session: path to a JSON-encoded container session token
|
|
timeout: Timeout for the operation.
|
|
Returns:
|
|
(str, bytes) - path to the file with range content and content of this file as bytes
|
|
"""
|
|
test_file = TestFile(os.path.join(ASSETS_DIR, str(uuid.uuid4())))
|
|
|
|
cli = FrostfsCli(shell, FROSTFS_CLI_EXEC, wallet.config_path)
|
|
cli.object.range(
|
|
rpc_endpoint=endpoint,
|
|
cid=cid,
|
|
oid=oid,
|
|
range=range_cut,
|
|
file=test_file,
|
|
bearer=bearer,
|
|
xhdr=xhdr,
|
|
session=session,
|
|
timeout=timeout,
|
|
)
|
|
|
|
with open(test_file, "rb") as file:
|
|
content = file.read()
|
|
return test_file, content
|
|
|
|
|
|
@reporter.step("Lock Object")
|
|
def lock_object(
|
|
wallet: WalletInfo,
|
|
cid: str,
|
|
oid: str,
|
|
shell: Shell,
|
|
endpoint: str,
|
|
lifetime: Optional[int] = None,
|
|
expire_at: Optional[int] = None,
|
|
address: Optional[str] = None,
|
|
bearer: Optional[str] = None,
|
|
session: Optional[str] = None,
|
|
ttl: Optional[int] = None,
|
|
xhdr: Optional[dict] = None,
|
|
timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
|
|
) -> str:
|
|
"""
|
|
Locks object in container.
|
|
|
|
Args:
|
|
address: Address of wallet account.
|
|
bearer: File with signed JSON or binary encoded bearer token.
|
|
cid: Container ID.
|
|
oid: Object ID.
|
|
lifetime: Lock lifetime.
|
|
expire_at: Lock expiration epoch.
|
|
shell: executor for cli command
|
|
endpoint: FrostFS endpoint to send request to, appends to `--rpc-endpoint` key
|
|
session: Path to a JSON-encoded container session token.
|
|
ttl: TTL value in request meta header (default 2).
|
|
wallet: WIF (NEP-2) string or path to the wallet or binary key.
|
|
xhdr: Dict with request X-Headers.
|
|
timeout: Timeout for the operation.
|
|
|
|
Returns:
|
|
Lock object ID
|
|
"""
|
|
|
|
cli = FrostfsCli(shell, FROSTFS_CLI_EXEC, wallet.config_path)
|
|
result = cli.object.lock(
|
|
rpc_endpoint=endpoint,
|
|
lifetime=lifetime,
|
|
expire_at=expire_at,
|
|
address=address,
|
|
cid=cid,
|
|
oid=oid,
|
|
bearer=bearer,
|
|
xhdr=xhdr,
|
|
session=session,
|
|
ttl=ttl,
|
|
timeout=timeout,
|
|
)
|
|
|
|
# Splitting CLI output to separate lines and taking the penultimate line
|
|
id_str = result.stdout.strip().split("\n")[0]
|
|
oid = id_str.split(":")[1]
|
|
return oid.strip()
|
|
|
|
|
|
@reporter.step("Search object")
|
|
def search_object(
|
|
wallet: WalletInfo,
|
|
cid: str,
|
|
shell: Shell,
|
|
endpoint: str,
|
|
bearer: str = "",
|
|
filters: Optional[dict] = None,
|
|
expected_objects_list: Optional[list] = None,
|
|
xhdr: Optional[dict] = None,
|
|
session: Optional[str] = None,
|
|
phy: bool = False,
|
|
root: bool = False,
|
|
timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
|
|
) -> list:
|
|
"""
|
|
SEARCH an Object.
|
|
|
|
Args:
|
|
wallet: wallet on whose behalf SEARCH is done
|
|
cid: ID of Container where we get the Object from
|
|
shell: executor for cli command
|
|
bearer: path to Bearer Token file, appends to `--bearer` key
|
|
endpoint: FrostFS endpoint to send request to, appends to `--rpc-endpoint` key
|
|
filters: key=value pairs to filter Objects
|
|
expected_objects_list: a list of ObjectIDs to compare found Objects with
|
|
xhdr: Request X-Headers in form of Key=Value
|
|
session: path to a JSON-encoded container session token
|
|
phy: Search physically stored objects.
|
|
root: Search for user objects.
|
|
timeout: Timeout for the operation.
|
|
|
|
Returns:
|
|
list of found ObjectIDs
|
|
"""
|
|
|
|
cli = FrostfsCli(shell, FROSTFS_CLI_EXEC, wallet.config_path)
|
|
result = cli.object.search(
|
|
rpc_endpoint=endpoint,
|
|
cid=cid,
|
|
bearer=bearer,
|
|
xhdr=xhdr,
|
|
filters=[f"{filter_key} EQ {filter_val}" for filter_key, filter_val in filters.items()] if filters else None,
|
|
session=session,
|
|
phy=phy,
|
|
root=root,
|
|
timeout=timeout,
|
|
)
|
|
|
|
found_objects = re.findall(r"(\w{43,44})", result.stdout)
|
|
|
|
if expected_objects_list:
|
|
if sorted(found_objects) == sorted(expected_objects_list):
|
|
logger.info(f"Found objects list '{found_objects}' " f"is equal for expected list '{expected_objects_list}'")
|
|
else:
|
|
logger.warning(f"Found object list {found_objects} " f"is not equal to expected list '{expected_objects_list}'")
|
|
|
|
return found_objects
|
|
|
|
|
|
@reporter.step("Get netmap netinfo")
|
|
def get_netmap_netinfo(
|
|
wallet: WalletInfo,
|
|
shell: Shell,
|
|
endpoint: str,
|
|
address: Optional[str] = None,
|
|
ttl: Optional[int] = None,
|
|
xhdr: Optional[dict] = None,
|
|
timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
|
|
) -> dict[str, Any]:
|
|
"""
|
|
Get netmap netinfo output from node
|
|
|
|
Args:
|
|
wallet (WalletInfo): wallet on whose behalf request is done
|
|
shell: executor for cli command
|
|
endpoint (optional, str): FrostFS endpoint to send request to, appends to `--rpc-endpoint` key
|
|
address: Address of wallet account
|
|
ttl: TTL value in request meta header (default 2)
|
|
wallet: Path to the wallet or binary key
|
|
xhdr: Request X-Headers in form of Key=Value
|
|
timeout: Timeout for the operation.
|
|
|
|
Returns:
|
|
(dict): dict of parsed command output
|
|
"""
|
|
|
|
cli = FrostfsCli(shell, FROSTFS_CLI_EXEC, wallet.config_path)
|
|
output = cli.netmap.netinfo(
|
|
rpc_endpoint=endpoint,
|
|
address=address,
|
|
ttl=ttl,
|
|
xhdr=xhdr,
|
|
timeout=timeout,
|
|
)
|
|
|
|
settings = dict()
|
|
|
|
patterns = [
|
|
(re.compile("(.*): (\d+)"), int),
|
|
(re.compile("(.*): (false|true)"), bool),
|
|
(re.compile("(.*): (\d+\.\d+)"), float),
|
|
]
|
|
for pattern, func in patterns:
|
|
for setting, value in re.findall(pattern, output.stdout):
|
|
settings[setting.lower().strip().replace(" ", "_")] = func(value)
|
|
|
|
return settings
|
|
|
|
|
|
@reporter.step("Head object")
|
|
def head_object(
|
|
wallet: WalletInfo,
|
|
cid: str,
|
|
oid: str,
|
|
shell: Shell,
|
|
endpoint: str,
|
|
bearer: str = "",
|
|
xhdr: Optional[dict] = None,
|
|
json_output: bool = True,
|
|
is_raw: bool = False,
|
|
is_direct: bool = False,
|
|
session: Optional[str] = None,
|
|
timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
|
|
):
|
|
"""
|
|
HEAD an Object.
|
|
|
|
Args:
|
|
wallet (WalletInfo): wallet on whose behalf HEAD is done
|
|
cid (str): ID of Container where we get the Object from
|
|
oid (str): ObjectID to HEAD
|
|
shell: executor for cli command
|
|
bearer (optional, str): path to Bearer Token file, appends to `--bearer` key
|
|
endpoint(optional, str): FrostFS endpoint to send request to
|
|
json_output(optional, bool): return response in JSON format or not; this flag
|
|
turns into `--json` key
|
|
is_raw(optional, bool): send "raw" request or not; this flag
|
|
turns into `--raw` key
|
|
is_direct(optional, bool): send request directly to the node or not; this flag
|
|
turns into `--ttl 1` key
|
|
xhdr (optional, dict): Request X-Headers in form of Key=Value
|
|
session (optional, dict): path to a JSON-encoded container session token
|
|
timeout: Timeout for the operation.
|
|
Returns:
|
|
depending on the `json_output` parameter value, the function returns
|
|
(dict): HEAD response in JSON format
|
|
or
|
|
(str): HEAD response as a plain text
|
|
"""
|
|
|
|
cli = FrostfsCli(shell, FROSTFS_CLI_EXEC, wallet.config_path)
|
|
result = cli.object.head(
|
|
rpc_endpoint=endpoint,
|
|
cid=cid,
|
|
oid=oid,
|
|
bearer=bearer,
|
|
json_mode=json_output,
|
|
raw=is_raw,
|
|
ttl=1 if is_direct else None,
|
|
xhdr=xhdr,
|
|
session=session,
|
|
timeout=timeout,
|
|
)
|
|
|
|
if not json_output:
|
|
return result
|
|
|
|
try:
|
|
decoded = json.loads(result.stdout)
|
|
except Exception as exc:
|
|
# If we failed to parse output as JSON, the cause might be
|
|
# the plain text string in the beginning of the output.
|
|
# Here we cut off first string and try to parse again.
|
|
logger.info(f"failed to parse output: {exc}")
|
|
logger.info("parsing output in another way")
|
|
fst_line_idx = result.stdout.find("\n")
|
|
decoded = json.loads(result.stdout[fst_line_idx:])
|
|
|
|
# if response
|
|
if "chunks" in decoded.keys():
|
|
logger.info("decoding ec chunks")
|
|
return decoded["chunks"]
|
|
|
|
# If response is Complex Object header, it has `splitId` key
|
|
if "splitId" in decoded.keys():
|
|
return json_utils.decode_split_header(decoded)
|
|
|
|
# If response is Last or Linking Object header,
|
|
# it has `header` dictionary and non-null `split` dictionary
|
|
if "split" in decoded["header"].keys():
|
|
if decoded["header"]["split"]:
|
|
return json_utils.decode_linking_object(decoded)
|
|
|
|
if decoded["header"]["objectType"] == "STORAGE_GROUP":
|
|
return json_utils.decode_storage_group(decoded)
|
|
|
|
if decoded["header"]["objectType"] == "TOMBSTONE":
|
|
return json_utils.decode_tombstone(decoded)
|
|
|
|
return json_utils.decode_simple_header(decoded)
|
|
|
|
|
|
@reporter.step("Run neo-go dump-keys")
|
|
def neo_go_dump_keys(shell: Shell, wallet: WalletInfo) -> dict:
|
|
"""
|
|
Run neo-go dump keys command
|
|
|
|
Args:
|
|
shell: executor for cli command
|
|
wallet: wallet path to dump from
|
|
Returns:
|
|
dict Address:Wallet Key
|
|
"""
|
|
neogo = NeoGo(shell, neo_go_exec_path=NEOGO_EXECUTABLE)
|
|
output = neogo.wallet.dump_keys(wallet=wallet).stdout
|
|
first_line = ""
|
|
try:
|
|
# taking first line from command's output contain wallet address
|
|
first_line = output.split("\n")[0]
|
|
except Exception:
|
|
logger.error(f"Got empty output (neo-go dump keys): {output}")
|
|
address_id = first_line.split()[0]
|
|
# taking second line from command's output contain wallet key
|
|
wallet_key = output.split("\n")[1]
|
|
return {address_id: wallet_key}
|
|
|
|
|
|
@reporter.step("Run neo-go query height")
|
|
def neo_go_query_height(shell: Shell, endpoint: str) -> dict:
|
|
"""
|
|
Run neo-go query height command
|
|
|
|
Args:
|
|
shell: executor for cli command
|
|
endpoint: endpoint to execute
|
|
Returns:
|
|
dict->
|
|
Latest block: {value}
|
|
Validated state: {value}
|
|
|
|
"""
|
|
neogo = NeoGo(shell, neo_go_exec_path=NEOGO_EXECUTABLE)
|
|
output = neogo.query.height(rpc_endpoint=endpoint).stdout
|
|
first_line = ""
|
|
try:
|
|
# taking first line from command's output contain the latest block in blockchain
|
|
first_line = output.split("\n")[0]
|
|
except Exception:
|
|
logger.error(f"Got empty output (neo-go query height): {output}")
|
|
latest_block = first_line.split(":")
|
|
# taking second line from command's output contain wallet key
|
|
second_line = output.split("\n")[1]
|
|
if second_line != "":
|
|
validated_state = second_line.split(":")
|
|
return {
|
|
latest_block[0].replace(":", ""): int(latest_block[1]),
|
|
validated_state[0].replace(":", ""): int(validated_state[1]),
|
|
}
|
|
return {latest_block[0].replace(":", ""): int(latest_block[1])}
|
|
|
|
|
|
@wait_for_success()
|
|
@reporter.step("Search object nodes")
|
|
def get_object_nodes(
|
|
cluster: Cluster,
|
|
cid: str,
|
|
oid: str,
|
|
alive_node: ClusterNode,
|
|
bearer: str = "",
|
|
xhdr: Optional[dict] = None,
|
|
is_direct: bool = False,
|
|
verify_presence_all: bool = False,
|
|
timeout: Optional[str] = CLI_DEFAULT_TIMEOUT,
|
|
) -> list[ClusterNode]:
|
|
shell = alive_node.host.get_shell()
|
|
endpoint = alive_node.storage_node.get_rpc_endpoint()
|
|
wallet = alive_node.storage_node.get_remote_wallet_path()
|
|
wallet_config = alive_node.storage_node.get_remote_wallet_config_path()
|
|
|
|
cli = FrostfsCli(shell, FROSTFS_CLI_EXEC, wallet_config)
|
|
|
|
response = cli.object.nodes(
|
|
rpc_endpoint=endpoint,
|
|
cid=cid,
|
|
oid=oid,
|
|
bearer=bearer,
|
|
ttl=1 if is_direct else None,
|
|
json=True,
|
|
xhdr=xhdr,
|
|
timeout=timeout,
|
|
verify_presence_all=verify_presence_all,
|
|
)
|
|
|
|
response_json = json.loads(response.stdout)
|
|
# Currently, the command will show expected and confirmed nodes.
|
|
# And we (currently) count only nodes which are both expected and confirmed
|
|
object_nodes_id = {
|
|
required_node
|
|
for data_object in response_json["data_objects"]
|
|
for required_node in data_object["required_nodes"]
|
|
if required_node in data_object["confirmed_nodes"]
|
|
}
|
|
|
|
netmap_nodes_list = parse_netmap_output(
|
|
cli.netmap.snapshot(
|
|
rpc_endpoint=endpoint,
|
|
wallet=wallet,
|
|
).stdout
|
|
)
|
|
netmap_nodes = [
|
|
netmap_node for object_node in object_nodes_id for netmap_node in netmap_nodes_list if object_node == netmap_node.node_id
|
|
]
|
|
|
|
object_nodes = [
|
|
cluster_node for netmap_node in netmap_nodes for cluster_node in cluster.cluster_nodes if netmap_node.node == cluster_node.host_ip
|
|
]
|
|
|
|
return object_nodes
|