mirror of
https://github.com/aquasecurity/kube-hunter.git
synced 2026-05-19 23:59:28 +00:00
6d63f55d18
* Fix "none" logging Test for different logging levels, existing and none existing Co-authored-by: yoavrotems <yoavrotems97@gmail.com> Co-authored-by: Yehuda Chikvashvili <yehudaac1@gmail.com>
23 lines
643 B
Python
23 lines
643 B
Python
import logging
|
|
|
|
from kube_hunter.conf.logging import setup_logger
|
|
|
|
|
|
def test_setup_logger_level():
|
|
test_cases = [
|
|
("INFO", logging.INFO),
|
|
("Debug", logging.DEBUG),
|
|
("critical", logging.CRITICAL),
|
|
("NOTEXISTS", logging.INFO),
|
|
("BASIC_FORMAT", logging.INFO),
|
|
]
|
|
for level, expected in test_cases:
|
|
setup_logger(level)
|
|
actual = logging.getLogger().getEffectiveLevel()
|
|
assert actual == expected, f"{level} level should be {expected} (got {actual})"
|
|
|
|
|
|
def test_setup_logger_none():
|
|
setup_logger("NONE")
|
|
assert logging.getLogger().manager.disable == logging.CRITICAL
|