OwlCyberSecurity - MANAGER

Edit File: check_log.py

from __future__ import annotations
from collections.abc import Iterable
from typing import Callable

from .pseudo_traceback import _build_pseudo_trace_str

should_use_color = False
COLOR_RED = "\x1b[31m"
COLOR_RESET = "\x1b[0m"
_failures: list[str] = []
_stop_on_fail = False

_default_max_fail = None
_default_max_report = None
_default_max_tb = 1

_max_fail: int | None = _default_max_fail
_max_report: int | None = _default_max_report
_max_tb = _default_max_tb
_num_failures = 0
_fail_function: Callable[[str], None] | None = None

_showlocals = False

def clear_failures() -> None:
    # gets called at the beginning of each test function
    global _failures, _num_failures
    global _max_fail, _max_report, _max_tb
    _failures = []
    _num_failures = 0
    _max_fail = _default_max_fail
    _max_report = _default_max_report
    _max_tb = _default_max_tb

def any_failures() -> bool:
    return bool(get_failures())

def get_failures() -> list[str]:
    return _failures

def log_failure(
    msg: object = "", check_str: str = "", tb: Iterable[str] | None = None
) -> None:
    global _num_failures
    __tracebackhide__ = True
    _num_failures += 1

msg = str(msg).strip()

if check_str:
        msg = f"{msg}: {check_str}"

if (_max_report is None) or (_num_failures <= _max_report):
        if _num_failures <= _max_tb:
            pseudo_trace_str = _build_pseudo_trace_str(_showlocals,
                                                       tb,
                                                       should_use_color)
            msg = f"{msg}\n{pseudo_trace_str}"

if should_use_color:
            msg = f"{COLOR_RED}FAILURE: {COLOR_RESET}{msg}"
        else:
            msg = f"FAILURE: {msg}"
        _failures.append(msg)
        if _fail_function:
            _fail_function(str(msg))

if _max_fail and (_num_failures >= _max_fail):
        assert_msg = f"pytest-check max fail of {_num_failures} reached"
        assert _num_failures < _max_fail, assert_msg

if _stop_on_fail:
        assert False, "Stopping on first failure"