[#350] Cleanup utility keywords

Signed-off-by: Vladimir Domnich <v.domnich@yadro.com>

robot/resources/lib/python_keywords/object_access.py

@@ -1,6 +1,7 @@
 from typing import Optional
 
 import allure
+from file_helper import get_file_hash
 from grpc_responses import OBJECT_ACCESS_DENIED, error_matches_status
 from python_keywords.neofs_verbs import (
     delete_object,
@@ -11,7 +12,6 @@ from python_keywords.neofs_verbs import (
     put_object,
     search_object,
 )
-from python_keywords.utility_keywords import get_file_hash
 
 OPERATION_ERROR_TYPE = RuntimeError
 

robot/resources/lib/python_keywords/utility_keywords.py

@@ -1,150 +0,0 @@
-#!/usr/bin/python3.8
-
-import hashlib
-import logging
-import os
-import tarfile
-import uuid
-from typing import Optional, Tuple
-
-import allure
-import docker
-import wallet
-from cli_helpers import _cmd_run
-from common import ASSETS_DIR, SIMPLE_OBJ_SIZE
-
-logger = logging.getLogger("NeoLogger")
-
-
-def generate_file(size: int = SIMPLE_OBJ_SIZE) -> str:
-    """
-    Function generates a binary file with the specified size in bytes.
-    Args:
-        size (int): the size in bytes, can be declared as 6e+6 for example
-    Returns:
-        (str): the path to the generated file
-    """
-    file_path = f"{os.getcwd()}/{ASSETS_DIR}/{str(uuid.uuid4())}"
-    with open(file_path, "wb") as fout:
-        fout.write(os.urandom(size))
-    logger.info(f"file with size {size} bytes has been generated: {file_path}")
-
-    return file_path
-
-
-@allure.step("Generate file")
-def generate_file_and_file_hash(size: int) -> Tuple[str, str]:
-    """
-    Function generates a binary file with the specified size in bytes
-    and its hash.
-    Args:
-        size (int): the size in bytes, can be declared as 6e+6 for example
-    Returns:
-        (str): the path to the generated file
-        (str): the hash of the generated file
-    """
-    file_path = generate_file(size)
-    file_hash = get_file_hash(file_path)
-
-    return file_path, file_hash
-
-
-@allure.step("Get File Hash")
-def get_file_hash(filename: str, len: Optional[int] = None):
-    """
-    This function generates hash for the specified file.
-    Args:
-        filename (str): the path to the file to generate hash for
-        len (int): how many bytes to read
-    Returns:
-        (str): the hash of the file
-    """
-    file_hash = hashlib.sha256()
-    with open(filename, "rb") as out:
-        if len:
-            file_hash.update(out.read(len))
-        else:
-            file_hash.update(out.read())
-    return file_hash.hexdigest()
-
-
-@allure.step("Generate Wallet")
-def generate_wallet():
-    return wallet.init_wallet(ASSETS_DIR)
-
-
-@allure.step("Get Docker Logs")
-def get_container_logs(testcase_name: str) -> None:
-    client = docker.APIClient(base_url="unix://var/run/docker.sock")
-    logs_dir = os.getenv("${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()
-
-
-@allure.step("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)
-
-
-@allure.step("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)
-
-
-@allure.step("Concatenation set of files to one file")
-def concat_files(list_of_parts: list, new_file_name: Optional[str] = None) -> str:
-    """
-    Concatenates a set of files into a single file.
-    Args:
-        list_of_parts (list): list with files to concratination
-        new_file_name (str): file name to the generated file
-    Returns:
-        (str): the path to the generated file
-    """
-    if not new_file_name:
-        new_file_name = f"{os.getcwd()}/{ASSETS_DIR}/{str(uuid.uuid4())}"
-    with open(new_file_name, "wb") as f:
-        for file in list_of_parts:
-            with open(file, "rb") as part_file:
-                f.write(part_file.read())
-    return new_file_name