forked from TrueCloudLab/frostfs-testcases
132 lines
4.3 KiB
Python
132 lines
4.3 KiB
Python
import logging
|
|
import os
|
|
import re
|
|
import shutil
|
|
|
|
import allure
|
|
import pytest
|
|
from robot.api import deco
|
|
|
|
import wallet
|
|
from cli_helpers import _cmd_run
|
|
from common import (ASSETS_DIR, FREE_STORAGE, INFRASTRUCTURE_TYPE, MAINNET_WALLET_PATH,
|
|
NEOFS_NETMAP_DICT)
|
|
from payment_neogo import neofs_deposit, transfer_mainnet_gas
|
|
from python_keywords.node_management import node_healthcheck, create_ssh_client
|
|
|
|
|
|
def robot_keyword_adapter(name=None, tags=(), types=()):
|
|
return allure.step(name)
|
|
|
|
|
|
deco.keyword = robot_keyword_adapter
|
|
|
|
logger = logging.getLogger('NeoLogger')
|
|
|
|
|
|
@pytest.fixture(scope='session')
|
|
def cloud_infrastructure_check():
|
|
if not is_cloud_infrastructure():
|
|
pytest.skip('Test only works on SberCloud infrastructure')
|
|
yield
|
|
|
|
|
|
def is_cloud_infrastructure():
|
|
return INFRASTRUCTURE_TYPE == "CLOUD_VM"
|
|
|
|
|
|
@pytest.fixture(scope='session', autouse=True)
|
|
@allure.title('Check binary versions')
|
|
def check_binary_versions(request):
|
|
environment_dir = request.config.getoption('--alluredir')
|
|
is_cloud = is_cloud_infrastructure()
|
|
# Collect versions of neo binaries
|
|
binaries = ['neo-go', 'neofs-cli', 'neofs-authmate']
|
|
env_out = _get_binaries_version_local(binaries)
|
|
|
|
if is_cloud:
|
|
binaries = ['neo-go',
|
|
'neofs-adm',
|
|
'neofs-cli',
|
|
'neofs-http-gw',
|
|
'neofs-ir',
|
|
'neofs-lens',
|
|
'neofs-node',
|
|
'neofs-s3-authmate',
|
|
'neofs-s3-gw',
|
|
'neogo-morph-cn']
|
|
env_out = _get_binaries_version_remote(binaries)
|
|
|
|
# Get version of aws binary
|
|
out = _cmd_run('aws --version')
|
|
out_lines = out.split("\n")
|
|
env_out["AWS"] = out_lines[0] if out_lines else 'Unknown'
|
|
|
|
if environment_dir:
|
|
with open(f'{environment_dir}/environment.properties', 'w') as out_file:
|
|
for env, env_value in env_out.items():
|
|
out_file.write(f'{env}={env_value}\n')
|
|
|
|
|
|
def _get_binaries_version_local(binaries: list) -> dict:
|
|
env_out = {}
|
|
for binary in binaries:
|
|
out = _cmd_run(f'{binary} --version')
|
|
version = re.search(r'version[:\s]*(.+)', out, re.IGNORECASE)
|
|
env_out[binary.upper()] = version.group(1) if version else 'Unknown'
|
|
return env_out
|
|
|
|
|
|
def _get_binaries_version_remote(binaries: list) -> dict:
|
|
env_out = {}
|
|
|
|
for node_name in NEOFS_NETMAP_DICT:
|
|
with create_ssh_client(node_name) as ssh_client:
|
|
for binary in binaries:
|
|
out = ssh_client.exec(f'{binary} --version').stdout
|
|
version = re.search(r'version[:\s]*(.+)', out, re.IGNORECASE)
|
|
version = version.group(1) if version else 'Unknown'
|
|
if not env_out.get(binary.upper()):
|
|
env_out[binary.upper()] = version
|
|
else:
|
|
msg = f'Expected binary {binary} versions on node s1 and {node_name} are the same'
|
|
assert env_out[binary.upper()] == version, msg
|
|
return env_out
|
|
|
|
|
|
@pytest.fixture(scope='session', autouse=True)
|
|
@allure.title('Run health check for all storage nodes')
|
|
def run_health_check():
|
|
failed_nodes = []
|
|
for node_name in NEOFS_NETMAP_DICT.keys():
|
|
health_check = node_healthcheck(node_name)
|
|
if health_check.health_status != 'READY' or health_check.network_status != 'ONLINE':
|
|
failed_nodes.append(node_name)
|
|
|
|
if failed_nodes:
|
|
raise AssertionError(f'Nodes {failed_nodes} are not healthy')
|
|
|
|
|
|
@pytest.fixture(scope='session')
|
|
@allure.title('Prepare tmp directory')
|
|
def prepare_tmp_dir():
|
|
full_path = f'{os.getcwd()}/{ASSETS_DIR}'
|
|
shutil.rmtree(full_path, ignore_errors=True)
|
|
os.mkdir(full_path)
|
|
yield full_path
|
|
shutil.rmtree(full_path)
|
|
|
|
|
|
@pytest.fixture(scope='session')
|
|
@allure.title('Prepare wallet and deposit')
|
|
def prepare_wallet_and_deposit(prepare_tmp_dir):
|
|
wallet_path, addr, _ = wallet.init_wallet(ASSETS_DIR)
|
|
logger.info(f'Init wallet: {wallet_path},\naddr: {addr}')
|
|
allure.attach.file(wallet_path, os.path.basename(wallet_path), allure.attachment_type.JSON)
|
|
|
|
if not FREE_STORAGE:
|
|
deposit = 30
|
|
transfer_mainnet_gas(wallet_path, deposit + 1, wallet_path=MAINNET_WALLET_PATH)
|
|
neofs_deposit(wallet_path, deposit)
|
|
|
|
return wallet_path
|