forked from TrueCloudLab/frostfs-testcases
92 lines
3 KiB
Python
92 lines
3 KiB
Python
#!/usr/bin/python3.8
|
|
|
|
import os
|
|
import tarfile
|
|
import uuid
|
|
import docker
|
|
|
|
from neo3 import wallet
|
|
from common import SIMPLE_OBJ_SIZE, ASSETS_DIR
|
|
from robot.api.deco import keyword
|
|
from robot.api import logger
|
|
from robot.libraries.BuiltIn import BuiltIn
|
|
from cli_helpers import _cmd_run
|
|
|
|
|
|
ROBOT_AUTO_KEYWORDS = False
|
|
|
|
@keyword('Generate file of bytes')
|
|
def generate_file_of_bytes(size: str) -> str:
|
|
"""
|
|
Function generates big binary file with the specified size in bytes.
|
|
:param size: the size in bytes, can be declared as 6e+6 for example
|
|
"""
|
|
size = int(float(size))
|
|
filename = f"{os.getcwd()}/{ASSETS_DIR}/{str(uuid.uuid4())}"
|
|
with open(filename, 'wb') as fout:
|
|
fout.write(os.urandom(size))
|
|
logger.info(f"file with size {size} bytes has been generated: {filename}")
|
|
return filename
|
|
|
|
@keyword('Get Docker Logs')
|
|
def get_container_logs(testcase_name: str) -> None:
|
|
client = docker.APIClient(base_url='unix://var/run/docker.sock')
|
|
logs_dir = BuiltIn().get_variable_value("${OUTPUT_DIR}")
|
|
tar_name = f"{logs_dir}/dockerlogs({testcase_name}).tar.gz"
|
|
tar = tarfile.open(tar_name, "w:gz")
|
|
for container in client.containers():
|
|
container_name = container['Names'][0][1:]
|
|
if client.inspect_container(container_name)['Config']['Domainname'] == "neofs.devenv":
|
|
file_name = f"{logs_dir}/docker_log_{container_name}"
|
|
with open(file_name,'wb') as out:
|
|
out.write(client.logs(container_name))
|
|
logger.info(f"Collected logs from container {container_name}")
|
|
tar.add(file_name)
|
|
os.remove(file_name)
|
|
tar.close()
|
|
|
|
@keyword('WIF to Binary')
|
|
def wif_to_binary(wif: str) -> str:
|
|
priv_key = wallet.Account.private_key_from_wif(wif)
|
|
path = f"{os.getcwd()}/{ASSETS_DIR}/{str(uuid.uuid4())}"
|
|
with open(path, "wb") as out:
|
|
out.write(priv_key)
|
|
return path
|
|
|
|
@keyword('Make Up')
|
|
def make_up(services: list=[], config_dict: dict={}):
|
|
test_path = os.getcwd()
|
|
dev_path = os.getenv('DEVENV_PATH', '../neofs-dev-env')
|
|
os.chdir(dev_path)
|
|
|
|
if len(services) > 0:
|
|
for service in services:
|
|
if config_dict != {}:
|
|
with open(f"{dev_path}/.int_test.env", "a") as out:
|
|
for key, value in config_dict.items():
|
|
out.write(f'{key}={value}')
|
|
cmd = f'make up/{service}'
|
|
_cmd_run(cmd)
|
|
else:
|
|
cmd = f'make up/basic; make update.max_object_size val={SIMPLE_OBJ_SIZE}'
|
|
_cmd_run(cmd, timeout=120)
|
|
|
|
os.chdir(test_path)
|
|
|
|
@keyword('Make Down')
|
|
def make_down(services: list=[]):
|
|
test_path = os.getcwd()
|
|
dev_path = os.getenv('DEVENV_PATH', '../neofs-dev-env')
|
|
os.chdir(dev_path)
|
|
|
|
if len(services) > 0:
|
|
for service in services:
|
|
cmd = f'make down/{service}'
|
|
_cmd_run(cmd)
|
|
with open(f"{dev_path}/.int_test.env", "w"):
|
|
pass
|
|
else:
|
|
cmd = 'make down; make clean'
|
|
_cmd_run(cmd, timeout=60)
|
|
|
|
os.chdir(test_path)
|