forked from TrueCloudLab/frostfs-testcases
62 lines
2 KiB
Python
62 lines
2 KiB
Python
|
#!/usr/bin/python3.7
|
||
|
|
|
||
|
|
import docker
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
import tarfile
|
||
|
|
import uuid
|
||
|
|
|
||
|
|
from robot.api.deco import keyword
|
||
|
|
from robot.api import logger
|
||
|
|
|
||
|
|
from common import *
|
||
|
|
|
||
|
|
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 = TEMP_DIR + str(uuid.uuid4())
|
||
|
|
with open(filename, 'wb') as fout:
|
||
|
|
fout.write(os.urandom(size))
|
||
|
|
logger.info(f"Random binary file with size {size} bytes has been generated.")
|
||
|
|
return f"{os.getcwd()}/{filename}"
|
||
|
|
|
||
|
|
@keyword('Get Docker Logs')
|
||
|
|
def get_container_logs(testcase_name: str) -> None:
|
||
|
|
client = docker.APIClient(base_url='unix://var/run/docker.sock')
|
||
|
|
tar_name = f"artifacts/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"artifacts/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('Cleanup Files')
|
||
|
|
def cleanup_file() -> None:
|
||
|
|
if os.path.isdir(TEMP_DIR):
|
||
|
|
try:
|
||
|
|
shutil.rmtree(TEMP_DIR)
|
||
|
|
logger.info(f"File '{TEMP_DIR}' has been deleted.")
|
||
|
|
except OSError as e:
|
||
|
|
raise Exception(f"Error: '{TEMP_DIR}' - {e.strerror}.")
|
||
|
|
else:
|
||
|
|
logger.warn(f"Error: '{TEMP_DIR}' file not found")
|
||
|
|
|
||
|
|
@keyword('Create Temporary Directory')
|
||
|
|
def create_temp_dir() -> None:
|
||
|
|
if not os.path.exists(TEMP_DIR):
|
||
|
|
os.makedirs(TEMP_DIR)
|
||
|
|
logger.info(f"Created temporary directory: {TEMP_DIR}")
|