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
|
# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Tests for the logging utility functions."""
import logging
import sys
from contextlib import ExitStack as does_not_raise
from pathlib import Path
from typing import Any
from typing import Optional
import pytest
from mlia.cli.logging import create_log_handler
@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: Optional[Path],
stream: Optional[Any],
log_filter: Optional[logging.Filter],
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)
|
