anurindm/frostfs-testcases
1
0
Fork 0
forked from TrueCloudLab/frostfs-testcases
Code Pull requests Activity

(#66) TEMP_DIR -> ASSETS_DIR

Browse source 
Signed-off-by: anastasia prasolova <anastasia@nspcc.ru>
This commit is contained in:
anastasia prasolova 2021-06-01 15:07:31 +03:00 • committed by Anastasia Prasolova
parent 8d50305982
commit 28af257f8f
39 changed files with 303 additions and 478 deletions
Download patch file Download diff file Expand all files Collapse all files

24
robot/resources/lib/utility_keywords.py
View file

@ -2,7 +2,6 @@
import docker
import os
import shutil
import tarfile
import uuid
@ -22,11 +21,11 @@ def generate_file_of_bytes(size: str) -> str:
:param size: the size in bytes, can be declared as 6e+6 for example
"""
size = int(float(size))
filename = TEMP_DIR + str(uuid.uuid4())
filename = f"{os.getcwd()}/{ASSETS_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}"
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:
@ -43,20 +42,3 @@ def get_container_logs(testcase_name: str) -> None:
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}")