mirror of
https://github.com/aquasecurity/kube-hunter.git
synced 2026-05-20 08:06:24 +00:00
24 lines
677 B
Python
24 lines
677 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),
|
|
]
|
|
logFile = None
|
|
for level, expected in test_cases:
|
|
setup_logger(level, logFile)
|
|
actual = logging.getLogger().getEffectiveLevel()
|
|
assert actual == expected, f"{level} level should be {expected} (got {actual})"
|
|
|
|
|
|
def test_setup_logger_none():
|
|
setup_logger("NONE", None)
|
|
assert logging.getLogger().manager.disable == logging.CRITICAL
|