1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Tests for the logging utility functions."""
from __future__ import annotations
import logging
import sys
from contextlib import ExitStack as does_not_raise
from pathlib import Path
from typing import Any
from typing import Callable
from unittest.mock import MagicMock
import pytest
from mlia.utils.logging import capture_raw_output
from mlia.utils.logging import create_log_handler
from mlia.utils.logging import redirect_output
from mlia.utils.logging import redirect_raw_output
@pytest.mark.parametrize(
"file_path, stream, log_filter, delay, expected_error, expected_class",
[
(
"test.log",
None,
None,
True,
does_not_raise(),
logging.FileHandler,
),
(
None,
sys.stdout,
None,
None,
does_not_raise(),
logging.StreamHandler,
),
(
None,
None,
None,
None,
pytest.raises(Exception, match="Unable to create logging handler"),
None,
),
],
)
def test_create_log_handler(
file_path: Path | None,
stream: Any | None,
log_filter: logging.Filter | None,
delay: bool,
expected_error: Any,
expected_class: type,
) -> None:
"""Test function test_create_log_handler."""
with expected_error:
handler = create_log_handler(
file_path=file_path,
stream=stream,
log_level=logging.INFO,
log_format="%(name)s - %(message)s",
log_filter=log_filter,
delay=delay,
)
assert isinstance(handler, expected_class)
@pytest.mark.parametrize(
"redirect_context_manager",
[
redirect_raw_output,
redirect_output,
],
)
def test_output_redirection(redirect_context_manager: Callable) -> None:
"""Test output redirection via context manager."""
print("before redirect")
logger_mock = MagicMock()
with redirect_context_manager(logger_mock):
print("output redirected")
print("after redirect")
logger_mock.log.assert_called_once_with(logging.INFO, "output redirected")
def test_output_and_error_capture() -> None:
"""Test output/error capturing."""
with capture_raw_output(sys.stdout) as std_output, capture_raw_output(
sys.stderr
) as stderr_output:
print("hello from stdout")
print("hello from stderr", file=sys.stderr)
assert std_output == ["hello from stdout\n"]
assert stderr_output == ["hello from stderr\n"]
|
