Source code for testapp.util

import io
import re
import sys
from functools import wraps
from pathlib import Path
from typing import Union

from testapp.constants import SSD_MIN_VALUE, SSD_MAX_VALUE, SSD_START_LBA, SSD_END_LBA



RESULT_FILE = "result.txt"



READ_COMMAND = "R"



WRITE_COMMAND = "W"



ERASE_COMMAND = "E"



FLUSH_COMMAND = "F"



SSD_COMMAND = "ssd"



SSD_MODULE = "ssd"




BASE_DIR = Path(__file__).parent.parent





def get_ssd_result() -> str:
    file_path = BASE_DIR / SSD_MODULE / RESULT_FILE
    with open(file_path, "r", encoding="utf-8") as f:
        return f.read()





def validate_ssd_command(command: str) -> None:
    if not isinstance(command, str):
        raise TypeError
    args = command.split(" ")
    if args[0] != SSD_COMMAND:
        raise ValueError(f"Applied wrong command: {command}")
    op = args[1]
    if op == WRITE_COMMAND and len(args) == 4:  # ssd W lba value
        validate_ssd_lba(args[2])
        validate_ssd_value(args[3])
    elif op == READ_COMMAND and len(args) == 3:  # ssd R lba
        validate_ssd_lba(args[2])
    elif op == ERASE_COMMAND and len(args) == 4:  # ssd E lba total_size
        validate_ssd_lba(args[2])
    elif op == FLUSH_COMMAND and len(args) == 2:  # ssd F
        pass
    else:
        raise ValueError(f"Applied wrong command: {command}")





def validate_ssd_lba(lba: Union[str, int]) -> None:
    if not isinstance(lba, (str, int)):
        raise TypeError("LBA는 숫자 타입이여야 합니다.")
    elif not 0 <= int(lba) < 100:
        raise ValueError("LBA는 0에서 99사이의 정수여야 합니다.")




def validate_ssd_end_lba(lba: Union[str, int]) -> None:
    if not isinstance(lba, (str, int)):
        raise TypeError("LBA는 숫자 타입이여야 합니다.")
    elif not 0 <= int(lba) <= 100:
        raise ValueError("End LBA는 0에서 100사이의 정수여야 합니다.")





def validate_ssd_value(value: Union[str, int]) -> None:
    if not isinstance(value, (str, int)):
        raise TypeError("value는 16진수 형식의 정수값만 허용됩니다.")
    elif isinstance(value, str):
        if not value.startswith("0x"):
            raise ValueError("value는 0x00000000 형식이여야 합니다.")
        try:
            value = int(value, 16)  # 16 진수 to
        except:  # noqa
            raise ValueError("value가 16진수가 아닙니다.")
    if not SSD_MIN_VALUE <= value <= SSD_MAX_VALUE:
        raise ValueError("value는 0x00000000에서 0xFFFFFFFF 사이의 값이여야 합니다.")





def capture_output(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        output = io.StringIO()
        sys.stdout = output
        try:
            func(*args, **kwargs)
        finally:
            sys.stdout = sys.__stdout__
        captured_output = output.getvalue()
        return captured_output

    return wrapper





def is_in_range_lba(lba: str) -> bool:
    try:
        num = int(lba)
        return SSD_START_LBA <= num <= SSD_END_LBA
    except ValueError:
        return False





def is_valid_hex(s: str):
    if not re.fullmatch(r"0x[0-9A-Fa-f]{8}", s):
        raise ValueError("value는 0x00000000 형식이여야 합니다.")

    num = int(s, 16)
    if not SSD_MIN_VALUE <= num <= SSD_MAX_VALUE:
        raise ValueError("value는 0x00000000부터 0xFFFFFFFF 사이의 값이여야 합니다.")





def validate_size(size: str):
    num = int(size)
    if 1 > num:
        raise ValueError("size는 1이상의 정수여야 합니다.")